added user login (not tested)

This commit is contained in:
Linus E 2022-05-03 17:52:48 +02:00
parent 2e95660945
commit 28ef0a6ff7

View File

@ -18,6 +18,7 @@ import telebot
from dotenv import load_dotenv from dotenv import load_dotenv
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from telebot import types from telebot import types
from random import randrange
from db import User, session, Product from db import User, session, Product
from fetcher import * from fetcher import *
@ -50,15 +51,27 @@ def start_name_setter(message):
Args: Args:
message (Message): Message to react to message (Message): Message to react to
""" """
if str(message.text).lower() == "cancel": # Set user name to user user_id = int(message.from_user.id)
bot.reply_to(message, "Your username will be set to default_user. \ user_name = ""
\nType /gameinfo for information about GuessThePrice \
\nType /help for an overview of all commands") if str(message.text).lower() == "cancel" or "auto" or "stop": # Set user name to user
user_name = "NewUser" + str(randrange(0, 9999999)) # generate random name, user can change it with /changename
else: else:
bot.reply_to(message, f"Thank you for setting your name {message.text} \ user_name = str(message.text)
bot.reply_to(message, f"Thank you for setting your name {user_name} \
\nType /gameinfo for information about GuessThePrice \ \nType /gameinfo for information about GuessThePrice \
\nType /help for an overview of all commands") \nType /help for an overview of all commands")
try:
user = User(telegram_id=user_id, username=user_name, admin=False)
session.add(user)
session.commit()
except sqlalchemy.exc.IntegrityError:
session.rollback()
bot.reply_to(message, "You are already registered, change name with /changename")
telebot.logger.setLevel(logging.DEBUG) telebot.logger.setLevel(logging.DEBUG)