flask-postgresql-audit 1.0.6__tar.gz → 1.0.7__tar.gz

{flask_postgresql_audit-1.0.6 → flask_postgresql_audit-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flask-postgresql_audit
-Version: 1.0.6
+Version: 1.0.7
 Summary: Versioning and auditing extension for PostgreSQL and SQLAlchemy.
 Project-URL: Homepage, https://github.com/higamigu/flask-postgresql-audit
 Author-email: Higamigu <higamigu@gmail.com>, Konsta Vesterinen <konsta@fastmonkeys.com>

{flask_postgresql_audit-1.0.6 → flask_postgresql_audit-1.0.7}/flask_postgresql_audit/__init__.py

@@ -1,4 +1,4 @@
 # ruff: noqa: F401
 from .core import Audit, PostgreSQLAudit
 
-__version__ = "1.0.6"
+__version__ = "1.0.7"

{flask_postgresql_audit-1.0.6 → flask_postgresql_audit-1.0.7}/flask_postgresql_audit/core.py

@@ -251,7 +251,13 @@ class PostgreSQLAudit:
     def fetch_activity(self, obj: type[Audit] | Audit | t.Sequence[Audit]):
         if inspect.isclass(obj):
             source = obj
-            activity_join_cond = self.Activity.table_name == source.__tablename__
+            relid = sa.text(f"'{source.__tablename__}'::regclass::oid")
+            jsonb = sa.func.to_jsonb(source.__table__.table_valued())
+            activity_join_cond = sa.and_(
+                self.Activity.table_name == source.__tablename__,
+                self.Activity.row_key == self.func.get_pk_values(relid, jsonb),
+            )
+
         else:
             objects = list(obj) if isinstance(obj, t.Sequence) else list([obj])
             unions = []

{flask_postgresql_audit-1.0.6 → flask_postgresql_audit-1.0.7}/tests/app.py

@@ -14,11 +14,8 @@ class DefaultConfig:
     DB_PASSWORD = os.environ.get("POSTGRESQL_AUDIT_TEST_PASSWORD", "")
     DB_NAME = os.environ.get("POSTGRESQL_AUDIT_TEST_DB", "postgresql_audit_test")
 
-    # SQLALCHEMY_DATABASE_URI = (
-    #     f"postgresql://{DB_USER}:{DB_PASSWORD}@localhost/{DB_NAME}"
-    # )
     SQLALCHEMY_DATABASE_URI = (
-        "postgresql://postgres:postgres@localhost:5492/pg_audit_test"
+        f"postgresql://{DB_USER}:{DB_PASSWORD}@localhost/{DB_NAME}"
     )
     SQLALCHEMY_ECHO = False