fix: Psycopg2

Author: Gu1nness

lib/charms/data_platform_libs/v1/data_interfaces.py

@@ -287,9 +287,9 @@ def _on_database_requested(self, event: DatabaseRequestedEvent) -> None:
 from typing_extensions import TypeAliasType, override
 
 try:
-    import psycopg
+    import psycopg2
 except ImportError:
-    psycopg = None
+    psycopg2 = None
 
 # The unique Charmhub library identifier, never change it
 LIBID = "6c3e6b6680d64e9c89e611d1a15f65be"
@@ -2488,17 +2488,17 @@ def are_all_resources_created(self, rel_id: int) -> bool:
     def _is_pg_plugin_enabled(plugin: str, connection_string: str) -> bool:
         # Actual checking method.
         # No need to check for psycopg here, it's been checked before.
-        if not psycopg:
+        if not psycopg2:
             return False
 
         try:
-            with psycopg.connect(connection_string) as connection:
+            with psycopg2.connect(connection_string) as connection:
                 with connection.cursor() as cursor:
                     cursor.execute(
                         "SELECT TRUE FROM pg_extension WHERE extname=%s::text;", (plugin,)
                     )
                     return cursor.fetchone() is not None
-        except psycopg.Error as e:
+        except psycopg2.Error as e:
             logger.exception(
                 f"failed to check whether {plugin} plugin is enabled in the database: %s",
                 str(e),
@@ -2512,7 +2512,7 @@ def is_postgresql_plugin_enabled(self, plugin: str, relation_id: int = 0) -> boo
             plugin: name of the plugin to check.
             relation_id: Optional index to check the database (default: 0 - first relation).
         """
-        if not psycopg:
+        if not psycopg2:
             return False
 
         # Can't check a non existing relation.