mirror of
https://git.wownero.com/lza_menace/tg-bot.git
synced 2024-08-15 00:23:12 +00:00
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
import logging
|
|
from tipbot import wownero
|
|
from tipbot.db import User
|
|
from tipbot.helpers.decorators import wallet_rpc_required, log_event, check_debug
|
|
from tipbot.helpers.utils import reply_user
|
|
|
|
|
|
@wallet_rpc_required
|
|
@log_event
|
|
@check_debug
|
|
def register(update, context):
|
|
wallet = wownero.Wallet()
|
|
msg = update.message
|
|
uid = msg.from_user['id']
|
|
un = getattr(msg.from_user, 'username', None)
|
|
if un is None:
|
|
_m = 'You need a username configured in Telegram to use this bot.'
|
|
reply_user(msg, context, _m)
|
|
return False
|
|
|
|
if User.filter(telegram_id=uid):
|
|
if User.filter(telegram_id=uid, telegram_user=un):
|
|
_m = 'You are already registered. Use /help to see available bot commands.'
|
|
reply_user(msg, context, _m)
|
|
else:
|
|
try:
|
|
u = User.get(telegram_id=uid)
|
|
u.telegram_user = un
|
|
u.save()
|
|
_m = f'You have been registered again as Telegram ID {uid} but with username {un}.'
|
|
reply_user(msg, context, _m)
|
|
return True
|
|
except Exception as e:
|
|
logging.error(f'Unable to update user in DB: {e}. Debug: {msg}')
|
|
_m = 'Unable to update your existing account. Ask for help.'
|
|
reply_user(msg, context, _m)
|
|
return False
|
|
else:
|
|
try:
|
|
account = wallet.new_account(label=un)
|
|
except Exception as e:
|
|
logging.error(f'Unable to create a new account in wallet RPC: {e}. Debug: {msg}')
|
|
_m = 'Unable to create a new account for you. Ask for help.'
|
|
reply_user(msg, context, _m)
|
|
return False
|
|
try:
|
|
u = User(
|
|
telegram_id=uid,
|
|
telegram_user=un,
|
|
account_index=account[0],
|
|
address=account[1]
|
|
)
|
|
u.save()
|
|
reply_text = [
|
|
f'You have been registered as Telegram ID {uid} and username {un} and can now send and receive tips.',
|
|
'Ask for /help to see all available bot commands. Maybe start with /deposit to get your deposit address.'
|
|
]
|
|
_m = ' '.join(reply_text)
|
|
reply_user(msg, context, _m)
|
|
return True
|
|
except Exception as e:
|
|
logging.error(f'Unable to register user in DB: {e}. Debug: {msg}')
|
|
_m = 'Unable to create a new account for you. Ask for help.'
|
|
reply_user(msg, context, _m)
|
|
return False
|