sqlite-database 0.7.5__tar.gz → 0.7.6__tar.gz

{sqlite_database-0.7.5/sqlite_database.egg-info → sqlite_database-0.7.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlite_database
-Version: 0.7.5
+Version: 0.7.6
 Summary: A weird wrapper for SQLite Connection
 Home-page: https://github.com/RimuEirnarn/sqlite_database
 Author: RimuEirnarn

{sqlite_database-0.7.5 → sqlite_database-0.7.6}/sqlite_database/__init__.py

@@ -2,7 +2,7 @@
 
 from .models import BaseModel, model, Foreign, Primary, Unique, hook, validate
 from .database import Database
-from ._utils import null, Row, Null
+from ._utils import Row, Null
 from .column import Column, text, integer, blob, real
 from .signature import op
 from .operators import this
@@ -14,14 +14,13 @@ def test_installed():
     return True
 
 
-__version__ = "0.7.5"
+__version__ = "0.7.6"
 __all__ = [
     "Database",
     "Table",
     "this",
     "op",
     "Column",
-    "null",
     "Null",
     "Row",
     "text",

{sqlite_database-0.7.5 → sqlite_database-0.7.6}/sqlite_database/models/query_builder.py

@@ -1,14 +1,17 @@
 """Model QueryBuilder"""
 
+# pylint: disable=protected-access
+
 from __future__ import annotations
 from typing import Any, Generic, Type, TypeVar
 from .errors import NoDataReturnedError
 from ..column import check_one
 from ..functions import Function
-from .. import models # pylint: disable=unused-import
+from .. import models  # pylint: disable=unused-import
 
 T = TypeVar("T", bound="models.BaseModel")
-count = Function('COUNT')
+count = Function("COUNT")
+
 
 class QueryBuilder(Generic[T]):
     # pylint: disable=protected-access
@@ -27,7 +30,7 @@ class QueryBuilder(Generic[T]):
         self._failing = True
         return self
 
-    def where(self, **kwargs):
+    def where(self, **kwargs: Any):
         """Sets conditioning"""
         self._filters.update(kwargs)
         return self
@@ -50,26 +53,51 @@ class QueryBuilder(Generic[T]):
     def fetch(self) -> list[T]:
         """Fetch data from table"""
         records = self._model._tbl.select(  # pylint: disable=protected-access
-                self._filters, limit=self._limit, offset=self._offset, order=self._order
-            )
+            self._filters, limit=self._limit, offset=self._offset, order=self._order
+        )
         if len(records) == 0 and self._failing:
-            raise NoDataReturnedError(f"Model {self._model.__name__} has no data for current scope")
-        return [
-            self._model(**record) for record in records
-        ]
+            raise NoDataReturnedError(
+                f"Model {self._model.__name__} has no data for current scope"
+            )
+        return [self._model(**record) for record in records]
 
     def fetch_one(self) -> T | None:
         """Fetch one data from table"""
         # pylint: disable=protected-access
-        record = self._model._tbl.select_one(
-            self._filters, order=self._order
-        )
+        record = self._model._tbl.select_one(self._filters, order=self._order)
         if not record and self._failing:
-            raise NoDataReturnedError(f"Model {self._model.__name__} has no data for current scope")
+            raise NoDataReturnedError(
+                f"Model {self._model.__name__} has no data for current scope"
+            )
         if not record:
             return None
         return self._model(**record)
 
+    def patch(self, **kwargs: Any):
+        """Update a data based on the filter according to passed keyword args"""
+        affected_rows = self._model._tbl.update(
+            self._filters, kwargs, self._limit, self._order
+        )
+
+        if affected_rows == 0 and self._failing:
+            raise NoDataReturnedError(
+                f"Model {self._model.__name__} updates no data for current scope"
+            )
+        return affected_rows
+
+    def delete(self):
+        """Delete data based on the filter"""
+        affected_rows = self._model._tbl.delete(self._filters, self._limit, self._order)
+
+        if affected_rows == 0 and self._failing:
+            raise NoDataReturnedError(
+                f"Model {self._model.__name__} delete exactly no rows"
+            )
+
+        return affected_rows
+
     def count(self) -> int:
         """Count how much data is within this operation"""
-        return self._model._tbl.select(self._filters, what=count('*'))
+        return self._model._tbl.select(self._filters, what=count("*"))
+
+    update = patch

{sqlite_database-0.7.5 → sqlite_database-0.7.6/sqlite_database.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlite_database
-Version: 0.7.5
+Version: 0.7.6
 Summary: A weird wrapper for SQLite Connection
 Home-page: https://github.com/RimuEirnarn/sqlite_database
 Author: RimuEirnarn

{sqlite_database-0.7.5 → sqlite_database-0.7.6}/tests/database/model_api/test_model_api.py

@@ -9,6 +9,9 @@ from sqlite_database.models.errors import ValidationError, NoDataReturnedError
 
 from ..setup import setup_model_api
 
+def auto_id():
+    return str(uuid4())
+
 def test_model_api():
     # pylint: disable=protected-access
     """Test 1100 Model API"""
@@ -147,9 +150,6 @@ def test_model_auto_id():
 
     db = Database(":memory:")
 
-    def auto_id():
-        return str(uuid4())
-
     @model(db)
     class Users(BaseModel):
         """Base User class"""
@@ -172,9 +172,6 @@ def test_model_hidden():
 
     db = Database(":memory:")
 
-    def auto_id():
-        return str(uuid4())
-
     @model(db)
     class Users(BaseModel):
         """Base User class"""
@@ -196,9 +193,6 @@ def test_model_fail():
 
     db = Database(":memory:")
 
-    def auto_id():
-        return str(uuid4())
-
     @model(db)
     class Users(BaseModel):
         """Base User class"""
@@ -219,9 +213,6 @@ def test_model_api_runtime_typechecking():
 
     db = Database(":memory:")
 
-    def auto_id():
-        return str(uuid4())
-
     @model(db, True)
     class Users(BaseModel):
         """Base User class"""
@@ -237,3 +228,22 @@ def test_model_api_runtime_typechecking():
         Users.create(id=0, username=0, password=0)
 
     assert Users.first() is None, "How?"
+
+def test_model_api_query_builder_update():
+    """Test Model API Querybuilder update()"""
+
+    db = Database(":memory:")
+
+    Users, _ = setup_model_api(db)
+    admin = Users.create(id=auto_id(), username="admin", is_active=True)
+    assert Users.where(id=admin.id).throw().patch(username="system") == 1, "Changes should be 1"
+
+def test_model_api_query_builder_delete():
+    """Test Model API QueryBuilder delete()"""
+
+    db = Database(":memory:")
+
+    Users, _ = setup_model_api(db)
+    admin = Users.create(id=auto_id(), username='admin', is_active=True)
+
+    assert Users.where(id=admin.id).throw().delete() == 1, "Changes should be 1"