allow lvlup multiple levels
This commit is contained in:
parent
aa77da14df
commit
207bc43487
GPG key ID:
9D9CE43650FF2BAA
2 changed files with 49 additions and 15 deletions
64
cards.py
64
cards.py
|
|
@ -6,6 +6,9 @@ from telegram.ext import run_async
|
|||
from data import get_player
|
||||
from random import randrange
|
||||
from time import time
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger('tgmsbot.cards')
|
||||
|
||||
MAX_LEVEL: int = 100
|
||||
MID_LEVEL: int = 80
|
||||
|
|
@ -45,6 +48,7 @@ def _msg_users(update):
|
|||
|
||||
@run_async
|
||||
def getperm(update, context):
|
||||
logging.info(f'getperm from {getattr(update.effective_user, "id", None)}')
|
||||
(from_user, reply_to_user) = _msg_users(update)
|
||||
if not from_user:
|
||||
return
|
||||
|
|
@ -59,6 +63,7 @@ def getperm(update, context):
|
|||
|
||||
@run_async
|
||||
def setperm(update, context):
|
||||
logging.info(f'setperm from {getattr(update.effective_user, "id", None)}')
|
||||
(from_user, reply_to_user) = _msg_users(update)
|
||||
if not from_user:
|
||||
return
|
||||
|
|
@ -87,39 +92,64 @@ def lvlup(update, context):
|
|||
'''
|
||||
use LVL_UP_CARDS cards to level up 1 lvl
|
||||
'''
|
||||
logging.info(f'lvlup from {getattr(update.effective_user, "id", None)}')
|
||||
LVLUP_TIMEOUT = 10
|
||||
last_time = context.user_data.setdefault('lvlup_time', 0.0)
|
||||
ctime = time()
|
||||
if ctime - last_time < LVLUP_TIMEOUT:
|
||||
update.message.reply_text('别急，你不是刚刚才来过吗\nTips: /lvlup n 可以一次升n级哦')
|
||||
return
|
||||
else:
|
||||
context.user_data['lvlup_time'] = ctime
|
||||
(from_user, reply_to_user) = _msg_users(update)
|
||||
if context.args and len(context.args) == 1:
|
||||
try:
|
||||
amount = int(context.args[0])
|
||||
except ValueError:
|
||||
update.message.reply_text('数字不合法')
|
||||
return
|
||||
else:
|
||||
amount = 1
|
||||
if not from_user:
|
||||
return
|
||||
if reply_to_user:
|
||||
fplayer = get_player(int(from_user.id))
|
||||
tplayer = get_player(int(reply_to_user.id))
|
||||
if fplayer.immunity_cards >= LVL_UP_CARDS:
|
||||
fplayer.immunity_cards -= LVL_UP_CARDS
|
||||
if tplayer.permission <= MAX_LEVEL - 2 or tplayer.permission >= MAX_LEVEL:
|
||||
tplayer.permission += 1
|
||||
amount = abs(amount)
|
||||
if fplayer.immunity_cards >= (used_cards := LVL_UP_CARDS * amount):
|
||||
fplayer.immunity_cards -= used_cards
|
||||
tplayer.permission = MAX_LEVEL - 1 if (_tpp := tplayer.permission + amount) >= MAX_LEVEL - 1 \
|
||||
and tplayer.permission < MAX_LEVEL else _tpp
|
||||
fplayer.save()
|
||||
tplayer.save()
|
||||
update.message.reply_text((f"{display_username(from_user)} 消耗了{LVL_UP_CARDS}张免疫卡，"
|
||||
f"为 {display_username(reply_to_user)} 升了1级"),
|
||||
update.message.reply_text((f"{display_username(from_user)} 消耗了{used_cards}张免疫卡，"
|
||||
f"为 {display_username(reply_to_user)} 升了{amount}级"),
|
||||
parse_mode="Markdown")
|
||||
else:
|
||||
update.message.reply_text(f"您的免疫卡不足({fplayer.immunity_cards})，{LVL_UP_CARDS}张免疫卡兑换1等级",
|
||||
update.message.reply_text(f"您的免疫卡不足({fplayer.immunity_cards})，{used_cards}张免疫卡兑换{amount}等级",
|
||||
parse_mode="Markdown")
|
||||
else:
|
||||
fplayer = get_player(int(from_user.id))
|
||||
if fplayer.immunity_cards >= LVL_UP_CARDS:
|
||||
fplayer.immunity_cards -= LVL_UP_CARDS
|
||||
if fplayer.permission <= MAX_LEVEL - 2 or fplayer.permission >= MAX_LEVEL:
|
||||
fplayer.permission += 1
|
||||
fplayer.save()
|
||||
update.message.reply_text((f"{display_username(from_user)} 消耗了{LVL_UP_CARDS}张免疫卡，"
|
||||
"为 自己 升了1级"), parse_mode="Markdown")
|
||||
if fplayer.immunity_cards >= (used_cards := LVL_UP_CARDS * amount):
|
||||
if amount < 0:
|
||||
if fplayer.permission + amount >= 0:
|
||||
fplayer.immunity_cards += abs(used_cards)
|
||||
else:
|
||||
update.message.reply_text(f"您的免疫卡不足({fplayer.immunity_cards})，{LVL_UP_CARDS}张免疫卡兑换1等级",
|
||||
fplayer.immunity_cards += LVL_UP_CARDS * fplayer.permission
|
||||
else:
|
||||
fplayer.immunity_cards -= abs(used_cards)
|
||||
fplayer.permission = MAX_LEVEL - 1 if (_fpp := fplayer.permission + amount) >= MAX_LEVEL - 1 \
|
||||
and fplayer.permission < MAX_LEVEL else _fpp
|
||||
fplayer.save()
|
||||
update.message.reply_text((f"{display_username(from_user)} 消耗了{used_cards}张免疫卡，"
|
||||
f"为 自己 升了{amount}级"), parse_mode="Markdown")
|
||||
else:
|
||||
update.message.reply_text(f"您的免疫卡不足({fplayer.immunity_cards})，{used_cards}张免疫卡兑换{amount}等级",
|
||||
parse_mode="Markdown")
|
||||
|
||||
@run_async
|
||||
def transfer_cards(update, context):
|
||||
logging.info(f'transfer_cards from {getattr(update.effective_user, "id", None)}')
|
||||
(from_user, reply_to_user) = _msg_users(update)
|
||||
if not from_user:
|
||||
return
|
||||
|
|
@ -160,6 +190,7 @@ def transfer_cards(update, context):
|
|||
|
||||
@run_async
|
||||
def rob_cards(update, context):
|
||||
logging.info(f'rob_cards from {getattr(update.effective_user, "id", None)}')
|
||||
ROB_TIMEOUT = 10
|
||||
last_time = context.user_data.setdefault('rob_time', 0.0)
|
||||
ctime = time()
|
||||
|
|
@ -225,6 +256,7 @@ def rob_cards(update, context):
|
|||
|
||||
@run_async
|
||||
def cards_lottery(update, context):
|
||||
logging.info(f'cards_lottery from {getattr(update.effective_user, "id", None)}')
|
||||
LOTTERY_TIMEOUT = 10
|
||||
last_time = context.user_data.setdefault('lottery_time', 0.0)
|
||||
ctime = time()
|
||||
|
|
@ -249,6 +281,7 @@ def cards_lottery(update, context):
|
|||
|
||||
@run_async
|
||||
def dist_cards(update, context):
|
||||
logging.info(f'dist_cards from {getattr(update.effective_user, "id", None)}')
|
||||
(from_user, _) = _msg_users(update)
|
||||
if not from_user:
|
||||
return
|
||||
|
|
@ -275,6 +308,7 @@ def dist_cards(update, context):
|
|||
|
||||
@run_async
|
||||
def dist_cards_btn_click(update, context):
|
||||
logging.info(f'dist_cards_btn_click from {getattr(update.effective_user, "id", None)}')
|
||||
data = update.callback_query.data
|
||||
user = update.callback_query.from_user
|
||||
omsg = update.callback_query.message
|
||||
|
|
|
|||
|
|
@ -15,7 +15,7 @@ import time
|
|||
import logging
|
||||
|
||||
logging.basicConfig(level=logging.INFO,format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||
logger = logging.getLogger(__name__)
|
||||
logger = logging.getLogger('tgmsbot')
|
||||
|
||||
token = "token_here"
|
||||
updater = Updater(token, workers=8, use_context=True)
|
||||
|
|
|
|||
Loading…
Reference in a new issue