fix(python-driver): Python 3.9-safe hints and correct $user in search_path

- Use Optional[str] for configure_connection graph_name (PEP 604 unions are
  invalid on Python 3.9).
- Import Any/Optional from typing for annotations.
- Quote $user in SET search_path; align unit test expectations.

Made-with: Cursor

Author: uesleilima

drivers/python/age/age.py

@@ -14,6 +14,8 @@
 # under the License.
 
 import re
+from typing import Any, Optional
+
 import psycopg
 from psycopg.types import TypeInfo
 from psycopg import sql
@@ -144,7 +146,7 @@ def setUpAge(conn:psycopg.connection, graphName:str, load_from_plugins:bool=Fals
         else:
             cursor.execute("LOAD 'age';")
 
-        cursor.execute("SET search_path = ag_catalog, '$user', public;")
+        cursor.execute('SET search_path = ag_catalog, "$user", public;')
 
         ag_info = TypeInfo.fetch(conn, 'agtype')
 
@@ -165,7 +167,7 @@ def setUpAge(conn:psycopg.connection, graphName:str, load_from_plugins:bool=Fals
 
 def configure_connection(
     conn: psycopg.connection,
-    graph_name: str | None = None,
+    graph_name: Optional[str] = None,
     load: bool = False,
     load_from_plugins: bool = False,
 ) -> None:
@@ -209,7 +211,7 @@ def configure_connection(
             else:
                 cursor.execute("LOAD 'age';")
 
-        cursor.execute("SET search_path = ag_catalog, '$user', public;")
+        cursor.execute('SET search_path = ag_catalog, "$user", public;')
 
         ag_info = TypeInfo.fetch(conn, 'agtype')
 

drivers/python/test_age_py.py

@@ -122,7 +122,7 @@ def test_default_does_not_load(self):
              unittest.mock.patch("age.age.checkGraphCreated"):
             age.age.configure_connection(mock_conn)
         mock_cursor.execute.assert_called_once_with(
-            "SET search_path = ag_catalog, '$user', public;"
+            'SET search_path = ag_catalog, "$user", public;'
         )
 
     def test_load_true_executes_load(self):
@@ -154,7 +154,7 @@ def test_always_sets_search_path(self):
              unittest.mock.patch("age.age.checkGraphCreated"):
             age.age.configure_connection(mock_conn)
         mock_cursor.execute.assert_any_call(
-            "SET search_path = ag_catalog, '$user', public;"
+            'SET search_path = ag_catalog, "$user", public;'
         )
 
     def test_registers_agtype_adapters(self):