From 87d95fbfeda6e98c774f652cde803b5f9c89d1af Mon Sep 17 00:00:00 2001 From: Davte Date: Tue, 7 Apr 2020 14:00:39 +0200 Subject: [PATCH] Create users table if missing --- davtelepot/__init__.py | 2 +- davtelepot/bot.py | 44 ++++++++++++++++++++++++++++++------------ 2 files changed, 33 insertions(+), 13 deletions(-) diff --git a/davtelepot/__init__.py b/davtelepot/__init__.py index 8f61fad..2f758e9 100644 --- a/davtelepot/__init__.py +++ b/davtelepot/__init__.py @@ -14,7 +14,7 @@ __author__ = "Davide Testa" __email__ = "davide@davte.it" __credits__ = ["Marco Origlia", "Nick Lee @Nickoala"] __license__ = "GNU General Public License v3.0" -__version__ = "2.4.15" +__version__ = "2.4.16" __maintainer__ = "Davide Testa" __contact__ = "t.me/davte" diff --git a/davtelepot/bot.py b/davtelepot/bot.py index da0f5cd..827abb7 100644 --- a/davtelepot/bot.py +++ b/davtelepot/bot.py @@ -223,18 +223,38 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject): self.shared_data = dict() self.Role = None # Add `users` table with its fields if missing - self.db['users'].upsert( - dict( - telegram_id=000000000, - privileges=100, - username="username", - first_name="First", - last_name="Last", - language_code="en", - selected_language_code="en" - ), - ['telegram_id'] - ) + if 'users' not in self.db.tables: + table = self.db.create_table( + table_name='users' + ) + table.create_column( + 'telegram_id', + self.db.types.integer + ) + table.create_column( + 'privileges', + self.db.types.integer + ) + table.create_column( + 'username', + self.db.types.string + ) + table.create_column( + 'first_name', + self.db.types.string + ) + table.create_column( + 'last_name', + self.db.types.string + ) + table.create_column( + 'language_code', + self.db.types.string + ) + table.create_column( + 'selected_language_code', + self.db.types.string + ) return @property