TriggerBotSqlite.py 文件源码

def check_args():
    print("checking args")
    parser = argparse.ArgumentParser()
    parser.add_argument('-t', '--token', help="Token received from botfather.")
    parser.add_argument('-o', '--owner', help="Telegram chat ID to send status messages.", type=int)
    parser.add_argument('-d', '--dict', help="Keep database as a dict object stored in ram", default=False)
    args = parser.parse_args()
    if(args.dict):
        global triggers_dict
        triggers_dict = True
    if (args.token):
        if (args.owner):
            try:
                dummy_bot = telebot.TeleBot(args.token)
                bot_info = dummy_bot.get_me()
                bot_user, created = TGUserModel.get_or_create(
                    chat_id=bot_info.id,
                    first_name=bot_info.first_name,
                    username=bot_info.username
                )
                bot_user.save()
                try:
                    dummy_bot.send_message(args.owner, "This bot is ready!")
                    bot_cfg = ConfigModel.create(bot_user=bot_user, token=args.token, owner=args.owner)
                    bot_cfg.save()
                except ApiException as ae:
                    print('''Make sure you have started your bot https://telegram.me/{}.
                        And configured the owner variable.
                        ApiException: {}'''.format(bot_info.username, ae))
                    exit(1)
            except ApiException as ApiError:
                print("Invalid token[{}]: {}".format(args.token, ApiError))
                exit(1)
            except Exception as e:
                print(e)
                exit(1)
        else:
            print("Owner ID not supplied")
            exit(1)