Initial commit

Author: coder2020official

.gitignore

@@ -0,0 +1 @@
+.DS_STORE

bot.py

@@ -0,0 +1,49 @@
+import asyncio
+
+# filters
+from tgbot.filters.admin_filter import AdminFilter
+
+# handlers
+from tgbot.handlers.admin import admin_user
+from tgbot.handlers.spam_command import anti_spam
+from tgbot.handlers.user import any_user
+
+# middlewares
+from tgbot.middlewares.antiflood_middleware import AntiFloodMiddleware
+
+# states
+from tgbot.states.register_state import Register
+
+# utils
+from tgbot.utils.database import Database
+
+# telebot
+from telebot.async_telebot import AsyncTeleBot
+
+# config
+from tgbot import config
+
+db = Database()
+
+
+bot = AsyncTeleBot(config.TOKEN)
+
+def register_handlers():
+    bot.register_message_handler(admin_user, commands=['start'], admin=True, pass_bot=True)
+    bot.register_message_handler(any_user, commands=['start'], admin=False, pass_bot=True)
+    bot.register_message_handler(anti_spam, commands=['spam'], pass_bot=True)
+
+register_handlers()
+
+# Middlewares
+bot.setup_middleware(AntiFloodMiddleware(limit=2, bot=bot))
+
+
+# custom filters
+bot.add_custom_filter(AdminFilter())
+
+async def run():
+    await bot.polling(non_stop=True)
+
+
+asyncio.run(run())

requirements.txt

@@ -0,0 +1,2 @@
+aiogram
+aioredis

systemd/tgbot.service

@@ -0,0 +1,12 @@
+[Unit]
+After=network.target
+Description=Template Bot
+
+[Service]
+Type=simple
+WorkingDirectory=/root/folder/
+ExecStart=/usr/bin/python3 bot.py
+Restart=always
+
+[Install]
+WantedBy=multi-user.target

tgbot/config.py

@@ -0,0 +1,3 @@
+# any configuration should be stored here
+
+TOKEN = 'token' # configure env if you need;

tgbot/filters/__init__.py

@@ -0,0 +1 @@
+# register filters here or in different folders.

tgbot/filters/admin_filter.py

@@ -0,0 +1,13 @@
+from telebot.asyncio_filters import SimpleCustomFilter
+from tgbot.models.users_model import Admin
+
+
+class AdminFilter(SimpleCustomFilter):
+    """
+    Filter for admin users
+    """
+
+    key = 'admin'
+    async def check(self, message):
+
+        return int(message.chat.id) == int(Admin.ADMIN.value)

tgbot/handlers/__init__.py

@@ -0,0 +1 @@
+# Create files for handlers in this folder.

tgbot/handlers/admin.py

@@ -0,0 +1,8 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message
+
+async def admin_user(message: Message, bot: AsyncTeleBot):
+    """
+    You can create a function and use parameter pass_bot.
+    """
+    await bot.send_message(message.chat.id, "Hello, admin!")

tgbot/handlers/spam_command.py

@@ -0,0 +1,16 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message
+
+
+async def anti_spam(message: Message, bot: AsyncTeleBot):
+    """
+    You can create a function and use parameter pass_bot.
+    """
+
+    await bot.send_message(
+        message.chat.id,
+        """This is demo spam command.
+If you send this command more than once within 2 seconds, 
+bot will warn you.
+This is made by using middlewares."""
+)

tgbot/handlers/user.py

@@ -0,0 +1,8 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message
+
+async def any_user(message: Message, bot: AsyncTeleBot):
+    """
+    You can create a function and use parameter pass_bot.
+    """
+    await bot.send_message(message.chat.id, "Hello, user!")

tgbot/middlewares/__init__.py

@@ -0,0 +1 @@
+# Create files for your middlewares in this folder.

tgbot/middlewares/antiflood_middleware.py

@@ -0,0 +1,28 @@
+from telebot.asyncio_handler_backends import BaseMiddleware
+from telebot.async_telebot import CancelUpdate
+
+class AntiFloodMiddleware(BaseMiddleware):
+    def __init__(self, limit, bot) -> None:
+        self.last_time = {}
+        self.limit = limit
+        self.update_types = ['message']
+        self.bot = bot
+        # Always specify update types, otherwise middlewares won't work
+
+
+    async def pre_process(self, message, data):
+        if message.text != '/spam': return # make it work only for this command
+        if not message.from_user.id in self.last_time:
+            # User is not in a dict, so lets add and cancel this function
+            self.last_time[message.from_user.id] = message.date
+            return
+        if message.date - self.last_time[message.from_user.id] < self.limit:
+            # User is flooding
+            await self.bot.send_message(message.chat.id, 'You are making request too often')
+            return CancelUpdate()
+        # write the time of the last request
+        self.last_time[message.from_user.id] = message.date
+
+        
+    async def post_process(self, message, data, exception):
+        pass

tgbot/models/__init__.py

@@ -0,0 +1 @@
+# Register models in this folder.

tgbot/models/users_model.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+# Admin role
+class Admin(Enum):
+    ADMIN = 927266710
+
+
+

tgbot/states/__init__.py

@@ -0,0 +1 @@
+# register states in this folder.

tgbot/states/register_state.py

@@ -0,0 +1,12 @@
+# Create your states in this folder.
+
+
+from telebot.asyncio_handler_backends import State, StatesGroup
+
+
+class Register(StatesGroup):
+    """
+    Group of states for registering
+    """
+    name = State()
+    surname = State()

tgbot/utils/database.py

@@ -0,0 +1,9 @@
+# aioredis
+import aioredis
+
+# Create a connection
+class Database:
+    def __init__(self) -> None:
+        self.redis = aioredis.from_url("redis://localhost")
+    # write other methods if needed.
+