bear-utils 0.8.22__py3-none-any.whl → 0.8.23__py3-none-any.whl

bear_utils/database/_db_manager.py

@@ -32,42 +32,35 @@ class DatabaseManager:
             raise ValueError("Base class is not set, failed to set base.")
         return cls._base
 
-    def __init__(self, db_url: str | Path | None = None):
+    def __init__(self, db_url: str | Path | None = None, default_schema: str = "sqlite:///"):
         if db_url is None or db_url == "":
             raise ValueError("Database URL cannot be None or empty.")
-        if isinstance(db_url, str) and not db_url.startswith("sqlite://"):
-            db_url = f"sqlite:///{db_url}"
+        if isinstance(db_url, str) and not db_url.startswith(default_schema):
+            db_url = f"{default_schema}{db_url}"
         self.db_url: str = str(db_url)
         self.engine: Engine = create_engine(self.db_url, echo=False)
         base: DeclarativeMeta = DatabaseManager.get_base()
         self.metadata: MetaData = base.metadata
-        self.SessionFactory = sessionmaker(bind=self.engine)
-        self.session = scoped_session(self.SessionFactory)
+        self.SessionFactory: sessionmaker[Session] = sessionmaker(bind=self.engine)
+        self.session: scoped_session[Session] = scoped_session(self.SessionFactory)
         atexit.register(self.close_all)
         self.create_tables()
 
     def get_all_records(self, table_obj: type[TableType]) -> list[TableType]:
         """Get all records from a table."""
-        with self.open_session() as session:
-            return session.query(table_obj).all()
+        return self.session().query(table_obj).all()
 
     def count_records(self, table_obj: type[TableType]) -> int:
         """Count the number of records in a table."""
-        with self.open_session() as session:
-            count: int = session.query(table_obj).count()
-        return count
+        return self.session().query(table_obj).count()
 
     def get_records_by_var(self, table_obj: type[TableType], variable: str, value: str) -> list[TableType]:
         """Get records from a table by a specific variable."""
-        with self.open_session() as session:
-            records: list[TableType] = session.query(table_obj).filter(getattr(table_obj, variable) == value).all()
-        return records
+        return self.session().query(table_obj).filter(getattr(table_obj, variable) == value).all()
 
     def count_records_by_var(self, table_obj: type[TableType], variable: str, value: str) -> int:
         """Count the number of records in a table by a specific variable."""
-        with self.open_session() as session:
-            count: int = session.query(table_obj).filter(getattr(table_obj, variable) == value).count()
-        return count
+        return self.session().query(table_obj).filter(getattr(table_obj, variable) == value).count()
 
     @contextmanager
     def open_session(self) -> Generator[Session, Any]:

bear_utils/extras/responses/function_response.py

@@ -14,13 +14,14 @@ from bear_utils.extras._async_helpers import AsyncResponseModel, create_async_ta
 # Pydantic will yell if we put this into a TYPE_CHECKING block.
 from bear_utils.logger_manager import AsyncLoggerProtocol, LoggerProtocol  # noqa: TC001
 
-SUCCESS: list[str] = ["name", "success"]
-FAILURE: list[str] = ["name"]
-
 if TYPE_CHECKING:
     from collections.abc import Callable
 
 
+SUCCESS: list[str] = ["name", "success", "number_of_tasks"]
+FAILURE: list[str] = ["name", "number_of_tasks"]
+
+
 class FunctionResponse(BaseModel):
     """A class to represent the response of a function call, including success status, content, and error messages."""
 
@@ -337,10 +338,8 @@ class FunctionResponse(BaseModel):
             content = []
         if not isinstance(error, (list | str)):
             error = []
-
         if not content and not error:
             return
-
         res: AsyncResponseModel = create_async_task(
             _log_messages,
             content=content,

{bear_utils-0.8.22.dist-info → bear_utils-0.8.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bear-utils
-Version: 0.8.22
+Version: 0.8.23
 Summary: Various utilities for Bear programmers, including a rich logging utility, a disk cache, and a SQLite database wrapper amongst other things.
 Author-email: chaz <bright.lid5647@fastmail.com>
 Requires-Python: >=3.12
@@ -24,7 +24,7 @@ Provides-Extra: gui
 Requires-Dist: pyqt6>=6.9.0; extra == 'gui'
 Description-Content-Type: text/markdown
 
-# Bear Utils v# Bear Utils v0.8.22
+# Bear Utils v# Bear Utils v0.8.23
 
 Personal set of tools and utilities for Python projects, focusing on modularity and ease of use. This library includes components for caching, database management, logging, time handling, file operations, CLI prompts, image processing, clipboard interaction, gradient utilities, event systems, and async helpers.
 

{bear_utils-0.8.22.dist-info → bear_utils-0.8.23.dist-info}/RECORD

@@ -28,7 +28,7 @@ bear_utils/constants/date_related.py,sha256=vwgPPGamVJ8p12iRFu3IzTksmGfhqYULwrk-
 bear_utils/constants/server.py,sha256=1gWA3h1HH12uk5nqX3r5EFkffPKVDTo3he51teHSDbk,271
 bear_utils/constants/time_related.py,sha256=Mykb27THqC-K0HFrbbrdkTNJUKhzBEOfmPJ0rx2-L6E,705
 bear_utils/database/__init__.py,sha256=dS_HHJKESpsI6BFDLVx055o0GCJV9CMYOQVuHs7EfgY,192
-bear_utils/database/_db_manager.py,sha256=rW4yq6arMF7UbgKrA7C2Wi37O-byqzae39tIZixfRZk,3977
+bear_utils/database/_db_manager.py,sha256=bbg5zG06DcOvd37oafLIqc480BwF9ZW9qWF3ldZHR8o,3805
 bear_utils/events/__init__.py,sha256=EFqmuzhaEYK9kjkGlrM7bjdjPwFEDbKn6RjJKfIBEJY,414
 bear_utils/events/events_class.py,sha256=vPDjWrbut8L3TFn7byyYFZpWYM5ADIqtW2Aeh-qtKfQ,1632
 bear_utils/events/events_module.py,sha256=DkQsDZ5WzM1MH1Msg7mRny40a17Jl31CXbpw4-pICxc,2349
@@ -37,7 +37,7 @@ bear_utils/extras/_async_helpers.py,sha256=4buULZZkR0n2z13Q8_FK59dMchj0Mup03YBaq
 bear_utils/extras/_tools.py,sha256=-rOH_-0pRd7_-o_l2QqByYBB2yhbXcTC8fuQf4Sx-eg,7671
 bear_utils/extras/platform_utils.py,sha256=Ai7ow7S-_cKb5zFwFh8dkC8xmbMJFy-0_-w3NCERdEw,1362
 bear_utils/extras/responses/__init__.py,sha256=XbE4VKemrKRwx9E5jqy__OiM_AAjA58ebnqQ2hytnT0,225
-bear_utils/extras/responses/function_response.py,sha256=VFhYGoSXbqdkgeBWYdmHcHcJ2Z0BFg_Y8EmdkaF442Y,16220
+bear_utils/extras/responses/function_response.py,sha256=aGUhxfliaqE2tXKGtVPDeOy5JJRjHN5M9gRgI1IsopY,16257
 bear_utils/extras/wrappers/__init__.py,sha256=crh4sKOLvuhNMVX5bJYjCFWtXtH7G47UgNPOHq3HXTk,43
 bear_utils/extras/wrappers/add_methods.py,sha256=z2XZG2ZoYOB1MaGiLli4NRyyTeRgBy7tuYsiy8mTa9s,4422
 bear_utils/files/__init__.py,sha256=mIdnFSXoDE64ElM43bN2m6KuafURnN82ki0pdqN8q2o,201
@@ -86,6 +86,6 @@ bear_utils/monitoring/__init__.py,sha256=9DKNIWTp_voLnaWgiP-wJ-o_N0hYixo-MzjUmg8
 bear_utils/monitoring/_common.py,sha256=LYQFxgTP9fk0cH71IQTuGwBYYPWCqHP_mMRNecoD76M,657
 bear_utils/monitoring/host_monitor.py,sha256=e0TYRJw9iDj5Ga6y3ck1TBFEeH42Cax5mQYaNU8yams,13241
 bear_utils/time/__init__.py,sha256=VctjJG17SyEHAFXytI1sZrOrq7zm3hVenIDOJFdaMN0,1424
-bear_utils-0.8.22.dist-info/METADATA,sha256=bI6HOJaPlDs8ml0B_n-if7qmczI67TKOllT-6JkXXhY,8763
-bear_utils-0.8.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bear_utils-0.8.22.dist-info/RECORD,,
+bear_utils-0.8.23.dist-info/METADATA,sha256=paN99eKZqMAyoDD6z25V7EFeHK1IcDaAvRwXBQPjyJ8,8763
+bear_utils-0.8.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bear_utils-0.8.23.dist-info/RECORD,,