From b185dff6640cd002fd30cf7550ec266516ddec80 Mon Sep 17 00:00:00 2001 From: Profitroll <47523801+profitrollgame@users.noreply.github.com> Date: Sun, 11 Dec 2022 19:52:14 +0100 Subject: [PATCH] Start is now Mongo-ready --- modules/commands/start.py | 31 +++++++++++++++++-------------- 1 file changed, 17 insertions(+), 14 deletions(-) diff --git a/modules/commands/start.py b/modules/commands/start.py index 42c4cf2..da6eb29 100644 --- a/modules/commands/start.py +++ b/modules/commands/start.py @@ -2,24 +2,27 @@ from app import app from os import sep from pyrogram import filters from pyrogram.types import ReplyKeyboardMarkup -from modules.utils import jsonLoad, jsonSave, configGet, configSet, locale, logWrite +from modules.utils import locale, logWrite +from modules.database import col_users # Start command ================================================================================================================ @app.on_message(~ filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"])) async def cmd_start(app, msg): - try: - user_stage = configGet("stage", file=str(msg.from_user.id)) - if user_stage != 0: - return - except FileNotFoundError: - jsonSave(jsonLoad(f"{configGet('data', 'locations')}{sep}user_default.json"), f"{configGet('data', 'locations')}{sep}users{sep}{msg.from_user.id}.json") - user_stage = configGet("stage", file=str(msg.from_user.id)) - configSet(["telegram_id"], str(msg.from_user.username), file=str(msg.from_user.id)) - configSet(["telegram_name"], f"{msg.from_user.first_name} {msg.from_user.last_name}", file=str(msg.from_user.id)) - configSet(["telegram_phone"], str(msg.from_user.phone_number), file=str(msg.from_user.id)) - configSet(["telegram_locale"], str(msg.from_user.language_code), file=str(msg.from_user.id)) + user = col_users.find_one({"user": msg.from_user.id}) - logWrite(f"User {msg.from_user.id} started bot interaction") - await msg.reply_text(locale("start", "message"), reply_markup=ReplyKeyboardMarkup(locale("welcome", "keyboard"), resize_keyboard=True)) + if user is None: + + col_users.insert_one({ + "user": msg.from_user.id, + "link": None, + "label": "", + "tg_name": msg.from_user.first_name, + "tg_phone": msg.from_user.phone_number, + "tg_locale": msg.from_user.language_code, + "tg_username": msg.from_user.username + }) + + logWrite(f"User {msg.from_user.id} started bot interaction") + await msg.reply_text(locale("start", "message"), reply_markup=ReplyKeyboardMarkup(locale("welcome", "keyboard"), resize_keyboard=True)) # ============================================================================================================================== \ No newline at end of file