initial

olgram/bot/bots.py

@@ -0,0 +1,16 @@
+from aiogram import Bot, Dispatcher, executor, types
+from aiogram.dispatcher import FSMContext
+
+from ..utils.router import Router
+
+router = Router()
+
+
+@router.message_handler(commands=["my_bots"])
+async def my_bots(message: types.Message, state: FSMContext):
+    await message.answer("У тебя много ботов )")
+
+
+@router.message_handler(commands=["add_bot"])
+async def add_bot(message: types.Message, state: FSMContext):
+    await message.answer("Добавь тогда ок")

olgram/bot/start.py

@@ -0,0 +1,17 @@
+from aiogram import Bot, Dispatcher, executor, types
+from aiogram.dispatcher import FSMContext
+
+from ..utils.router import Router
+
+router = Router()
+
+
+@router.message_handler(commands=["start"])
+async def start(message: types.Message, state: FSMContext):
+    """
+    Start command handler
+    """
+
+    await message.answer(
+        "Привет. Я бот Olgram"
+    )

olgram/models/bot.py

@@ -0,0 +1,11 @@
+from tortoise.models import Model
+from tortoise import fields
+
+
+class Bot(Model):
+    id = fields.IntField(pk=True)
+    token = fields.CharField(max_length=50, unique=True)
+    owner = fields.ForeignKeyField("models.User", related_name="bots")
+
+    class Meta:
+        table = 'bot'

olgram/models/user.py

@@ -0,0 +1,9 @@
+from tortoise import Model, fields
+
+
+class User(Model):
+    id = fields.IntField(pk=True)
+    telegram_id = fields.IntField(index=True)
+
+    class Meta:
+        table = 'user'

olgram/utils/database.py

@@ -0,0 +1,15 @@
+from tortoise import Tortoise
+from settings import DatabaseSettings
+
+
+async def init_database():
+    # Here we create a SQLite DB using file "db.sqlite3"
+    #  also specify the app name of "models"
+    #  which contain models from "app.models"
+    await Tortoise.init(
+        db_url=f'postgres://{DatabaseSettings.user()}:{DatabaseSettings.password()}'
+               f'@localhost:5431/{DatabaseSettings.database_name()}',
+        modules={'models': ['olgram.models.bot', 'olgram.models.user']}
+    )
+    # Generate the schema
+    await Tortoise.generate_schemas()

olgram/utils/router.py

@@ -0,0 +1,79 @@
+from dataclasses import dataclass
+from typing import Any, Dict, List, Tuple
+
+from aiogram.dispatcher import Dispatcher
+
+
+@dataclass()
+class Handler:
+    callback: Any
+    custom_filters: Tuple[Any]
+    kwargs: Dict[Any, Any]
+    commands: Any = None
+    regexp: Any = None
+    content_types: Any = None
+    state: Any = None
+    run_task: Any = None
+
+
+class Router:
+    def __init__(self):
+        self._message_handlers: List[Handler] = []
+        self._inline_handlers: List[Handler] = []
+        self._callback_handlers: List[Handler] = []
+
+    def message_handler(
+        self, *custom_filters, commands=None, regexp=None, content_types=None, state=None, run_task=None, **kwargs
+    ):
+        def decorator(callback):
+            self._message_handlers.append(
+                Handler(callback, custom_filters, kwargs, commands, regexp, content_types, state, run_task)
+            )
+            return callback
+
+        return decorator
+
+    def inline_handler(self, *custom_filters, state=None, run_task=None, **kwargs):
+        def decorator(callback):
+            self._inline_handlers.append(Handler(callback, custom_filters, kwargs, state=state, run_task=run_task))
+            return callback
+
+        return decorator
+
+    def callback_query_handler(self, *custom_filters, state=None, run_task=None, **kwargs):
+        def decorator(callback):
+            self._callback_handlers.append(Handler(callback, custom_filters, kwargs, state=state, run_task=run_task))
+            return callback
+
+        return decorator
+
+    def setup(self, dp: Dispatcher):
+        for handler in self._message_handlers:
+            dp.register_message_handler(
+                handler.callback,
+                *handler.custom_filters,
+                commands=handler.commands,
+                regexp=handler.regexp,
+                content_types=handler.content_types,
+                state=handler.state,
+                run_task=handler.run_task,
+                **handler.kwargs
+            )
+
+        for handler in self._inline_handlers:
+            dp.register_inline_handler(
+                handler.callback,
+                *handler.custom_filters,
+                state=handler.state,
+                run_task=handler.run_task,
+                **handler.kwargs
+            )
+
+        for handler in self._callback_handlers:
+            dp.register_callback_query_handler(
+                handler.callback,
+                *handler.custom_filters,
+                state=handler.state,
+                run_task=handler.run_task,
+                **handler.kwargs
+            )