aboutsummaryrefslogtreecommitdiff
path: root/main.py
blob: 4ed98a741026d858794f498bdcf8830ef606a672 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import optparse

from aiogram import executor, types as t, Dispatcher
import logging
from shared import config

logging.basicConfig(level=logging.INFO)

parser = optparse.OptionParser(conflict_handler="resolve")  # Делает прикол с аргументами
parser.add_option('-t', '--test',
                  action="store_true",
                  dest='test',
                  help='test token')
parser.add_option('-m', '--main',
                  action="store_true",
                  dest='main',
                  help='main token')
values, args = parser.parse_args()  # Либа хуйня

if values.test:
    config.token = config.test_token
elif values.main:
    config.token = config.main_token
else:
    config.token = config.test_token


async def on_start(dp: Dispatcher):
    from shared.commands import commands
    for scope, cmd in commands.items():
        await dp.bot.delete_my_commands(scope)
        await dp.bot.set_my_commands(cmd, scope)


if __name__ == '__main__':
    from shared.instances import dp
    import handlers

    executor.start_polling(
        dp, allowed_updates=t.AllowedUpdates.all(), on_startup=on_start, skip_updates=True
    )