arpakitlib 1.7.256__py3-none-any.whl → 1.8.0__py3-none-any.whl

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_1.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_2.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_3.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_4.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_5.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_6.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/_arpakit_project_template_v1/sandbox/sandbox_7.py

@@ -0,0 +1,14 @@
+import asyncio
+
+
+def __sandbox():
+    pass
+
+
+async def __async_sandbox():
+    pass
+
+
+if __name__ == '__main__':
+    __sandbox()
+    asyncio.run(__async_sandbox())

arpakitlib/{_arpakit_project_template/src/sqlalchemy_db/util.py → _arpakit_project_template_v1/sqlalchemy_db/sqlalchemy_db.py}

@@ -1,16 +1,8 @@
 from functools import lru_cache
 
-from arpakitlib.ar_sqlalchemy_model_util import BaseDBM
 from arpakitlib.ar_sqlalchemy_util import SQLAlchemyDb
-from src.core.settings import get_cached_settings
-
-
-def get_base_dbm() -> type[BaseDBM]:
-    from arpakitlib.ar_sqlalchemy_model_util import import_ar_sqlalchemy_models
-    import_ar_sqlalchemy_models()
-    from src.sqlalchemy_db.sqlalchemy_model import import_project_sqlalchemy_models
-    import_project_sqlalchemy_models()
-    return BaseDBM
+from core.settings import get_cached_settings
+from sqlalchemy_db.sqlalchemy_model import get_simple_dbm
 
 
 def create_sqlalchemy_db() -> SQLAlchemyDb:
@@ -18,7 +10,7 @@ def create_sqlalchemy_db() -> SQLAlchemyDb:
         sync_db_url=get_cached_settings().sqlalchemy_sync_db_url,
         async_db_url=get_cached_settings().sqlalchemy_async_db_url,
         db_echo=get_cached_settings().sqlalchemy_db_echo,
-        base_dbm=get_base_dbm()
+        base_dbm=get_simple_dbm()
     )
 
 

arpakitlib/_arpakit_project_template_v1/sqlalchemy_db/sqlalchemy_model.py

@@ -0,0 +1,129 @@
+from datetime import datetime, timedelta
+from typing import Any
+
+import sqlalchemy
+from sqlalchemy import func
+from sqlalchemy.dialects import postgresql
+from sqlalchemy.orm import mapped_column, Mapped
+
+from arpakitlib.ar_datetime_util import now_utc_dt
+from arpakitlib.ar_enumeration_util import Enumeration
+from arpakitlib.ar_sqlalchemy_util import get_string_info_from_declarative_base, BaseDBM
+from sqlalchemy_db.util import generate_default_long_id
+
+
+class SimpleDBM(BaseDBM):
+    __abstract__ = True
+
+    id: Mapped[int] = mapped_column(
+        sqlalchemy.INTEGER, primary_key=True, autoincrement=True, sort_order=-3, nullable=False
+    )
+    long_id: Mapped[str] = mapped_column(
+        sqlalchemy.TEXT, insert_default=generate_default_long_id, server_default=func.gen_random_uuid(),
+        unique=True, sort_order=-2, nullable=False
+    )
+    creation_dt: Mapped[datetime] = mapped_column(
+        sqlalchemy.TIMESTAMP(timezone=True), insert_default=now_utc_dt, server_default=func.now(),
+        index=True, sort_order=-1, nullable=False
+    )
+
+    def __repr__(self):
+        return f"{self.__class__.__name__.removesuffix('DBM')} (id={self.id})"
+
+
+class StoryLogDBM(SimpleDBM):
+    __tablename__ = "story_log"
+
+    class Levels(Enumeration):
+        info = "info"
+        warning = "warning"
+        error = "error"
+
+    class Types(Enumeration):
+        error_in_execute_operation = "error_in_execute_operation"
+        error_in_api_route = "error_in_api_route"
+
+    level: Mapped[str] = mapped_column(
+        sqlalchemy.TEXT, insert_default=Levels.info, server_default=Levels.info, index=True, nullable=False
+    )
+    type: Mapped[str | None] = mapped_column(sqlalchemy.TEXT, index=True, default=None, nullable=True)
+    title: Mapped[str | None] = mapped_column(sqlalchemy.TEXT, index=True, default=None, nullable=True)
+    data: Mapped[dict[str, Any]] = mapped_column(
+        postgresql.JSON, insert_default={}, server_default="{}", nullable=False
+    )
+
+
+class OperationDBM(SimpleDBM):
+    __tablename__ = "operation"
+
+    class Statuses(Enumeration):
+        waiting_for_execution = "waiting_for_execution"
+        executing = "executing"
+        executed_without_error = "executed_without_error"
+        executed_with_error = "executed_with_error"
+
+    class Types(Enumeration):
+        healthcheck_ = "healthcheck"
+        raise_fake_error_ = "raise_fake_error_"
+
+    status: Mapped[str] = mapped_column(
+        sqlalchemy.TEXT, index=True, insert_default=Statuses.waiting_for_execution,
+        server_default=Statuses.waiting_for_execution, nullable=False
+    )
+    type: Mapped[str] = mapped_column(
+        sqlalchemy.TEXT, index=True, insert_default=Types.healthcheck_, nullable=False
+    )
+    execution_start_dt: Mapped[datetime | None] = mapped_column(sqlalchemy.TIMESTAMP(timezone=True), nullable=True)
+    execution_finish_dt: Mapped[datetime | None] = mapped_column(sqlalchemy.TIMESTAMP(timezone=True), nullable=True)
+    input_data: Mapped[dict[str, Any]] = mapped_column(
+        postgresql.JSON,
+        insert_default={},
+        server_default="{}",
+        nullable=False
+    )
+    output_data: Mapped[dict[str, Any]] = mapped_column(
+        postgresql.JSON, insert_default={}, server_default="{}", nullable=False
+    )
+    error_data: Mapped[dict[str, Any]] = mapped_column(
+        postgresql.JSON, insert_default={}, server_default="{}", nullable=False
+    )
+
+    def raise_if_executed_with_error(self):
+        if self.status == self.Statuses.executed_with_error:
+            raise Exception(
+                f"Operation (id={self.id}, type={self.type}) executed with error, error_data={self.error_data}"
+            )
+
+    def raise_if_error_data(self):
+        if self.error_data:
+            raise Exception(
+                f"Operation (id={self.id}, type={self.type}) has error_data, error_data={self.error_data}"
+            )
+
+    @property
+    def duration(self) -> timedelta | None:
+        if self.execution_start_dt is None or self.execution_finish_dt is None:
+            return None
+        return self.execution_finish_dt - self.execution_start_dt
+
+    @property
+    def duration_total_seconds(self) -> float | None:
+        if self.duration is None:
+            return None
+        return self.duration.total_seconds()
+
+    @property
+    def sdp_duration_total_seconds(self) -> float | None:
+        return self.duration_total_seconds
+
+
+def get_simple_dbm() -> type[SimpleDBM]:
+    return SimpleDBM
+
+
+def import_sqlalchemy_models():
+    pass
+
+
+if __name__ == '__main__':
+    print(get_string_info_from_declarative_base(SimpleDBM))

arpakitlib/_arpakit_project_template_v1/sqlalchemy_db/util.py

@@ -0,0 +1,28 @@
+from uuid import uuid4
+
+from arpakitlib.ar_datetime_util import now_utc_dt
+
+
+def generate_api_key() -> str:
+    return (
+        f"apikey"
+        f"{str(uuid4()).replace('-', '')}"
+        f"{str(now_utc_dt().timestamp()).replace('.', '')}"
+    )
+
+
+def generate_default_long_id():
+    return (
+        f"longid"
+        f"{str(uuid4()).replace('-', '')}"
+        f"{str(now_utc_dt().timestamp()).replace('.', '')}"
+    )
+
+
+def __example():
+    print(generate_api_key())
+    print(generate_default_long_id())
+
+
+if __name__ == '__main__':
+    __example()

arpakitlib/{_arpakit_project_template/src/sandbox/sandbox_3.py → _arpakit_project_template_v1/test_data/make_test_data_1.py}

@@ -1,15 +1,20 @@
 import asyncio
 
 
-# ...
+def make_test_data_1():
+    pass
 
 
-def __example():
+async def async_make_test_data_1():
     pass
 
 
+def __example():
+    make_test_data_1()
+
+
 async def __async_example():
-    pass
+    await async_make_test_data_1()
 
 
 if __name__ == '__main__':

arpakitlib/{_arpakit_project_template/src/sandbox/sandbox_4.py → _arpakit_project_template_v1/test_data/make_test_data_2.py}

@@ -1,15 +1,20 @@
 import asyncio
 
 
-# ...
+def make_test_data_2():
+    pass
 
 
-def __example():
+async def async_make_test_data_2():
     pass
 
 
+def __example():
+    make_test_data_2()
+
+
 async def __async_example():
-    pass
+    await async_make_test_data_2()
 
 
 if __name__ == '__main__':

arpakitlib/{_arpakit_project_template/src/sandbox/sandbox_1.py → _arpakit_project_template_v1/test_data/make_test_data_3.py}

@@ -1,15 +1,20 @@
 import asyncio
 
 
-# ...
+def make_test_data_3():
+    pass
 
 
-def __example():
+async def async_make_test_data_3():
     pass
 
 
+def __example():
+    make_test_data_3()
+
+
 async def __async_example():
-    pass
+    await async_make_test_data_3()
 
 
 if __name__ == '__main__':

arpakitlib/{_arpakit_project_template/src/sandbox/sandbox_2.py → _arpakit_project_template_v1/test_data/make_test_data_4.py}

@@ -1,15 +1,20 @@
 import asyncio
 
 
-# ...
+def make_test_data_4():
+    pass
 
 
-def __example():
+async def async_make_test_data_4():
     pass
 
 
+def __example():
+    make_test_data_4()
+
+
 async def __async_example():
-    pass
+    await async_make_test_data_4()
 
 
 if __name__ == '__main__':

arpakitlib/_arpakit_project_template_v1/test_data/make_test_data_5.py

@@ -0,0 +1,22 @@
+import asyncio
+
+
+def make_test_data_5():
+    pass
+
+
+async def async_make_test_data_5():
+    pass
+
+
+def __example():
+    make_test_data_5()
+
+
+async def __async_example():
+    await async_make_test_data_5()
+
+
+if __name__ == '__main__':
+    __example()
+    asyncio.run(__async_example())

arpakitlib/_arpakit_project_template_v1/tg_bot/blank/blank.py

@@ -0,0 +1,5 @@
+from arpakitlib.ar_blank_util import BaseBlank
+
+
+class TgBotBlank(BaseBlank):
+    pass

arpakitlib/{_arpakit_project_template/src → _arpakit_project_template_v1}/tg_bot/blank/util.py

@@ -1,6 +1,6 @@
 from functools import lru_cache
 
-from src.tg_bot.blank.blank import TgBotBlank
+from tg_bot.blank.blank import TgBotBlank
 
 
 def get_create_tg_bot_blank() -> TgBotBlank:

arpakitlib/{_arpakit_project_template/src → _arpakit_project_template_v1}/tg_bot/const.py

@@ -1,11 +1,14 @@
 from arpakitlib.ar_enumeration_util import Enumeration
 
 
-class TgBotPublicCommands(Enumeration):
+class TgBotCommands(Enumeration):
+    # public
+    healthcheck = "healthcheck"
+    arpakitlib = "arpakitlib"
     start = "start"
+    about = "about"
 
-
-class TgBotPrivateCommands(Enumeration):
+    # private
     init_db = "init_db"
     reinit_db = "reinit_db"
     drop_db = "drop_db"

arpakitlib/_arpakit_project_template_v1/tg_bot/event.py

@@ -0,0 +1,51 @@
+import logging
+from typing import Callable
+
+from aiogram import Dispatcher
+
+from tg_bot.transmitted_tg_data import get_cached_transmitted_tg_bot_data
+
+_logger = logging.getLogger(__name__)
+
+
+async def on_startup():
+    _logger.info("start")
+
+    if get_cached_transmitted_tg_bot_data().media_file_storage_in_dir is not None:
+        get_cached_transmitted_tg_bot_data().media_file_storage_in_dir.init()
+
+    if get_cached_transmitted_tg_bot_data().cache_file_storage_in_dir is not None:
+        get_cached_transmitted_tg_bot_data().cache_file_storage_in_dir.init()
+
+    if get_cached_transmitted_tg_bot_data().dump_file_storage_in_dir is not None:
+        get_cached_transmitted_tg_bot_data().dump_file_storage_in_dir.init()
+
+    if get_cached_transmitted_tg_bot_data().settings.api_init_sqlalchemy_db:
+        get_cached_transmitted_tg_bot_data().sqlalchemy_db.init()
+
+    if get_cached_transmitted_tg_bot_data().settings.api_init_json_db:
+        get_cached_transmitted_tg_bot_data().json_db.init()
+
+    _logger.info("finish")
+
+
+async def on_shutdown(*args, **kwargs):
+    _logger.info("start")
+    _logger.info("finish")
+
+
+def get_tg_bot_startup_events() -> list[Callable]:
+    res = [on_startup]
+    return res
+
+
+def get_tg_bot_shutdown_events() -> list[Callable]:
+    res = [on_shutdown]
+    return res
+
+
+def add_events_to_tg_bot_dispatcher(*, tg_bot_dispatcher: Dispatcher):
+    for tg_bot_event in get_tg_bot_startup_events():
+        tg_bot_dispatcher.startup.register(tg_bot_event)
+    for tg_bot_event in get_tg_bot_shutdown_events():
+        tg_bot_dispatcher.shutdown.register(tg_bot_event)

arpakitlib/{_arpakit_project_template/src → _arpakit_project_template_v1}/tg_bot/filter/not_prod_mode.py

@@ -1,6 +1,6 @@
 from aiogram.filters import Filter
 
-from src.core.settings import get_cached_settings
+from core.settings import get_cached_settings
 
 
 class NotProdModeFilter(Filter):

arpakitlib/{_arpakit_project_template/src → _arpakit_project_template_v1}/tg_bot/filter/prod_mode.py

@@ -1,6 +1,6 @@
 from aiogram.filters import Filter
 
-from src.core.settings import get_cached_settings
+from core.settings import get_cached_settings
 
 
 class ProdModeFilter(Filter):

arpakitlib/_arpakit_project_template_v1/tg_bot/middleware/init_user.py

@@ -0,0 +1,24 @@
+import logging
+from typing import Any, Awaitable, Callable, Dict
+
+from aiogram import BaseMiddleware
+from aiogram.types import TelegramObject
+
+from tg_bot.transmitted_tg_data import TransmittedTgBotData
+
+_logger = logging.getLogger(__name__)
+
+
+class InitUserTgBotMiddleware(BaseMiddleware):
+
+    async def __call__(
+            self,
+            handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+            event: TelegramObject,
+            data: Dict[str, Any],
+    ) -> Any:
+        _logger.info("start")
+        transmitted_tg_bot_data: TransmittedTgBotData = data["transmitted_tg_bot_data"]
+        # ...
+        _logger.info("finish")
+        return await handler(event, data)

arpakitlib/_arpakit_project_template_v1/tg_bot/middleware/middleware.py

@@ -0,0 +1,12 @@
+import aiogram
+
+from tg_bot.middleware.init_user import InitUserTgBotMiddleware
+
+
+def register_tg_bot_middleware(
+        *,
+        tg_bot_dp: aiogram.Dispatcher,
+        **kwargs
+) -> aiogram.Dispatcher:
+    tg_bot_dp.update.outer_middleware.register(InitUserTgBotMiddleware())
+    return tg_bot_dp

arpakitlib/_arpakit_project_template_v1/tg_bot/router/arpakitlib_.py

@@ -0,0 +1,10 @@
+import aiogram.filters
+
+from tg_bot.const import TgBotCommands
+
+tg_bot_router = aiogram.Router()
+
+
+@tg_bot_router.message(aiogram.filters.Command(TgBotCommands.arpakitlib))
+async def _(m: aiogram.types.Message, **kwargs):
+    await m.answer(text="arpakitlib")

arpakitlib/_arpakit_project_template_v1/tg_bot/router/error.py

@@ -0,0 +1,15 @@
+import aiogram
+from aiogram import Router
+
+from tg_bot.transmitted_tg_data import TransmittedTgBotData
+
+tg_bot_router = Router()
+
+
+@tg_bot_router.error()
+async def _(
+        event: aiogram.types.ErrorEvent,
+        transmitted_tg_bot_data: TransmittedTgBotData,
+        **kwargs
+):
+    pass

arpakitlib/_arpakit_project_template_v1/tg_bot/router/healthcheck.py

@@ -0,0 +1,10 @@
+import aiogram.filters
+
+from tg_bot.const import TgBotCommands
+
+tg_bot_router = aiogram.Router()
+
+
+@tg_bot_router.message(aiogram.filters.Command(TgBotCommands.healthcheck))
+async def _(m: aiogram.types.Message, **kwargs):
+    await m.answer(text="healthcheck")

arpakitlib/_arpakit_project_template_v1/tg_bot/router/main_router.py

@@ -0,0 +1,14 @@
+from aiogram import Router
+
+from tg_bot.router import error
+from tg_bot.router import healthcheck
+
+main_tg_bot_router = Router()
+
+# Healthcheck
+
+main_tg_bot_router.include_router(router=healthcheck.tg_bot_router)
+
+# Error
+
+main_tg_bot_router.include_router(router=error.tg_bot_router)

arpakitlib/_arpakit_project_template_v1/tg_bot/start_tg_bot.py

@@ -0,0 +1,34 @@
+from arpakitlib.ar_aiogram_util import start_aiogram_tg_bot_with_webhook
+from core.util import setup_logging
+from tg_bot.event import add_events_to_tg_bot_dispatcher
+from tg_bot.tg_bot import create_tg_bot
+from tg_bot.tg_bot_dispatcher import create_tg_bot_dispatcher
+from tg_bot.transmitted_tg_data import get_cached_transmitted_tg_bot_data
+
+
+def start_tg_bot():
+    setup_logging()
+
+    transmitted_tg_bot_data = get_cached_transmitted_tg_bot_data()
+
+    tg_bot = create_tg_bot()
+
+    tg_bot_dispatcher = create_tg_bot_dispatcher()
+
+    add_events_to_tg_bot_dispatcher(tg_bot_dispatcher=tg_bot_dispatcher)
+
+    if transmitted_tg_bot_data.settings.tg_bot_webhook_enabled:
+        tg_bot_dispatcher.start_polling(tg_bot)
+    else:
+        start_aiogram_tg_bot_with_webhook(
+            dispatcher=tg_bot_dispatcher,
+            bot=tg_bot,
+            webhook_secret=transmitted_tg_bot_data.settings.tg_bot_webhook_secret,
+            webhook_path=transmitted_tg_bot_data.settings.tg_bot_webhook_path,
+            webhook_server_hostname=transmitted_tg_bot_data.settings.tg_bot_webhook_server_hostname,
+            webhook_server_port=transmitted_tg_bot_data.settings.tg_bot_webhook_server_port
+        )
+
+
+if __name__ == '__main__':
+    start_tg_bot()

arpakitlib/_arpakit_project_template_v1/tg_bot/tg_bot.py

@@ -0,0 +1,24 @@
+import aiogram
+from aiogram.client.default import DefaultBotProperties
+from aiogram.client.session.aiohttp import AiohttpSession
+from aiogram.enums import ParseMode
+
+from core.settings import get_cached_settings
+
+
+def create_tg_bot() -> aiogram.Bot:
+    session: AiohttpSession | None = None
+    if get_cached_settings().tg_bot_proxy_url is not None:
+        session = AiohttpSession(proxy=get_cached_settings().tg_bot_proxy_url)
+
+    tg_bot = aiogram.Bot(
+        token=get_cached_settings().tg_bot_token,
+        session=session,
+        default=DefaultBotProperties(
+            parse_mode=ParseMode.HTML,
+            disable_notification=False,
+            link_preview_is_disabled=True
+        )
+    )
+
+    return tg_bot