n0body
3692c0233d
Changes: Frontend: - Added and adjusted command handlers for /start, /cat, /subscribe, /unsubscribe, and /subscription_modify - Implemented mass send function to send images for all subscribed users - Configured scheduler to call function above at 12:00 every day - Added psycopg2 error exception for each handler Backend: - Changed PostgreSQL DB Users table structure(id, chat_id, images_amount) and adjusted all functions' arguments and executes to match these changes - Added a function to fetch selected images_amount for a set chat_id - Slightly optimized GetObjectExtension function - Allowed all functions that work with currentDay to accept int values - Added a script for batch population of MinIO bucket with images from ./files/<day number>/ directory Src: - Updated requirements.txt
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
import psycopg2
|
|
import Backend.ISwork, Backend.DBwork
|
|
from src import config
|
|
from src.Backend import DBwork
|
|
from src.Backend import ISwork
|
|
import logging
|
|
from aiogram import Bot, Dispatcher, Router
|
|
from aiogram.filters import Command, CommandObject
|
|
from aiogram.types import Message, URLInputFile
|
|
from aiogram.client.default import DefaultBotProperties
|
|
from aiogram.enums import ParseMode
|
|
from aiogram.fsm.storage.memory import MemoryStorage
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from datetime import datetime
|
|
|
|
|
|
current_day = datetime.now().weekday()
|
|
|
|
start_router = Router()
|
|
scheduler = AsyncIOScheduler(timezone = 'Europe/Moscow')
|
|
|
|
|
|
logging.basicConfig(level = logging.INFO, format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
logger = logging.getLogger(__name__)
|
|
|
|
bot = Bot(token = config.TG_token , default = DefaultBotProperties(parse_mode = ParseMode.HTML))
|
|
dp = Dispatcher(storage = MemoryStorage())
|
|
|
|
|
|
@dp.message(Command('start'))
|
|
async def cmd_start(message: Message):
|
|
await message.answer('''
|
|
This is a bot that sends images of cats.
|
|
|
|
List of available commands:
|
|
/cat - request an image of a cat from today's pool of images
|
|
/subscribe - subscribe to daily cat images sent at 12:00 UTC+3
|
|
/subscription_modify <number> - change the amount of images sent daily
|
|
/unsubscribe - cancel your subscription(but why would you want to? :3)
|
|
''')
|
|
|
|
|
|
@dp.message(Command('cat'))
|
|
async def cmd_cat(message: Message):
|
|
image_link = URLInputFile(ISwork.getDownloadURL(current_day), filename=datetime.now().strftime('%Y_%m_%d_%H_%M_%S'))
|
|
await message.answer_photo(image_link, caption='Look, a cat :3')
|
|
|
|
|
|
@dp.message(Command('subscription_modify'))
|
|
async def subscription_modify(message: Message, command: CommandObject):
|
|
if command.args is None or command.args.isdigit() == False:
|
|
await message.answer('Please write the number of images you would like\n'
|
|
'to receive in the same message as a command.\n'
|
|
'Example:'
|
|
'/subscription_modify <number of daily images>')
|
|
return
|
|
try:
|
|
cursor, connection = DBwork.set_connection()
|
|
chat_id = message.chat.id
|
|
amount = command.args
|
|
DBwork.change_images_amount(chat_id, amount, connection, cursor)
|
|
DBwork.close_connection(connection, cursor)
|
|
except psycopg2.Error:
|
|
await message.answer('You are not yet subscribed.')
|
|
return
|
|
await message.answer('Amount of daily images was changed successfully!')
|
|
|
|
|
|
@dp.message(Command('subscribe'))
|
|
async def cmd_subscribe(message: Message):
|
|
try:
|
|
cursor, connection = DBwork.set_connection()
|
|
chat_id = message.chat.id
|
|
DBwork.add_user(chat_id, connection, cursor)
|
|
DBwork.close_connection(connection, cursor)
|
|
except psycopg2.Error as e:
|
|
await message.answer('You are already subscribed.')
|
|
print(e.pgerror)
|
|
return
|
|
await message.answer('''
|
|
You have successfully subscribed to daily cat photos!
|
|
You will get 1 photo a day by default,
|
|
use /subscription_modify to change that amount.
|
|
''')
|
|
|
|
|
|
@dp.message(Command('unsubscribe'))
|
|
async def cmd_unsubscribe(message: Message):
|
|
try:
|
|
cursor, connection = DBwork.set_connection()
|
|
chat_id = message.chat.id
|
|
DBwork.delete_user(chat_id, connection, cursor)
|
|
DBwork.close_connection(connection, cursor)
|
|
except psycopg2.Error:
|
|
await message.answer('You are not yet subscribed.')
|
|
return
|
|
await message.answer('You have successfully unsubscribed.')
|
|
|
|
|
|
async def send_daily_images():
|
|
cursor, connection = DBwork.set_connection()
|
|
max_id = DBwork.get_last_id(cursor)
|
|
for id in range(1, max_id + 1):
|
|
chat_id = DBwork.get_chat_id(id, cursor)
|
|
images_amount = DBwork.get_images_amount(chat_id, connection, cursor)
|
|
for _ in range(images_amount):
|
|
image_link = URLInputFile(ISwork.getDownloadURL(current_day), filename=datetime.now().strftime('%Y_%m_%d_%H_%M_%S'))
|
|
await bot.send_photo(chat_id = chat_id, photo = image_link)
|
|
DBwork.close_connection(connection, cursor)
|
|
|
|
|
|
scheduler.add_job(send_daily_images, 'cron', hour = 12, minute = 0)
|
|
|
|
|
|
async def main():
|
|
scheduler.start()
|
|
await dp.start_polling(bot)
|