python-plugins 0.1.3__tar.gz → 0.1.4__tar.gz

{python_plugins-0.1.3 → python_plugins-0.1.4}/CHANGES.rst

@@ -1,3 +1,11 @@
+v0.1.4
+------
+
+Released 2024-09-26
+
+- remove_pycache
+- models.update
+
 v0.1.3
 ------
 

{python_plugins-0.1.3 → python_plugins-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: python-plugins
-Version: 0.1.3
+Version: 0.1.4
 Summary: A collection of Python functions and classes.
 Project-URL: Documentation, https://python-plugins.readthedocs.io
 Project-URL: Source, https://github.com/ojso/python-plugins

{python_plugins-0.1.3 → python_plugins-0.1.4}/docs/usage.rst

@@ -50,3 +50,12 @@ mixins
 
    class User(PrimaryKeyMixin,UserMixin, TimestampMixin,db.models):
       pass
+
+remove_pycache
+=======================
+
+.. code-block :: python
+
+    from  python_plugins.utils.remove_pycache import remove_pycache
+
+    remove_pycache(".")

python_plugins-0.1.4/src/python_plugins/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.1.4"

python_plugins-0.1.4/src/python_plugins/models/update.py

@@ -0,0 +1,33 @@
+def update_obj(db, old_obj, new_data: dict, NewClass, force=None):
+    """insert or update object
+
+    :param db: sqlalchemy db
+    :param old_obj: old object
+    :param new_data: new dict
+    :param NewClass: object class
+    :param force: update= update old object with new attribute
+    """
+    # 1. old is not exist
+    if old_obj is None:
+        new_obj = NewClass()
+        for k in new_data:
+            if hasattr(new_obj, k):
+                setattr(new_obj, k, new_data.get(k))
+        db.session.add(new_obj)
+        db.session.commit()
+        print(f"{new_obj} inserted")
+        return
+
+    # 2. old is exist
+    match force:
+        case None:
+            print(f"{old_obj} exists")
+        case "update":
+            # update old
+            for k in new_data:
+                if hasattr(old_obj, k):
+                    setattr(old_obj, k, new_data.get(k))
+            db.session.commit()
+            print(f"{old_obj} updated")
+        case _:
+            raise Exception

python_plugins-0.1.4/src/python_plugins/utils/remove_pycache.py

@@ -0,0 +1,13 @@
+import os
+import shutil
+
+def remove_pycache(dir_path):
+    for root, dirs, files in os.walk(dir_path):
+        if "venv" in root or "git" in root:
+            continue
+        for dir in dirs:
+            if dir == "__pycache__":
+                pycache_path = os.path.join(root, dir)
+                print(f"Removing {pycache_path}")
+                shutil.rmtree(pycache_path)
+

{python_plugins-0.1.3 → python_plugins-0.1.4}/tests/test_sqlalchemy.py

@@ -2,6 +2,7 @@ from sqlalchemy.orm import Mapped
 from sqlalchemy.orm import mapped_column
 from python_plugins.models.mixins import PrimaryKeyMixin
 from python_plugins.models.mixins import DataMixin
+from python_plugins.models.mixins import TimestampMixin
 from sqlalchemy.orm import DeclarativeBase
 from sqlalchemy.schema import CreateTable
 
@@ -12,14 +13,17 @@ class Base(DeclarativeBase):
     pass
 
 
-class Demo(PrimaryKeyMixin, DataMixin, Base):
+class Demo(PrimaryKeyMixin, DataMixin, TimestampMixin,Base):
     __tablename__ = "demo"
 
 
-def test_info():
+def test_createtable():
 
     assert Demo.__table__.name == "demo"
     stmt = CreateTable(Demo.__table__)
     # print(stmt)
     assert "id INTEGER" in str(stmt)
+    assert "PRIMARY KEY" in str(stmt)
     assert "data JSON" in str(stmt)
+    assert "created_at DATETIME" in str(stmt)
+    assert "updated_at DATETIME" in str(stmt)

python_plugins-0.1.4/tests/test_utils.py

@@ -0,0 +1,8 @@
+import pytest
+
+from python_plugins.utils.remove_pycache import remove_pycache
+
+@pytest.mark.skip
+def test_remove_pycache():
+    remove_pycache("./tests")
+

python_plugins-0.1.3/src/python_plugins/__about__.py

@@ -1 +0,0 @@
-__version__ = "0.1.3"