Use engine.connect

ddxv · Oct 3, 2024 · ec14626 · ec14626
1 parent ff3919d
commit ec14626
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/adscrawler/connection.py b/adscrawler/connection.py
@@ -22,7 +22,7 @@ def __init__(self, db_name: str, db_ip: str, db_port: str) -> None:
         self.db_name = db_name
         self.db_ip = db_ip
         self.db_port = db_port
-        self.engine: Engine | None = None
+        self.engine: Engine
 
         try:
             self.db_pass = CONFIG[self.db_name]["db_password"]

diff --git a/adscrawler/queries.py b/adscrawler/queries.py
@@ -267,10 +267,10 @@ def query_pub_domains(
 def delete_app_url_mapping(app_url_id: int, database_connection: PostgresCon) -> None:
     del_query = text("DELETE FROM app_urls_map WHERE id = :app_url_id")
     logger.info(f"{app_url_id=} delete app_urls_map start")
-    raw_conn = database_connection.engine.raw_connection()
-    with raw_conn.cursor() as cur:
-        cur.execute(del_query, {"app_url_id": app_url_id})
 
+    with database_connection.engine.connect() as conn:
+        conn.execute(del_query, {"app_url_id": app_url_id})
+        conn.commit()
 
 
 def query_store_apps(