56 lines
3.4 KiB
Python
56 lines
3.4 KiB
Python
from os import path, sep
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from datetime import datetime, timedelta
|
|
from app import app
|
|
from pyrogram.enums.chat_members_filter import ChatMembersFilter
|
|
from modules.utils import configGet, locale, logWrite
|
|
from dateutil.relativedelta import relativedelta
|
|
from modules.database import col_applications
|
|
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
# for user_file in listdir(f"{configGet('data', 'locations')}{sep}users{sep}"):
|
|
# filename = fsdecode(f"{configGet('data', 'locations')}{sep}users{sep}{user_file}")
|
|
# if filename.endswith(".json"):
|
|
# user = jsonLoad(filename)
|
|
# if isinstance(user["application"]["2"], str):
|
|
# try:
|
|
# if ".".join([((user["application"]["2"]).split("."))[0], ((user["application"]["2"]).split("."))[1]]) == datetime.now().strftime("%d.%m"):
|
|
# tg_user = await app.get_users(int(user_file.replace(".json", "")))
|
|
# await app.send_message( configGet("admin_group"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), datetime.strptime(user["application"]["2"], '%d.%m.%Y')).years)) )
|
|
# except AttributeError:
|
|
# continue
|
|
|
|
if configGet("enabled", "scheduler", "cache_avatars"):
|
|
@scheduler.scheduled_job(trigger="date", run_date=datetime.now()+timedelta(seconds=10))
|
|
@scheduler.scheduled_job(trigger="interval", hours=configGet("interval", "scheduler", "cache_avatars"))
|
|
async def cache_avatars():
|
|
list_of_users = []
|
|
async for member in app.get_chat_members(configGet("destination_group"), filter=ChatMembersFilter.SEARCH, query=""):
|
|
list_of_users.append(member.user)
|
|
for user in list_of_users:
|
|
if user.photo != None:
|
|
if not path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{user.photo.big_file_id}'):
|
|
print(f'Pre-cached avatar {user.photo.big_file_id} of {user.id}', flush=True)
|
|
await app.download_media(user.photo.big_file_id, file_name=f'{configGet("cache", "locations")}{sep}avatars{sep}{user.photo.big_file_id}')
|
|
logWrite("Avatars caching performed")
|
|
|
|
|
|
if configGet("enabled", "scheduler", "birthdays"):
|
|
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "birthdays"))
|
|
async def check_birthdays():
|
|
for entry in col_applications.find():
|
|
if entry["application"]["2"].strftime("%d.%m") == datetime.now().strftime("%d.%m"):
|
|
try:
|
|
tg_user = await app.get_users(entry["user"])
|
|
await app.send_message( configGet("admin_group"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), entry["application"]["2"], '%d.%m.%Y').years)) ) # type: ignore
|
|
logWrite(f"Notified admins about {entry['user']}'s birthday")
|
|
except Exception as exp:
|
|
logWrite(f"Could not find user {entry['user']} to send a message about birthday due to '{exp}'")
|
|
continue
|
|
logWrite("Birthdays check performed")
|
|
|
|
if configGet("enabled", "scheduler", "sponsorships"):
|
|
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "sponsorships"))
|
|
async def check_sponsors():
|
|
logWrite("Sponsorships check performed") |