arpakitlib 1.8.28__tar.gz → 1.8.29__tar.gz

{arpakitlib-1.8.28 → arpakitlib-1.8.29}/PKG-INFO

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

{arpakitlib-1.8.28 → arpakitlib-1.8.29}/arpakitlib/ar_sqlalchemy_util.py

@@ -6,6 +6,7 @@ from typing import Any
 from urllib.parse import quote_plus
 from uuid import uuid4
 
+import sqlalchemy
 from sqlalchemy import create_engine, QueuePool, text, func, inspect, AsyncAdaptedQueuePool
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
 from sqlalchemy.orm import sessionmaker, DeclarativeBase
@@ -243,6 +244,19 @@ class SQLAlchemyDb:
     def generate_creation_dt(self) -> datetime:
         return now_utc_dt()
 
+    async def async_get_table_name_to_amount(self) -> dict[str, int]:
+        res = {}
+
+        async with self.new_async_session() as async_session:
+            for table_name, table in BaseDBM.metadata.tables.items():
+                res[table_name] = await async_session.scalar(
+                    sqlalchemy.select(
+                        sqlalchemy.func.count(1)
+                    ).select_from(table)
+                )
+
+        return res
+
 
 def get_string_info_from_declarative_base(class_: type[DeclarativeBase]):
     res = f"Db Models: {len(class_.__subclasses__())}"

{arpakitlib-1.8.28 → arpakitlib-1.8.29}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arpakitlib"
-version = "1.8.28"
+version = "1.8.29"
 description = "arpakitlib"
 authors = [
     { name = "arpakit_support", email = "support@arpakit.com" }