checkmate5 4.0.92__py3-none-any.whl → 4.0.93__py3-none-any.whl

checkmate/lib/backend.py

@@ -0,0 +1,119 @@
+from blitzdb import FileBackend as BlitzDBFileBackend
+from blitzdb.backends.sql import Backend as BlitzDBSQLBackend
+from typing import Any, Optional, Dict
+import logging
+from contextlib import contextmanager
+
+logger = logging.getLogger(__name__)
+
+class FileBackend(BlitzDBFileBackend):
+    def __init__(self, path: str) -> None:
+        """Initialize FileBackend with specified path.
+        
+        Args:
+            path: Path to store the database files
+            
+        Raises:
+            ValueError: If path is invalid
+        """
+        logger.debug(f"Initializing FileBackend with path: {path}")
+        super().__init__(path)
+        self.path = path
+
+    def test_connection(self) -> bool:
+        """Test the file access.
+        
+        Returns:
+            bool: True if file access is successful
+            
+        Raises:
+            ConnectionError: If file access test fails
+        """
+        try:
+            self.begin()
+            self.commit()
+            logger.info("File access test successful")
+            return True
+        except Exception as e:
+            logger.error(f"File access test failed: {str(e)}")
+            raise ConnectionError(f"File access test failed: {str(e)}")
+
+    @contextmanager
+    def session(self):
+        """Context manager for file operations.
+        
+        Yields:
+            FileBackend: 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
+
+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

@@ -9,7 +9,7 @@ from functools import reduce
 import argparse
 from sqlalchemy import create_engine, MetaData, Table, Column, Integer, String
 from sqlalchemy.exc import SQLAlchemyError
-from ...lib.models import Backend
+from checkmate.lib.backend import Backend
 
 def get_project_path(path=None):
     if not path:

{checkmate5-4.0.92.dist-info → checkmate5-4.0.93.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: checkmate5
-Version: 4.0.92
+Version: 4.0.93
 Summary: A meta-code checker written in Python.
 Author: Andreas Dewes
 License: AGPL-3.0

{checkmate5-4.0.92.dist-info → checkmate5-4.0.93.dist-info}/RECORD

@@ -71,6 +71,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=dO3rMqI70D8-9CrgL3qu3fqKf0BXvJCDIgSIys8IRyM,3683
 checkmate/lib/models.py,sha256=Myzjf7ChJnObPpMLXv34znMXkeP0Fma54iUN9uSl8AI,20191
 checkmate/lib/analysis/__init__.py,sha256=_JpM1GkChWCfLKqPqEz3-8DCPwNe7lPwQDMoF_6Ore0,45
 checkmate/lib/analysis/base.py,sha256=R9Zy6rKKCw1LSAsBBBaBbgFEo6Fkdx8DTtp7bsYoywE,3309
@@ -81,7 +82,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=XSvNilmlDiw-qubAN2oVtMHcRz6VAMaOYmULfYLukvs,5403
+checkmate/management/helpers.py,sha256=8AQ2qNBBJ16H4T4G5FolH9RhdLQ1adVdzxrkLEaxcsU,5411
 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
@@ -109,9 +110,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=HJoZIin2BHT_gBCngGZNh7WgSNLqIDTK--iW7hXez4Q,5084
-checkmate5-4.0.92.dist-info/LICENSE.txt,sha256=9cSf5btiPmglu3yzhXkXcf7GNT8XC9o7Y54tfhvfDp8,213739
-checkmate5-4.0.92.dist-info/METADATA,sha256=KMpTUwrZ2KV2_07ZoWCxbdndGqcEihZMDL6XLhdunlE,1234
-checkmate5-4.0.92.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-checkmate5-4.0.92.dist-info/entry_points.txt,sha256=FbGnau5C4z98WmBYpMJqUzobQEr1AIi9aZApSavNojQ,60
-checkmate5-4.0.92.dist-info/top_level.txt,sha256=tl6eIJXedpLZbcbmYEwlhEzuTaSt0TvIRUesOb8gtng,10
-checkmate5-4.0.92.dist-info/RECORD,,
+checkmate5-4.0.93.dist-info/LICENSE.txt,sha256=9cSf5btiPmglu3yzhXkXcf7GNT8XC9o7Y54tfhvfDp8,213739
+checkmate5-4.0.93.dist-info/METADATA,sha256=aTicVld8Hcve_pknVkSm8KttlJgOF5HL_tp748EJoqc,1234
+checkmate5-4.0.93.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+checkmate5-4.0.93.dist-info/entry_points.txt,sha256=FbGnau5C4z98WmBYpMJqUzobQEr1AIi9aZApSavNojQ,60
+checkmate5-4.0.93.dist-info/top_level.txt,sha256=tl6eIJXedpLZbcbmYEwlhEzuTaSt0TvIRUesOb8gtng,10
+checkmate5-4.0.93.dist-info/RECORD,,