checkmate5 4.1.0.dev34__py3-none-any.whl → 4.1.0.dev36__py3-none-any.whl

checkmate/lib/backend.py

@@ -6,4 +6,67 @@ from ..helpers.exceptions import DatabaseError
 
 logger = logging.getLogger(__name__)
 
-Backend = BlitzDBSQLBackend
+
+class SQLBackend(BlitzDBSQLBackend):
+    def __init__(self, connection: Any) -> None:
+        """Initialize SQL Backend with database connection.
+        
+        Args:
+            connection: SQLAlchemy engine or connection object
+            
+        Raises:
+            ConnectionError: If database connection fails
+        """
+        logger.debug(f"Initializing SQL Backend with connection: {connection}")
+        super().__init__(connection)
+        self.connection = connection
+
+    def test_connection(self) -> bool:
+        """Test the database connection.
+        
+        Returns:
+            bool: True if connection is successful
+            
+        Raises:
+            ConnectionError: If connection test fails
+        """
+        try:
+            with self.connection.connect() as conn:
+                conn.execute('SELECT 1')
+            logger.info("Database connection test successful")
+            return True
+        except Exception as e:
+            logger.error(f"Database connection test failed: {str(e)}")
+            raise ConnectionError(f"Database connection test failed: {str(e)}")
+
+    @contextmanager
+    def session(self):
+        """Context manager for database sessions.
+        
+        Yields:
+            SQLBackend: Self with active session
+        """
+        try:
+            self.begin()
+            yield self
+            self.commit()
+        except Exception as e:
+            logger.error(f"Session error: {str(e)}")
+            self.rollback()
+            raise
+        finally:
+            if hasattr(self.connection, 'dispose'):
+                self.connection.dispose()
+
+    def close(self) -> None:
+        """Clean up database connections."""
+        try:
+            if hasattr(self.connection, 'dispose'):
+                self.connection.dispose()
+            logger.info("Closed SQL Backend connection")
+        except Exception as e:
+            logger.error(f"Error closing backend: {str(e)}")
+            raise
+
+# For backward compatibility
+Backend = SQLBackend  # Default to SQLBackend for existing code

checkmate/management/helpers.py

@@ -106,8 +106,8 @@ def get_project(project_path, project_config, settings, backend):
     project_class = project_config.get('project_class', 'Project')
     ProjectClass = settings.models[project_class]
 
-    # Attempt to find the project by project_id
-    project = backend.filter({'pk': project_config['project_id']}).first()
+    project = backend.filter(ProjectClass).first()
+
 
     # If the project doesn't exist, create a new one
     if project is None:

{checkmate5-4.1.0.dev34.dist-info → checkmate5-4.1.0.dev36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: checkmate5
-Version: 4.1.0.dev34
+Version: 4.1.0.dev36
 Summary: A meta-code checker written in Python.
 Author: Andreas Dewes
 License: AGPL-3.0

{checkmate5-4.1.0.dev34.dist-info → checkmate5-4.1.0.dev36.dist-info}/RECORD

@@ -72,7 +72,7 @@ checkmate/helpers/hashing.py,sha256=TcwBFJlciJkLlETQcQBxEWRgwyh2HK9MO4BzN2VMlDU,
 checkmate/helpers/issue.py,sha256=7wImtI8uZR5VcE_1u7mI8qvEXU97p6Rzkb1bbJkqXKQ,3841
 checkmate/helpers/settings.py,sha256=97zsz4vNq7EpUpRME4XQvvp5LUp3618ZvSfNTce_4j4,322
 checkmate/lib/__init__.py,sha256=iwhKnzeBJLKxpRVjvzwiRE63_zNpIBfaKLITauVph-0,24
-checkmate/lib/backend.py,sha256=7302LiR5NTaePEsBDxdIHUFqMfnzvMKZi4SyZ7Mc8ms,268
+checkmate/lib/backend.py,sha256=_mMw7snfCdjiy3Hh0buSJt2AXFB-SP09-Xbx1YzG2PU,2305
 checkmate/lib/models.py,sha256=bmS_QDDBEMaqC0TiKNaoRBCHuVUK9S-TSBB7BYYJz7Q,7404
 checkmate/lib/analysis/__init__.py,sha256=_JpM1GkChWCfLKqPqEz3-8DCPwNe7lPwQDMoF_6Ore0,45
 checkmate/lib/analysis/base.py,sha256=R9Zy6rKKCw1LSAsBBBaBbgFEo6Fkdx8DTtp7bsYoywE,3309
@@ -83,7 +83,7 @@ checkmate/lib/stats/helpers.py,sha256=X21Pb7dz3PWubDRoqUjbxwonl43TRxqKEcwyM80tS6
 checkmate/lib/stats/mapreduce.py,sha256=rGdA_HninEZSeowtBn5R02c6wyT5dxcu-0T2ftlMnY0,782
 checkmate/management/__init__.py,sha256=iwhKnzeBJLKxpRVjvzwiRE63_zNpIBfaKLITauVph-0,24
 checkmate/management/decorators.py,sha256=iwhKnzeBJLKxpRVjvzwiRE63_zNpIBfaKLITauVph-0,24
-checkmate/management/helpers.py,sha256=Kdd7P876dPeU0h6Stovbe3pSn7MmbZNrSVKvrSo_hAc,4619
+checkmate/management/helpers.py,sha256=k4STjKlijvZvkC3EJc2IKGycU23bBbrjctYDZbO_umI,4548
 checkmate/management/commands/__init__.py,sha256=XAi0y8z1NviyGvLB68Oxnzr6Nw5AP8xgbcSSnc1Zcvw,766
 checkmate/management/commands/alembic.py,sha256=DTWH8hcY6UmWDo0wMwcID_wQIa-es7nyJHV6p1_T4FQ,718
 checkmate/management/commands/analyze.py,sha256=J1rawVEGjbrXT9ArfUVpfoLNs2JaFkV4mCrBnEIrXaw,1159
@@ -111,9 +111,9 @@ checkmate/scripts/manage.py,sha256=mpioBaxzirAKXZtbxO-y4dbOcc6UoP0MaAMsNuKHbz0,4
 checkmate/settings/__init__.py,sha256=z32hPz-kGS-tTGa6dWCFjrrrbS_eagLd-YrqBP3gjWI,33
 checkmate/settings/base.py,sha256=3WBXZITqoWepIja96bo5JTi-TDpQALPTCugL0E8z-yE,4551
 checkmate/settings/defaults.py,sha256=uK1KB50ukDbk2rACyiCQuXTNSr2M7GXE-2_GTFupbv0,2728
-checkmate5-4.1.0.dev34.dist-info/licenses/LICENSE.txt,sha256=SGQTFjJQjkYGoK1PCFfMKpfgRLm3yL0h9Mq2o26sm2E,151451
-checkmate5-4.1.0.dev34.dist-info/METADATA,sha256=mx1GYxnsddRZAcS5nqS1auVwGWgiVRBkfQhtq3vBZok,1262
-checkmate5-4.1.0.dev34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-checkmate5-4.1.0.dev34.dist-info/entry_points.txt,sha256=FbGnau5C4z98WmBYpMJqUzobQEr1AIi9aZApSavNojQ,60
-checkmate5-4.1.0.dev34.dist-info/top_level.txt,sha256=tl6eIJXedpLZbcbmYEwlhEzuTaSt0TvIRUesOb8gtng,10
-checkmate5-4.1.0.dev34.dist-info/RECORD,,
+checkmate5-4.1.0.dev36.dist-info/licenses/LICENSE.txt,sha256=SGQTFjJQjkYGoK1PCFfMKpfgRLm3yL0h9Mq2o26sm2E,151451
+checkmate5-4.1.0.dev36.dist-info/METADATA,sha256=Q-RqbapM82Y7q2BuyHVqCS96jkDK3LoDtUn_zWKUOOI,1262
+checkmate5-4.1.0.dev36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+checkmate5-4.1.0.dev36.dist-info/entry_points.txt,sha256=FbGnau5C4z98WmBYpMJqUzobQEr1AIi9aZApSavNojQ,60
+checkmate5-4.1.0.dev36.dist-info/top_level.txt,sha256=tl6eIJXedpLZbcbmYEwlhEzuTaSt0TvIRUesOb8gtng,10
+checkmate5-4.1.0.dev36.dist-info/RECORD,,