Merge pull request #13 from semiotic-ai/custom-pg-ports

Added custom ports for postgres

autoagora_processor/db_utils.py

@@ -34,6 +34,13 @@
     required=True,
     help="Password for the graph-node database.",
 )
+argsparser.add_argument(
+    "--graph-postgres-port ",
+    env_var="GRAPH_POSTGRES_PORT",
+    default=5432,
+    required=True,
+    help="Port for the graph-node database.",
+)
 
 
 def get_indexed_subgraphs() -> Set[str]:
@@ -42,6 +49,7 @@ def get_indexed_subgraphs() -> Set[str]:
         user=args.graph_postgres_username,
         password=args.graph_postgres_password,
         host=args.graph_postgres_host,
+        port=args.graph_postgres_port,
     )
     cur = conn.cursor()
 

autoagora_processor/logs_db.py

@@ -15,8 +15,11 @@ def __init__(
         user: str,
         password: str,
         host: str,
+        port: str,
     ) -> None:
-        self.conn = pg.connect(dbname=dbname, user=user, password=password, host=host)
+        self.conn = pg.connect(
+            dbname=dbname, user=user, password=password, host=host, port=port
+        )
         self.conn.autocommit = True
         self.cur = self.conn.cursor()
 

autoagora_processor/main.py

@@ -105,6 +105,13 @@
     required=True,
     help="Password for the logs database.",
 )
+argsparser.add_argument(
+    "--postgres-port",
+    env_var="POSTGRES_PORT",
+    default=5432,
+    required=True,
+    help="Port for the logs database.",
+)
 argsparser.add_argument(
     "--log-level",
     env_var="LOG_LEVEL",
@@ -121,6 +128,7 @@
     user=args.postgres_username,
     password=args.postgres_password,
     host=args.postgres_host,
+    port=args.postgress_port,
 )
 
 subgraph_schemas = SubgraphSchemas()