From 47df0f93f2c4e91246f2a7ef36086dd159c24adf Mon Sep 17 00:00:00 2001
From: Igor <50257429+igorechek06@users.noreply.github.com>
Date: Sat, 28 Aug 2021 21:14:07 +0900
Subject: 0.0.1

---
 main.py             | 23 +++++++++++++++++++++++
 shared/instances.py |  4 ++--
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index d9dce7e..af60614 100644
--- a/main.py
+++ b/main.py
@@ -1,17 +1,40 @@
+import optparse
+
 from aiogram import executor, types as t, Dispatcher
 from shared.instances import dp, bot
 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 bot.set_my_commands(cmd, scope)
 
+
 if __name__ == '__main__':
     import handlers
+
     executor.start_polling(
         dp, allowed_updates=t.AllowedUpdates.all(), on_startup=on_start
     )
diff --git a/shared/instances.py b/shared/instances.py
index 7eb883c..0a6153d 100644
--- a/shared/instances.py
+++ b/shared/instances.py
@@ -1,5 +1,5 @@
 from aiogram import Bot, Dispatcher
-from .config import TOKEN
+from .config import token
 
-bot = Bot(token=TOKEN)
+bot = Bot(token=token)
 dp = Dispatcher(bot)
-- 
cgit v1.2.3