72 lines
2.0 KiB
Python
Executable File
72 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import logging
|
|
from aiogram import executor
|
|
from database import db, Member, Restriction
|
|
|
|
from load import dp, bot
|
|
import filters
|
|
|
|
dp.filters_factory.bind(filters.AvaibleRolesFilter)
|
|
dp.filters_factory.bind(filters.ReplayMessageFilter)
|
|
|
|
import handlers
|
|
import config
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
|
|
|
|
WEBAPP_HOST = '127.0.0.1'
|
|
WEBAPP_PORT = 3001
|
|
|
|
# Don`t touch anything!
|
|
WEBHOOK_HOST = f'http://{WEBAPP_HOST}:{WEBAPP_PORT}'
|
|
WEBHOOK_PATH = f'/bot{config.token}/'
|
|
WEBHOOK_URL = f"{WEBHOOK_HOST}{WEBHOOK_PATH}"
|
|
|
|
async def on_startup(dp):
|
|
if not db.get_columns('members'):
|
|
db.create_tables([Member,Restriction])
|
|
logging.warning("Member table is empty")
|
|
await bot.send_message(config.second_group_id,"First launch successful!")
|
|
await bot.send_message(config.second_group_id,"Member table is empty, run: `!reload`",parse_mode="Markdown")
|
|
|
|
elif Member.select().count() == 0:
|
|
await bot.send_message(config.second_group_id,"Member table is empty, run `!reload`",parse_mode="Markdown")
|
|
logging.warning("Member table is empty")
|
|
|
|
from utils.notify_start import notify_started_bot
|
|
await notify_started_bot(bot)
|
|
|
|
from utils.default_commands import set_default_commands
|
|
await set_default_commands(dp)
|
|
|
|
from load import tgc
|
|
await tgc.client.start()
|
|
|
|
await bot.set_webhook(WEBHOOK_URL)
|
|
|
|
async def on_shutdown(dp):
|
|
await bot.delete_webhook()
|
|
|
|
# Close Redis connection.
|
|
await dp.storage.close()
|
|
await dp.storage.wait_closed()
|
|
|
|
def main() -> None:
|
|
|
|
if config.USE_WEBHOOK:
|
|
executor.start_webhook(
|
|
dispatcher=dp,
|
|
webhook_path=WEBHOOK_PATH,
|
|
on_startup=on_startup,
|
|
on_shutdown=on_shutdown,
|
|
skip_updates=True,
|
|
host=WEBAPP_HOST,
|
|
port=WEBAPP_PORT,
|
|
)
|
|
|
|
else:
|
|
executor.start_polling(dp,skip_updates=True)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|