arpakitlib 1.8.156__py3-none-any.whl → 1.8.157__py3-none-any.whl

arpakitlib/_arpakit_project_template_v_5/command/remove_rows_from_tables_sqlalchemy_db.py

@@ -0,0 +1,11 @@
+from project.core.util import setup_logging
+from project.sqlalchemy_db_.sqlalchemy_db import get_cached_sqlalchemy_db
+
+
+def __command():
+    setup_logging()
+    get_cached_sqlalchemy_db().remove_rows_from_tables()
+
+
+if __name__ == '__main__':
+    __command()

arpakitlib/ar_sqlalchemy_util.py

@@ -220,19 +220,27 @@ class SQLAlchemyDb:
         self.remove_celery_tables_data()
         self._logger.info("all reinited")
 
-    def remove_rows_from_tables(self):
+    def remove_rows_from_tables(self, exclude_tables: list[str] | None = None):
+        if exclude_tables is None:
+            exclude_tables = []
         with self.new_session() as session:
             for table_name, table in BaseDBM.metadata.tables.items():
-                session.execute(sqlalchemy.delete(table))
+                if table_name not in exclude_tables:
+                    session.execute(sqlalchemy.delete(table))
             session.commit()
-        self._logger.info(f"rows from tables ({BaseDBM.metadata.tables.keys()}) were removed")
+        removed_tables = [t for t in BaseDBM.metadata.tables.keys() if t not in exclude_tables]
+        self._logger.info(f"rows from tables ({removed_tables}) were removed")
 
-    async def async_remove_rows_from_tables(self):
+    async def async_remove_rows_from_tables(self, exclude_tables: list[str] | None = None):
+        if exclude_tables is None:
+            exclude_tables = []
         async with self.new_async_session() as async_session:
             for table_name, table in BaseDBM.metadata.tables.items():
-                await async_session.execute(sqlalchemy.delete(table))
+                if table_name not in exclude_tables:
+                    await async_session.execute(sqlalchemy.delete(table))
             await async_session.commit()
-        self._logger.info(f"rows from tables ({BaseDBM.metadata.tables.keys()}) were removed")
+        removed_tables = [t for t in BaseDBM.metadata.tables.keys() if t not in exclude_tables]
+        self._logger.info(f"rows from tables ({removed_tables}) were removed")
 
     def check_conn(self):
         self.engine.connect()

{arpakitlib-1.8.156.dist-info → arpakitlib-1.8.157.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arpakitlib
-Version: 1.8.156
+Version: 1.8.157
 Summary: arpakitlib
 License: Apache-2.0
 Keywords: arpakitlib,arpakit,arpakit-company,arpakitcompany,arpakit_company

{arpakitlib-1.8.156.dist-info → arpakitlib-1.8.157.dist-info}/RECORD

@@ -53,6 +53,7 @@ arpakitlib/_arpakit_project_template_v_5/command/reinit_all_sqlalchemy_db.py,sha
 arpakitlib/_arpakit_project_template_v_5/command/reinit_json_db.py,sha256=NYY1dPVbJ_mO9aCu8VTiDT7kfloiGvG624hJ_5g2tBM,318
 arpakitlib/_arpakit_project_template_v_5/command/reinit_sqlalchemy_db.py,sha256=aCoLtUvB_dubMHMGlOFgVzxkhhPAL4lHeMU0LM4CBCs,343
 arpakitlib/_arpakit_project_template_v_5/command/remove_operations.py,sha256=l1BMAT9kQMy_aZ2ZNc20vTi1xXktmDrJvKDvo85b8xA,326
+arpakitlib/_arpakit_project_template_v_5/command/remove_rows_from_tables_sqlalchemy_db.py,sha256=9XDeYhjdELLq1T7ArskaMjzqpGerYuP38PjOwUFDnqQ,259
 arpakitlib/_arpakit_project_template_v_5/command/remove_story_logs.py,sha256=l-whsYahKY6ssk-c24ufiybF-U24iFqMdkNlnlASm2U,498
 arpakitlib/_arpakit_project_template_v_5/command/rm_all_records_in_json_db.py,sha256=qF5c_Aso7XpNcBZOInz7MWXX9Wm8L73rcOZBKGXlFfg,326
 arpakitlib/_arpakit_project_template_v_5/command/set_tg_bot_commands.py,sha256=6HbuL6acw-U8vkUdqLVIEffe_FcBsDWRKZ90AakZbv8,456
@@ -402,13 +403,13 @@ arpakitlib/ar_schedule_uust_api_client_util.py,sha256=rXI2_3OTaIBgR-GixM1Ti-Ue1f
 arpakitlib/ar_settings_util.py,sha256=Y5wi_cmsjDjfJpM0VJHjbo0NoVPKfypKaD1USowwDtQ,1327
 arpakitlib/ar_sleep_util.py,sha256=ggaj7ML6QK_ADsHMcyu6GUmUpQ_9B9n-SKYH17h-9lM,1045
 arpakitlib/ar_sqladmin_util.py,sha256=Jd33EX_fAso4bvrcDhNEYOUlUHkudbadNGOJDg7rqhQ,301
-arpakitlib/ar_sqlalchemy_util.py,sha256=J8AAPT0BgrrRYYkJJn_ezEla6YyF3OL34Q-DNpadziU,11285
+arpakitlib/ar_sqlalchemy_util.py,sha256=w_tGPTWIMVjHkTEYo9tVe1sfK_4vvfd3zGATLiYC2J4,11775
 arpakitlib/ar_ssh_runner_util.py,sha256=yvAwza480MkHKvLkDEsR7JNh2bYNs6P9rCVo4NA85NE,6865
 arpakitlib/ar_str_util.py,sha256=2lGpnXDf2h1cBZpVf5i1tX_HCv5iBd6IGnrCw4QWWlY,4350
 arpakitlib/ar_type_util.py,sha256=Cs_tef-Fc5xeyAF54KgISCsP11NHyzIsglm4S3Xx7iM,4049
 arpakitlib/ar_yookassa_api_client_util.py,sha256=VozuZeCJjmLd1zj2BdC9WfiAQ3XYOrIMsdpNK-AUlm0,5347
-arpakitlib-1.8.156.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.156.dist-info/METADATA,sha256=w7-HsemfipWlnRI-TBdYbQY3yqWzPO7mHUBd3iuBZYI,3706
-arpakitlib-1.8.156.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.156.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.156.dist-info/RECORD,,
+arpakitlib-1.8.157.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.157.dist-info/METADATA,sha256=JrajnfsEw_XyxLDxB-5qvd1cjf6LQSaVCmjGtjBahyQ,3706
+arpakitlib-1.8.157.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.157.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.157.dist-info/RECORD,,