scitex 2.0.0__py2.py3-none-any.whl

scitex/db/_PostgreSQLMixins/_TransactionMixin.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-02-27 22:15:42 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex_dev/src/scitex/db/_PostgreSQLMixins/_TransactionMixin.py
+
+THIS_FILE = (
+    "/home/ywatanabe/proj/scitex_repo/src/scitex/db/_PostgreSQLMixins/_TransactionMixin.py"
+)
+
+import psycopg2
+from .._BaseMixins._BaseTransactionMixin import _BaseTransactionMixin
+
+
+class _TransactionMixin(_BaseTransactionMixin):
+    def begin(self) -> None:
+        self.execute("BEGIN TRANSACTION")
+
+    def commit(self) -> None:
+        self.conn.commit()
+
+    def rollback(self) -> None:
+        self.conn.rollback()
+
+    def enable_foreign_keys(self) -> None:
+        # In PostgreSQL, foreign key constraints are always enabled
+        pass
+
+    def disable_foreign_keys(self) -> None:
+        # Warning: This is session-level and should be used carefully
+        self.execute("SET session_replication_role = 'replica'")
+
+    @property
+    def writable(self) -> bool:
+        try:
+            self.cursor.execute(
+                "SELECT current_setting('transaction_read_only') = 'off'"
+            )
+            return self.cursor.fetchone()[0]
+        except psycopg2.Error:
+            return True
+
+    @writable.setter
+    def writable(self, state: bool) -> None:
+        try:
+            if state:
+                self.execute("SET TRANSACTION READ WRITE")
+            else:
+                self.execute("SET TRANSACTION READ ONLY")
+        except psycopg2.Error as err:
+            raise ValueError(f"Failed to set writable state: {err}")
+
+    def _check_writable(self) -> None:
+        if not self.writable:
+            raise ValueError("Database is in read-only mode")
+
+
+# EOF

scitex/db/_PostgreSQLMixins/__init__.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Time-stamp: "2025-02-27 22:13:18 (ywatanabe)"
+# File: ./scitex_repo/src/scitex/db/_PostgreSQLMixins/__init__.py
+
+from ._BackupMixin import _BackupMixin
+from ._BatchMixin import _BatchMixin
+from ._BlobMixin import _BlobMixin
+from ._ConnectionMixin import _ConnectionMixin
+from ._ImportExportMixin import _ImportExportMixin
+from ._IndexMixin import _IndexMixin
+from ._MaintenanceMixin import _MaintenanceMixin
+from ._QueryMixin import _QueryMixin
+from ._RowMixin import _RowMixin
+from ._SchemaMixin import _SchemaMixin
+from ._TableMixin import _TableMixin
+from ._TransactionMixin import _TransactionMixin
+
+__all__ = [
+    "_BackupMixin",
+    "_BatchMixin",
+    "_BlobMixin",
+    "_ConnectionMixin",
+    "_ImportExportMixin",
+    "_IndexMixin",
+    "_MaintenanceMixin",
+    "_QueryMixin",
+    "_RowMixin",
+    "_SchemaMixin",
+    "_TableMixin",
+    "_TransactionMixin",
+]
+
+# EOF