diff --git a/import_users.py b/import_users.py index b05740e..914a614 100755 --- a/import_users.py +++ b/import_users.py @@ -11,6 +11,11 @@ if __name__ == '__main__': conn = get_conn() uids = input('Please input user ids split by spaces: ') for uid in uids.split(): + try: + uid = int(uid) + except (ValueError, TypeError): + pass + conn.root.subscribers[uid] = Subscriber.from_chat(bot.get_chat(uid)) commit() conn.close()