added user login (not tested)
This commit is contained in:
parent
2e95660945
commit
28ef0a6ff7
@ -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)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user