def resin(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
cur_resin = db.get_resin(uid)
msg = (f"❗ Argument must be an integer greater than "
f"{cur_resin} and lower than {ut.RESIN_MAX}, "
f"e.g. /resin, /resin 135")
if context.args:
try:
value = int(context.args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if cur_resin < value < ut.RESIN_MAX:
msg = ut.cap_format(uid, value)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
else:
msg = ut.cap_format(uid)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
_state(uid)
ut.send(update, msg)
def _timezone(args, uid, msg):
try:
datetime.strptime(args[0], "%H:%M")
except ValueError:
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if value == -1:
db.unset_timezone(uid)
msg = ut.text_format("Current timezone", "disabled")
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
else:
hour, minutes = map(int, args[0].split(':'))
bot_hour, bot_minutes = map(
int,
datetime.now().strftime("%H:%M").split(':'))
tz_hour = hour - bot_hour
tz_minutes = minutes - bot_minutes
db.set_timezone(uid, tz_hour, tz_minutes)
tz = ut.normalize_timezone(tz_hour, tz_minutes)
msg = (
f"{ut.text_format('Bot hour', f'{bot_hour:02}:{bot_minutes:02}')}\n\n" # noqa
f"{ut.text_format('Current timezone', f'{tz}')}")
db.dec_strikes(uid)
return msg
def notifications(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
msg = ("❗ Argument must be 1 or -1, "
"e.g. /notifications 1, /notifications -1")
if context.args:
try:
value = int(context.args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if abs(value) == 1:
if value == -1:
db.unset_notifications(uid)
value = "disabled"
else:
db.set_notifications(uid)
value = "enabled"
msg = ut.text_format("Current notifications status",
value)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
else:
status = ('enabled'
if db.get_notifications(uid) == 1 else 'disabled')
msg = ut.text_format("Current notifications status", status)
db.dec_strikes(uid)
ut.send(update, msg)
def _refill(args, uid, msg, current, max_resin):
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if 0 < value < max_resin:
db.inc_resin(uid, value)
msg = ut.text_format("Current resin", current + value)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
return msg
def _spend(args, uid, msg, current):
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if 0 < value < current:
db.dec_resin(uid, value)
msg = ut.text_format("Current resin", current - value)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
return msg
def _set_resin(args, uid, msg):
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if 0 < value < ut.RESIN_MAX:
db.set_resin(uid, value)
msg = ut.text_format("Current resin", value)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
return msg
def set_resin(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
msg = (f"❗ Argument must be an integer lower than {ut.RESIN_MAX}, "
f"e.g. /set, /set 12")
if context.args:
msg = _set_resin(context.args, uid, msg)
else:
msg = (f"Tell me your current resin "
f"(max: <code>{ut.RESIN_MAX}</code>)")
_state(uid, ut.CMD.SET)
db.dec_strikes(uid)
ut.send(update, msg)
def track(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
msg = ("❗ Argument must have format: mm:ss, "
"e.g. /track, /track 5:35m /track -1")
if context.args:
msg = _track(context.args, context.bot, uid, msg)
else:
msg = ("Tell me your genshin timer "
"in format <code>mm:ss</code>, "
"or <code>-1</code> to disable")
_state(uid, ut.CMD.TRACK)
db.dec_strikes(uid)
ut.send(update, msg)
def _warnings(args, uid, msg):
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if value == -1 or 0 < value < ut.RESIN_MAX:
if value == -1:
value = "disabled"
db.unset_warn(uid)
else:
db.set_warn(uid, value)
msg = ut.text_format("Current warning threshold", value)
msg = _synchronized(uid, msg)
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
return msg
def spend(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
cur_resin = db.get_resin(uid)
msg = (f"❗ Argument must be an integer greater than 0 "
f"and lower than {cur_resin}, "
f"e.g. /spend, /spend 20")
if context.args:
msg = _spend(context.args, uid, msg, cur_resin)
else:
msg = (f"Tell me how much resin to spend "
f"(max: <code>{cur_resin}</code>)")
_state(uid, ut.CMD.SPEND)
db.dec_strikes(uid)
ut.send(update, msg)
def refill(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
cur_resin = db.get_resin(uid)
max_resin = ut.RESIN_MAX - cur_resin
msg = (f"❗ Argument must be an integer greater than 0 "
f"and lower than {max_resin}, "
f"e.g. /refill, /refill 20")
if context.args:
msg = _refill(context.args, uid, msg, cur_resin, max_resin)
else:
msg = (f"Tell me how much resin to refill "
f"(max: <code>{max_resin}</code>)")
_state(uid, ut.CMD.REFILL)
db.dec_strikes(uid)
ut.send(update, msg)
def warnings(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
msg = (f"❗ Argument must be an integer greater than 0 "
f"and lower than {ut.RESIN_MAX}, or -1, "
f"e.g. /warnings, /warnings -1, /warning 140")
if context.args:
msg = _warnings(context.args, uid, msg)
else:
cur_warn = db.get_warn(uid)
if cur_warn == -1:
cur_warn = "disabled"
msg = (f"{ut.text_format('Warning threshold', cur_warn)}\n\n"
f"Tell me resin value to be warned at, "
f"or <code>-1</code> to disable")
_state(uid, ut.CMD.WARN)
db.dec_strikes(uid)
ut.send(update, msg)
def _track(args, bot, uid, msg):
try:
datetime.strptime(args[0], "%M:%S")
except ValueError:
try:
value = int(args[0])
except ValueError:
msg = ut.strike_user(uid, msg)
else:
if value == -1:
th.del_thread(uid)
msg = ut.text_format("Current tracking status", "disabled")
db.dec_strikes(uid)
else:
msg = ut.strike_user(uid, msg)
else:
minutes, seconds = map(int, args[0].split(':'))
timer = minutes * 60 + seconds
th.new_thread(bot, uid, timer)
msg = "Bot timer synchronized."
db.dec_strikes(uid)
return msg
def timezone(update, context):
uid = update.effective_message.chat.id
if not db.banned(uid):
if not db.cached(uid):
ut.not_started(update)
else:
msg = ("❗ Argument must have format(24h): hh:mm or -1, "
"e.g. /timezone, /timezone -1, /timezone 18:30")
if context.args:
msg = _timezone(context.args, uid, msg)
else:
tz_hour, tz_minutes = db.get_timezone(uid).split(':')
if tz_hour == 'null':
tz = "disabled"
else:
tz = ut.normalize_timezone(tz_hour, tz_minutes)
msg = (f"{ut.text_format('Current timezone', tz)}\n\n"
f"Tell me your current hour "
f"in format(24h): <code>hh:mm</code>, "
f"or <code>-1</code> to disable")
_state(uid, ut.CMD.TZ)
db.dec_strikes(uid)
ut.send(update, msg)
