Initial commit

This commit is contained in:
2025-09-15 17:45:55 +03:00
commit c14683a35c
13 changed files with 543 additions and 0 deletions
+175
View File
@@ -0,0 +1,175 @@
# ---> Python
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class
# C extensions
*.so
# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST
# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec
# Installer logs
pip-log.txt
pip-delete-this-directory.txt
# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
cover/
# Translations
*.mo
*.pot
# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal
# Flask stuff:
instance/
.webassets-cache
# Scrapy stuff:
.scrapy
# Sphinx documentation
docs/_build/
# PyBuilder
.pybuilder/
target/
# Jupyter Notebook
.ipynb_checkpoints
# IPython
profile_default/
ipython_config.py
# pyenv
# For a library or package, you might want to ignore these files since the code is
# intended to run in multiple environments; otherwise, check them in:
# .python-version
# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock
# UV
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
#uv.lock
# poetry
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
#poetry.lock
# pdm
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
#pdm.lock
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
# in version control.
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
.pdm.toml
.pdm-python
.pdm-build/
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
__pypackages__/
# Celery stuff
celerybeat-schedule
celerybeat.pid
# SageMath parsed files
*.sage.py
# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/
# Spyder project settings
.spyderproject
.spyproject
# Rope project settings
.ropeproject
# mkdocs documentation
/site
# mypy
.mypy_cache/
.dmypy.json
dmypy.json
# Pyre type checker
.pyre/
# pytype static type analyzer
.pytype/
# Cython debug symbols
cython_debug/
# PyCharm
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/
# Ruff stuff:
.ruff_cache/
# PyPI configuration file
.pypirc
+15
View File
@@ -0,0 +1,15 @@
Project template for simple telegram bots
Features:
- Base commands with FSM (no redis)
- JSON data storage
- Async worker function
- Cute project structure
.env example:
```.env
botToken="token:from-botfather"
whitelist=["userid"]
dataPath="data/"
jsonFilename="data.json"
```
+26
View File
@@ -0,0 +1,26 @@
aiofiles==24.1.0
aiogram==3.22.0
aiohappyeyeballs==2.6.1
aiohttp==3.12.15
aiosignal==1.4.0
annotated-types==0.7.0
anyio==4.10.0
attrs==25.3.0
certifi==2025.8.3
frozenlist==1.7.0
h11==0.16.0
httpcore==1.0.9
httpx==0.28.1
idna==3.10
loguru==0.7.3
magic-filter==1.0.12
multidict==6.6.4
propcache==0.3.2
pydantic==2.11.7
pydantic-settings==2.10.1
pydantic_core==2.33.2
python-dotenv==1.1.1
sniffio==1.3.1
typing-inspection==0.4.1
typing_extensions==4.15.0
yarl==1.20.1
+9
View File
@@ -0,0 +1,9 @@
import asyncio
from app.jsonData import start as start_json
from app.telegram import start_bot
def createApp():
start_json()
asyncio.run(start_bot())
+8
View File
@@ -0,0 +1,8 @@
import asyncio
from loguru import logger
async def internal():
logger.info("Internal")
await asyncio.sleep(4)
+11
View File
@@ -0,0 +1,11 @@
from .connect import load, start
from .utils import add, check, removeById, removeByParam
__all__ = [
"load",
"start",
"add",
"check",
"removeById",
"removeByParam",
]
+47
View File
@@ -0,0 +1,47 @@
import json
import os
import shutil
import sys
from datetime import datetime
from loguru import logger
from app.settings import settings
dataPath = settings.dataPath.removesuffix("/")
jsonRawFilename = settings.jsonFilename.removesuffix(".json")
backupPath = dataPath + "/backup"
filePath = dataPath + "/" + jsonRawFilename + ".json"
def start():
# Inits JSON file
try:
if not(os.path.exists(dataPath)):
os.mkdir(dataPath)
if not(os.path.exists(backupPath)):
os.mkdir(backupPath)
if os.path.exists(filePath):
timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
backupFilePath = backupPath + "/" + jsonRawFilename + f"_backup_{timestamp}.json"
shutil.copy(filePath, backupFilePath)
logger.info(f"JSON backup was created: {backupFilePath}")
with open(filePath, "w") as f:
json.dump({}, f, ensure_ascii = False, indent=4)
logger.success(f"Successfully initialized JSON with path: {filePath}")
except Exception as e:
logger.error(f"Failed to initialize JSON: {e}")
sys.exit(1)
def load():
# Returnes the contents of the JSON file as a dictionary
data = {}
try:
with open(filePath, "r") as f:
data = json.load(f)
except json.decoder.JSONDecodeError:
logger.warning("JSON file is empty!")
return data
+69
View File
@@ -0,0 +1,69 @@
import json
from loguru import logger
from app.jsonData.connect import load
from app.settings import settings
def getParamId(param):
# Returnes -1 if param not found and id if found
currentData = load()
id = -1
for (i, j) in [(i[0], i[1]) for i in currentData]:
if j == param:
id = i
break
return id
def check(param):
# Returnes 1 if param exists and 0 if not
currentData = load()
if param in [i[1] for i in currentData]:
return 1
else:
return 0
def removeById(id: int):
# Returnes 0 if deleted successfully and -1 if not
currentData = load()
if id in currentData:
del currentData[id]
with open(settings.jsonFilename, 'w', encoding = 'utf-8') as f:
json.dump(currentData, f, ensure_ascii = False, indent = 4)
logger.info(f'Id {id} was deleted successfully!')
return 0
else:
logger.info(f'Id {id} was not found in the data file when trying to delete it.')
return -1
def removeByParam(param):
# Returnes 0 if deleted successfully and -1 if not
id = getParamId(param)
if id != -1:
return removeById(id)
else:
logger.info(f'Param {param} was not found in the data file when trying to delete it.')
return -1
def add(param):
# Saves or updates data in JSON
currentData = load()
if currentData == {}:
id = 0
else:
id = currentData[-1][0] + 1
newData = {id: param}
with open(settings.jsonFilename, 'w', encoding = 'utf-8') as f:
currentData.update(newData)
json.dump(currentData, f, ensure_ascii = False, indent = 4)
logger.info(f"Param {param} was added with id {id}")
return id
+14
View File
@@ -0,0 +1,14 @@
from pydantic_settings import BaseSettings, SettingsConfigDict
class Settings(BaseSettings):
botToken: str
whitelist: list
dataPath: str
jsonFilename: str
model_config = SettingsConfigDict(env_file = ".env", env_file_encoding="utf-8")
settings = Settings() # type: ignore
+3
View File
@@ -0,0 +1,3 @@
from .telegram import start_bot
__all__ = ["start_bot"]
+32
View File
@@ -0,0 +1,32 @@
# Bot Statuses
startBot = "Bot started"
stopBot = "Bot stopped"
unexpectedError = "Unexpected error"
# Commands
startCommand = "Hi"
helpCommand = "This is help"
# Route status
askParam = "Enter param:"
successAdd = "Added successfully"
successRemove = "Removed successfully"
alreadyExists = "Key already exists"
# Data status
noData = "No data"
foundData = "Founde data:"
+131
View File
@@ -0,0 +1,131 @@
import asyncio
from aiogram import Bot, Dispatcher, F, Router
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode
from aiogram.filters import Command
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.fsm.storage.memory import MemoryStorage
from aiogram.types import BotCommand, BotCommandScopeDefault, Message
from loguru import logger
from app import jsonData
from app.internal.internal import internal
from app.settings import settings
from app.telegram import strings
async def setCommands():
commands = [
BotCommand(command='start', description='Start'),
BotCommand(command='help', description='Help'),
BotCommand(command='info', description='Info'),
BotCommand(command='add', description='Add'),
BotCommand(command='remove', description='Remove')
]
await bot.set_my_commands(commands, BotCommandScopeDefault())
class addForm(StatesGroup):
param = State()
class removeForm(StatesGroup):
param = State()
dp = Dispatcher(storage=MemoryStorage())
bot = Bot(
token=settings.botToken,
default=DefaultBotProperties(parse_mode=ParseMode.HTML),
)
strategyRouter = Router()
removeRouter = Router()
@dp.message(Command("start"))
async def commandStart(message: Message) -> None:
await message.answer(strings.startCommand)
@dp.message(Command("help"), F.chat.id.in_(settings.whitelist))
async def commandHelp(message: Message) -> None:
await message.answer(strings.helpCommand)
@dp.message(Command("info"), F.chat.id.in_(settings.whitelist))
async def commandInfo(message: Message) -> None:
data = jsonData.load()
msgText = ''
if data == {}:
msgText = strings.noData
else:
msgText = strings.foundData
for i in data:
msgText += (f"<b>{str(i[0])}</b>: </b>{str(i[1])}</b>\n")
await message.answer(msgText)
@strategyRouter.message(Command("add"), F.chat.id.in_(settings.whitelist))
async def commandAdd(message: Message, state: FSMContext):
await message.answer(strings.askParam)
await state.set_state(addForm.param)
@strategyRouter.message(F.text, addForm.param)
async def captureStartPair(message: Message, state: FSMContext):
await state.update_data(pair=message.text)
data = await state.get_data()
param = data.get("param")
t = jsonData.add(param)
msgText = strings.successAdd + str(t)
await asyncio.to_thread(internal)
await message.answer(msgText)
await state.clear()
@removeRouter.message(Command("remove"), F.chat.id.in_(settings.whitelist))
async def commandRemove(message: Message, state: FSMContext):
await message.answer(strings.askParam)
await state.set_state(removeForm.param)
@removeRouter.message(F.text, removeForm.param)
async def captureRemoveParam(message: Message, state: FSMContext):
await state.update_data(pair=message.text)
data = await state.get_data()
param = data.get("param")
t = jsonData.removeByParam(param)
if t == 0:
msgText = strings.successRemove
else:
msgText = strings.noData
await message.answer(msgText)
await state.clear()
async def startBot():
await setCommands()
try:
for i in settings.whitelist:
await bot.send_message(chat_id=i, text=strings.startBot)
except Exception as e:
logger.error(e)
logger.success("Started bot")
async def stopBot():
try:
for i in settings.whitelist:
await bot.send_message(chat_id=i, text=strings.stopBot)
except Exception as e:
logger.error(e)
logger.success("Stopped bot")
async def start_bot() -> None:
dp.startup.register(startBot)
dp.include_router(strategyRouter)
dp.include_router(removeRouter)
dp.shutdown.register(stopBot)
await dp.start_polling(bot)
+3
View File
@@ -0,0 +1,3 @@
from app.createApp import createApp
createApp()