Fixed DB namespace issue

Author: root

api/Comment.py

@@ -3,7 +3,7 @@
 from collections import defaultdict
 import Parameters
 from Helpers import *
-
+import DBFunctions
 
 class search:
     params = None
@@ -30,7 +30,7 @@ def getIds(self,ids):
             if id[:3] == "t1_":
                 id = id[3:]
             ids_to_get_from_db.append(base36decode(id))
-        rows = pgdb.execute("SELECT * FROM comment WHERE (json->>'id')::bigint IN %s LIMIT 5000",tuple(ids_to_get_from_db))
+        rows = DBFunctions.pgdb.execute("SELECT * FROM comment WHERE (json->>'id')::bigint IN %s LIMIT 5000",tuple(ids_to_get_from_db))
         results = []
         data = {}
         if rows:

api/DBFunctions.py

@@ -34,3 +34,6 @@ def execute(self,sql,params):
                     self.connect()
                 except:
                     raise
+
+pgdb = pgdb()
+

api/Helpers.py

@@ -1,6 +1,7 @@
 from collections import defaultdict
 import requests
 import json
+import DBFunctions
 
 def LooksLikeInt(s):
     try:
@@ -55,7 +56,7 @@ def getSubmissionsFromPg(ids):
     if not isinstance(ids, (list, tuple)):
         ids = [ids]
     ids_to_get_from_db = []
-    rows = pgdb.execute("SELECT * FROM submission WHERE (json->>'id')::int IN %s LIMIT 5000",tuple(ids))
+    rows = DBFunctions.pgdb.execute("SELECT * FROM submission WHERE (json->>'id')::int IN %s LIMIT 5000",tuple(ids))
     results = {}
     data = {}
     if rows:

api/api.py

@@ -77,7 +77,7 @@ def on_get(self, req, resp, submission_id):
         if submission_id[:3] == 't3_':
             submission_id = submission_id[3:]
         submission_id = base36decode(submission_id)
-        rows = pgdb.execute("SELECT (json->>'id')::bigint comment_id FROM comment WHERE (json->>'link_id')::int = %s ORDER BY comment_id ASC LIMIT 50000",submission_id)
+        rows = DBFunctions.pgdb.execute("SELECT (json->>'id')::bigint comment_id FROM comment WHERE (json->>'link_id')::int = %s ORDER BY comment_id ASC LIMIT 50000",submission_id)
         results = []
         data = {}
         if rows:
@@ -96,7 +96,6 @@ def database_connection():
 config.read ('credentials.ini')
 DB_PASSWORD = config.get('database','password')
 DB_USER = config.get('database','user')
-pgdb = DBFunctions.pgdb()
 r = redis.StrictRedis(host='localhost', port=6379, db=1)
 api = falcon.API()
 api.add_route('/reddit/search', Comment.search())