class TestNicknameUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_getNickname_empty(self):
nicknames = NicknameUtil.get_all_nicknames(1, 2)
assert len(nicknames) == 0
def test_setNickname_success(self):
NicknameUtil.set_nickname(guild_id=1, user_id=1, nickname="hello")
nicknames = NicknameUtil.get_all_nicknames(guild_id=1, user_id=1)
assert nicknames == ["hello"]
def test_setNicknames_success(self):
NicknameUtil.set_nickname(guild_id=1, user_id=1, nickname="hello")
NicknameUtil.set_nickname(guild_id=1, user_id=1, nickname="world")
nicknames = NicknameUtil.get_all_nicknames(guild_id=1, user_id=1)
assert nicknames == ["hello", "world"]
def test_setNicknames_clear(self):
NicknameUtil.set_nickname(guild_id=1, user_id=1, nickname="hello")
NicknameUtil.set_nickname(guild_id=1, user_id=1, nickname="world")
NicknameUtil.clear_nickname(guild_id=1, user_id=1)
nicknames = NicknameUtil.get_all_nicknames(guild_id=1, user_id=1)
assert len(nicknames) == 0
class TestStatusUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_mehtod(self, method):
self.database.teardown()
def test_createRestStatus_success(self):
expected_status = StatusUtil.createRestStatus(1, 1, 20, 30)
status = StatusUtil.getStatus(1, 1, StatusType.REST)
assert expected_status == status
def test_applyExpiredStatus_success(self):
c: RPGCharacter = RPGCharacterUtil.createNewRPGCharacter(1)
RPGCharacterUtil.changeHp(c, -1000)
StatusUtil.createRestStatus(1, 1, 20, -500) # expired status
StatusUtil.applyExpiredStatus()
assert None == StatusUtil.getStatus(1, 1, StatusType.REST)
c = RPGCharacterUtil.getRPGCharacter(1)
assert c.hp_current > 0
def test_createComaStatus_success(self):
c: RPGCharacter = RPGCharacterUtil.createNewRPGCharacter(1)
comaStatus = StatusUtil.createComaStatus(guild_id=1,
user=c,
hp_max=c.hp_max)
status = StatusUtil.getStatus(guild_id=1,
member_id=c.id,
type=StatusType.COMA)
assert comaStatus == status
statuslist = StatusUtil.reviveComaStatus()
assert len(statuslist) == 0
ch: RPGCharacter = RPGCharacterUtil.createNewRPGCharacter(2)
comaStatus1 = StatusUtil.createComaStatus(guild_id=1,
user=ch,
hp_max=ch.hp_max)
status1 = StatusUtil.getStatus(guild_id=1,
member_id=ch.id,
type=StatusType.COMA)
assert comaStatus1 == status1
statuslist = StatusUtil.reviveComaStatus()
assert len(statuslist) == 0
assert c.hp_max == c.hp_current
def test_removeComaStatusToRest(self):
c: RPGCharacter = RPGCharacterUtil.createNewRPGCharacter(1)
StatusUtil.createComaStatus(guild_id=1, user=c, hp_max=c.hp_max)
StatusUtil.startResting(guild_id=1, user=c)
status = StatusUtil.getStatus(guild_id=1,
member_id=c.id,
type=StatusType.COMA)
assert status is None
status = StatusUtil.getStatus(guild_id=1,
member_id=c.id,
type=StatusType.REST)
assert status is not None
def test_memberTokenMigration(self):
oldDbPath = getDBPath("old.db")
testDbPath = getDBPath("test.db")
shutil.copyfile(oldDbPath, testDbPath)
database = KfpDb(testDbPath)
assert KfpMigrator.KfpMigrate(database.sqliteDb)
database.teardown()
os.remove(testDbPath)
def test_itemDropHiddenMigration(self):
oldDbPath = getDBPath("item_has_hidden.db")
testDbPath = getDBPath("test.db")
shutil.copyfile(oldDbPath, testDbPath)
database = KfpDb(testDbPath)
assert KfpMigrator.KfpMigrate(database.sqliteDb)
database.teardown()
os.remove(testDbPath)
class TestKujiUtil():
def setup_method(self):
self.database = KfpDb(":memory:")
def teardown_method(self):
self.database.teardown()
def test_canDraw_true(self):
assert KujiUtil.canDrawJp(1)
assert KujiUtil.canDrawCn(1)
assert KujiUtil.canDrawLs(1)
def test_canDrawJp_false(self):
KujiUtil.updateMemberJp(1, 1)
assert not KujiUtil.canDrawJp(1)
def test_canDrawCn_false(self):
KujiUtil.updateMemberCn(1, 0, 0)
assert not KujiUtil.canDrawCn(1)
def test_canDrawLs_false(self):
KujiUtil.updateMemberLs(1, 1)
assert not KujiUtil.canDrawLs(1)
def test_canDraw_false(self):
KujiUtil.updateMemberJp(1, 1)
KujiUtil.updateMemberCn(1, 0, 0)
KujiUtil.updateMemberLs(1, 1)
assert not KujiUtil.canDrawLs(1)
def test_getHistoryJp(self):
KujiUtil.updateMemberJp(1, 1)
history = KujiUtil.getHistoryJp(1)
assert history[0] == 1
def test_getHistoryLs(self):
KujiUtil.updateMemberLs(1, 1)
history = KujiUtil.getHistoryLs(1)
assert history[0] == 1
def test_getHistoryCn(self):
KujiUtil.updateMemberCn(1, 1, 2)
history = KujiUtil.getHistoryCn(1)
assert history[0] == 1
assert history[1] == 2
def test_clearData(self):
KujiUtil.updateMemberJp(1, 1)
KujiUtil.clearData()
assert KujiUtil.canDrawJp(1)
KujiUtil.updateMemberCn(1, 1, 2)
assert not KujiUtil.canDrawCn(1)
class TestForwardUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_getAllForward_empty(self):
forwards = ForwardUtil.get_all_forward()
assert len(forwards) == 0
def test_getAllForward_empty(self):
ForwardUtil.create_forward(1, 1, 1, 2)
ForwardUtil.create_forward(1, 1, 1, 5)
forwards = ForwardUtil.get_all_forward()
assert len(forwards) == 2
def test_getForward_empty(self):
forwards = ForwardUtil.get_forward(1, 1)
assert len(forwards) == 0
def test_createForward_success(self):
ForwardUtil.create_forward(1, 1, 1, 2, True)
forward: Forward = ForwardUtil.get_forward(1, 1)[0]
assert forward.receive_channel_id == 2
assert forward.receive_guild_id == 1
assert forward.delete_original == True
def test_createForward_noOverride(self):
ForwardUtil.create_forward(1, 1, 1, 2)
ForwardUtil.create_forward(1, 1, 1, 3)
forwards = ForwardUtil.get_forward(1, 1)
assert forwards[0].receive_channel_id == 2
assert forwards[1].receive_channel_id == 3
def test_deleteForward_noRecord(self):
ForwardUtil.delete(123)
forwards = ForwardUtil.get_all_forward()
assert len(forwards) == 0
def test_deleteForward_success(self):
ForwardUtil.create_forward(1, 1, 1, 2)
ForwardUtil.delete(1)
forwards = ForwardUtil.get_all_forward()
assert len(forwards) == 0
async def on_ready():
print('Logged in as')
print(bot.user.name)
print(bot.user.id)
print(' -- -- -- ')
tmpFile = getTempFile()
if tmpFile.exists():
print(tmpFile)
f = open(tmpFile, 'r')
line = f.read().split(".")
guild_id = int(line[0])
pid = int(line[1])
f.close()
os.remove(tmpFile.absolute())
try:
os.kill(pid, signal.SIGKILL)
except OSError:
print(f"pid {pid} is not found. Ignore")
else:
print("restart successful, sending success message")
db = KfpDb()
channel: Channel = ChannelUtil.getRebootMessageChannel(guild_id)
if channel:
await bot.get_channel(channel.channel_id
).send("更新結束, 現在版本 {}".format(get_version()))
refreshStatus.start()
reviveComaStatus.start()
async def command_restart(ctx, *attr):
db = KfpDb()
ChannelUtil.setRebootMessageChannel(guild_id=ctx.guild.id,
channel_id=ctx.channel.id)
await ctx.send("現在版本 {}, 檢查更新中...".format(get_version()))
tmpFile = getTempFile()
tmpFile.touch()
f = open(tmpFile, "a")
f.write(f"{ctx.guild.id}.")
f.write(f"{os.getpid()}")
f.close()
Popen([os.sep.join((os.getcwd(), "update_and_restart.sh"))], shell=True)
class TestPoliceUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_getCurrentPolice_empty(self):
currentPoliceType = PoliceUtil.getCurrentPoliceType(guild_id=1,
user_id=1)
assert len(currentPoliceType) == 0
def test_createNewPolice_success(self):
assert PoliceUtil.createNewPolice(guild_id=1, user_id=1, type="SLEEP")
currentPoliceType = PoliceUtil.getCurrentPoliceType(guild_id=1,
user_id=1)
assert currentPoliceType == "SLEEP"
def test_createNewPolice_failed(self):
assert PoliceUtil.createNewPolice(guild_id=1, user_id=1, type="SLEEP")
assert not PoliceUtil.createNewPolice(
guild_id=1, user_id=1, type="STUDY")
def test_stopPolice_fail(self):
assert not PoliceUtil.stopPolice(guild_id=1, user_id=1)
def test_stopPolice_success(self):
assert PoliceUtil.createNewPolice(guild_id=1, user_id=1, type="SLEEP")
assert PoliceUtil.stopPolice(guild_id=1, user_id=1)
assert len(PoliceUtil.getCurrentPoliceType(guild_id=1, user_id=1)) == 0
def test_timeout_success(self):
assert PoliceUtil.createNewPoliceWithDuration(guild_id=1,
user_id=1,
duration_min=-61,
type="SLEEP")
assert len(PoliceUtil.getCurrentPoliceType(guild_id=1, user_id=1)) == 0
class TestChannelUtil():
def setup_method(self, method):
self.database = KfpDb(dbFile=":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_setChannel(self):
ChannelUtil.setChannel(123, 321, Util.ChannelType.RANK_UP)
result: Channel = ChannelUtil.GetChannelWithGuild(
123, Util.ChannelType.RANK_UP)[0]
assert result.channel_id == 321
assert result.channel_guild_id == 123
assert result.channel_type == Util.ChannelType.RANK_UP.value
def test_setRankupChannel(self):
ChannelUtil.setRankupChannel(0, 123)
assert ChannelUtil.getMessageChannelId(0) == 123
ChannelUtil.setRankupChannel(0, 456)
assert ChannelUtil.getMessageChannelId(0) == 456
def test_setRankupChannel_notExist(self):
assert not ChannelUtil.getMessageChannelId(123)
class TestRPGCharacterUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_hasAdventureStarted_noMember(self):
assert not RPGCharacterUtil.hasAdventureStared(100)
def test_hasAdventureStarted_hasMember_notStarted(self):
self.database.add_member(20)
assert not RPGCharacterUtil.hasAdventureStared(20)
def test_hasAdventureStarted_started(self):
member = self.database.add_member(33)
RPGCharacterUtil.createNewRPGCharacter(member)
assert RPGCharacterUtil.hasAdventureStared(33)
def test_createNewCharacter_success(self):
member = self.database.add_member(33)
character = RPGCharacterUtil.createNewRPGCharacter(member)
assert character.character == member
def test_createExistedCharacter_false(self):
member = self.database.add_member(33)
RPGCharacterUtil.createNewRPGCharacter(member)
character = RPGCharacterUtil.createNewRPGCharacter(member)
assert character == None
def test_retireCharacter_notExist(self):
assert not RPGCharacterUtil.retireRPGCharacter(123)
def test_retireCharacter_notAdventurer(self):
member = self.database.add_member(33)
assert not RPGCharacterUtil.retireRPGCharacter(member.member_id)
def test_retireCharacter_success(self):
member = self.database.add_member(33)
RPGCharacterUtil.createNewRPGCharacter(member)
RPGCharacterUtil.retireRPGCharacter(member.member_id)
assert not RPGCharacterUtil.hasAdventureStared(member.member_id)
def test_getRPGChracter_success(self):
member = self.database.add_member(100)
rpg = RPGCharacterUtil.createNewRPGCharacter(member)
rpg_get = RPGCharacterUtil.getRPGCharacter(member)
assert rpg_get == rpg
def test_getRPGCharacter_successWithId(self):
member = self.database.add_member(100)
rpg = RPGCharacterUtil.createNewRPGCharacter(member)
rpg_get = RPGCharacterUtil.getRPGCharacter(member.member_id)
assert rpg_get == rpg
def test_levelUpCharacter_success(self):
member = self.database.add_member(100)
rpg: RPGCharacterUtil = RPGCharacterUtil.createNewRPGCharacter(member)
old_hp = rpg.hp_max
old_mp = rpg.mp_max
old_atk = rpg.attack_basic
old_def = rpg.defense_basic
RPGCharacterUtil.levelUpCharacter(member.member_id, 1, 10)
rpg: RPGCharacterUtil = RPGCharacterUtil.getRPGCharacter(member.member_id)
assert rpg.hp_max > old_hp
assert rpg.mp_max > old_mp
assert rpg.attack_basic > old_atk
assert rpg.defense_basic > old_def
def test_changeHp_success(self):
member = self.database.add_member(100)
rpg: RPGCharacterUtil = RPGCharacterUtil.createNewRPGCharacter(member)
expected_hp = rpg.hp_current
dead = RPGCharacterUtil.changeHp(rpg, -10000)
assert (rpg.hp_current == 0) and (dead is True)
dead = RPGCharacterUtil.changeHp(rpg, +100)
assert (rpg.hp_current == expected_hp) and (dead is not True)
class TestMemberUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_getCountZero(self):
assert MemberUtil.get_member_count() == 0
def test_getCountSuccess(self):
self.database.add_member(1)
assert MemberUtil.get_member_count() == 1
self.database.add_member(2)
assert MemberUtil.get_member_count() == 2
def test_getTotalTokenZero(self):
assert MemberUtil.get_total_token() == 0
def test_getTotalTokenSuccess(self):
self.database.add_member(123)
assert MemberUtil.get_total_token() == 100
self.database.add_member(321)
assert MemberUtil.get_total_token() == 200
def test_getTotalCoinZero(self):
assert MemberUtil.get_total_coin() == 0
def test_getTotalCoinSuccess(self):
self.database.add_member(123)
self.database.add_coin(123, 100)
assert MemberUtil.get_total_coin() == 100
self.database.add_member(321)
self.database.add_coin(321, 100)
assert MemberUtil.get_total_coin() == 200
class TestItemUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_searchItem_success(self):
item1 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2)
result = ItemUtil.searchItem(guild_id=1, item_name="hello")
assert result == item1
def test_searchItem_fail(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2)
result = ItemUtil.searchItem(guild_id=1, item_name="he")
assert result == None
def test_createItem_success(self):
item1 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2)
result = ItemUtil.ListAllItem(guild_id=1)
assert len(result) == 1 and result[0] == item1
def test_createItem_failed_itemExist(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2)
result = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2)
assert result == -1
def test_deleteItem_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2,
level_required=0,
price=1)
ItemUtil.deleteItem(guild_id=1, item_name="hello")
result = ItemUtil.ListAllItem(guild_id=1)
assert result == []
def test_deleteItem_failed_noItem(self):
ItemUtil.deleteItem(guild_id=1, item_name="hello")
result = ItemUtil.ListAllItem(guild_id=1)
assert result == []
def test_deleteItems_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=1,
buff_value=-1,
buff_round=2,
level_required=0,
price=1)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=2,
buff_value=10,
buff_round=3,
level_required=0,
price=1)
ItemUtil.createItem(guild_id=1,
item_name="heap",
item_type=ItemType.STATUS,
buff_type=2,
buff_value=10,
buff_round=3,
level_required=0,
price=1)
ItemUtil.createItem(guild_id=1,
item_name="hot",
item_type=ItemType.STATUS,
buff_type=2,
buff_value=10,
buff_round=3,
level_required=0,
price=1)
ItemUtil.deleteItems(guild_id=1)
result = ItemUtil.ListAllItem(guild_id=1)
assert result == []
def test_deleteItems_failed_noItem(self):
ItemUtil.deleteItems(guild_id=1)
result = ItemUtil.ListAllItem(guild_id=1)
assert result == []
def test_IntToItemType(self):
assert ItemType.list()[0] == ItemType.NONE
def test_withRegularDatabase(self):
database = KfpDb(":memory:")
database.add_member(1)
member: Member = database.get_member(1)
assert member
class NewProfile(commands.Cog):
#TODO: add check permiision function, base on roles
db = None
__channels = []
def __init__(self, client, dbFile: str, isTest=False):
self.bot = client
self.db = KfpDb(dbFile)
self.isTest = isTest
@commands.Cog.listener('on_message')
async def profile_on_message(self, message: Message):
if message.author.bot:
return
if message.channel == None or not message.channel.guild.id in whitelist or message.author.bot:
return
if self.populateChannels(message, self.isTest):
return
if not self.channelAllowed(message.channel.id, self.isTest):
return
member: Member = self.db.get_member(message.author.id)
if not member:
self.db.add_member(message.author.id)
member = self.db.get_member(message.author.id)
increaseNumber = randint(10, 25)
rank = self.db.increase_exp(message.channel.guild.id,
message.channel.id, message.author.id,
increaseNumber)
assert rank > 0, 'method increase_xp should not retrun less than 1 in profile_on_message'
if member.rank != rank:
channel = ChannelUtil.getMessageChannelId(message.guild.id)
if channel == None:
channelToUse = message.channel
else:
channelToUse = message.guild.get_channel(channel)
RPGCharacterUtil.levelUpCharacter(message.author.id, member.rank,
rank)
await channelToUse.send('恭喜<@{}> 等級提升至{}。'.format(
message.author.id, rank))
await self.updateUserKfpRoles(message, rank, channelToUse)
self.db.increase_coin(message.guild.id, message.author.id,
increaseNumber)
@commands.group(name='profile', invoke_without_command=True)
async def profile_group(self, ctx: commands.Context, *attr):
if not isWhiteList(ctx):
if ctx.guild:
print(
"{} is not on white list, if you are a developer, add your server to the white list"
.format(ctx.guild.id))
return
if not ChannelUtil.hasChannel(ctx.guild.id, ctx.channel.id,
Util.ChannelType.PROFILE):
print(
"WARNING: cannot run on this channel, if you are a developer, try to use '!commandControll add profile'"
)
return
memberRow: Member = self.db.get_member(ctx.author.id)
if memberRow == None:
self.db.add_member(ctx.author.id)
memberRow = self.db.get_member(ctx.author.id)
profileByte = None
iconData = None
bgData = None
avatar_url = ctx.author.avatar_url_as(format='jpg', size=1024)
if avatar_url._url != None:
iconData = await avatar_url.read()
banner_url = ctx.guild.banner_url
if banner_url._url != None:
bgData = await banner_url.read()
with ProfileImage() as pf:
if iconData:
pf.setIcon(iconData)
if bgData:
pf.setBackGround(bgData)
pf.setCoin(memberRow.coin)
pf.setXp(memberRow.exp)
pf.setLevelNumber(memberRow.rank)
pf.setRankNumber(self.db.get_member_rank_order(ctx.author.id))
pf.setMemberName(ctx.author.display_name, ctx.author.name)
profileByte = pf.generateProfileImage()
discordFile = discord.File(io.BytesIO(profileByte),
filename='profile.png')
await ctx.channel.send(file=discordFile)
@profile_group.command(name='bind')
@commands.check(isWhiteList)
async def profile_group_bind_command(self, ctx: commands.Context, *arg):
channel = ctx.channel
ChannelUtil.setRankupChannel(ctx.guild.id, channel.id)
await channel.send('<@!{}> 設定升級訊息將會於此。'.format(ctx.author.id))
@profile_group.command(name='allowed')
async def profile_allowed_channels_command(self, ctx: commands.Context,
*arg):
msg = "```"
msg += "allowed channel list:\n"
for channel_id in self.__channels:
channel = ctx.guild.get_channel(channel_id)
msg += f"{channel.id}: {channel.name}\n"
msg += "```"
await ctx.channel.send(msg)
# @profile_group.command('force_set_level')
# @commands.check(isWhiteList)
# async def profile_force_set_level(self, ctx:commands.Context, rank=10):
# message = ctx.message
# member: Member = self.db.get_member(message.author.id)
# if not member:
# member = self.db.add_member(message.author.id)
# if member.rank != rank:
# self.db.force_update_rank(member.member_id, rank)
# channel = ChannelUtil.getMessageChannelId(message.guild.id)
# if channel == None:
# channelToUse = message.channel
# else:
# channelToUse = message.guild.get_channel(channel)
# await channelToUse.send('恭喜<@{}> 等級提升至{}。'.format(message.author.id, rank))
# await self.updateUserKfpRoles(message, rank, channelToUse)
# @profile_group.command('force_set_exp')
# async def profile_force_set_exp(self, ctx:commands.Context, exp=-1):
# if exp < 1:
# return
# message = ctx.message
# member: Member = self.db.get_member(message.author.id)
# if not member:
# member = self.db.add_member(message.author.id)
# self.db.set_exp(member.member_id, exp)
# member: Member = self.db.get_member(message.author.id)
# await ctx.send(f"你現在有經驗值:{member.exp}")
# @profile_group.command('get_rank_exp')
# async def profile_get_rank_exp(self, ctx:commands.Context, rank=0):
# if rank < 1:
# return
# exp_needed = Util.get_rank_exp(rank)
# await ctx.send(f"要升級到等級{rank} 需要經驗值:{exp_needed}")
@profile_group.command('items')
async def show_items_command(self, ctx: commands.Context):
if ctx.author.bot:
return # ignore bot
records = InventoryUtil.getAllItemsBelongToUser(
ctx.guild.id, ctx.author.id)
msg = ""
if len(records) > 0:
msg += "你現在有以下物品:\n"
record: InventoryRecord
for record in records:
msg += f"{record.item.name} x {record.amount}\n"
else:
msg += "你目前沒有任何物品"
await ctx.author.send(msg)
@profile_group.command('syncAllRank')
@commands.check(isWhiteList)
async def reset_everyone_rank(self, ctx: commands.Context, rank=0):
member_id_list = Member.select(
Member.member_id).where(Member.rank >= rank)
for member_id in member_id_list:
member: Member = self.db.get_member(member_id)
try:
user = await ctx.guild.fetch_member(member_id)
except NotFound as e:
continue
await self.__updateUserRole(ctx.guild, user, member, member.rank,
None, True)
@profile_group.command(name='leaderboard')
@commands.check(isWhiteList)
async def profile_leaderboard(self, ctx: commands.Context, limit=10):
max_limit = 25
if limit > max_limit:
await ctx.channel.send(f'{limit} 超過上限, 請選擇小於 {max_limit} 的數字')
return
top_leaders = self.db.get_leader_board(limit)
msg = "```"
msg += "員工等級排名:\n"
member: Member
for rank, member in enumerate(top_leaders):
guild_member = ctx.guild.get_member(member.member_id)
user = await self.bot.fetch_user(member.member_id)
if guild_member:
if guild_member.nick:
msg += f"第{rank+1}名: {guild_member.nick}\n"
else:
msg += f"第{rank+1}名: {guild_member.display_name}\n"
elif user:
msg += f"第{rank+1}名: {user.display_name}\n"
msg += "```"
await ctx.channel.send(msg)
async def updateUserKfpRoles(self, message: Message, rank: int,
channelToUse: GuildChannel):
member = Member.select().where(Member.member_id == message.author.id)
user = message.author
await self.__updateUserRole(message.guild, user, member, rank,
channelToUse, False)
async def __updateUserRole(self, guild: Guild, user: User, member: Member,
rank: int, channelToUse: GuildChannel,
internal: bool):
if user:
if member:
newRoles = RoleUtil.getKfpRolesFromLevel(guild.id, rank)
if len(newRoles) > 0:
for newRole in newRoles:
newGuildRole: Role = guild.get_role(newRole.role_id)
if newGuildRole:
if not newGuildRole in user.roles:
# 用戶有新身份組
# 先移除所有不符合的身份組
oldRoles: KfpRole = RoleUtil.getCurrentRoles(
guild.id,
Util.RoleCategory(newRole.category))
if oldRoles:
oldGuildRoles = []
for oldRole in oldRoles:
guildRole = guild.get_role(
oldRole.role_id)
if guildRole and guildRole in user.roles:
oldGuildRoles.append(guildRole)
for oldGuildRole in oldGuildRoles:
await user.remove_roles(oldGuildRole)
# 添加新的身份組
await user.add_roles(newGuildRole)
if internal:
print(
"adding role {} to member {} successed!"
.format(newGuildRole.name, user.name))
else:
embed = Embed()
embed.description = '恭喜<@!{}> 成為 {}'.format(
user.id, newGuildRole.name)
await channelToUse.send(embed=embed)
def populateChannels(self, message: Message, isTest: bool):
if isTest:
return False
if len(self.__channels) == 0:
categories = message.guild.categories
for category in categories:
if category.name == "🐔員工大廳-Hühnerfarm":
channels = category.channels
result = []
for channel in channels:
result.append(channel.id)
self.__channels = result
return True
return False
def channelAllowed(self, channel_id: int, isTest: bool):
if isTest:
return True
return channel_id in self.__channels
def __init__(self, client, dbFile: str, isTest=False):
self.bot = client
self.db = KfpDb(dbFile)
self.isTest = isTest
class TestGamblingUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
pass
def teardown_method(self, method):
pass
def test_gameCreation(self):
game: GamblingGame = GamblingUtil.create_game(123, "new_game", 5,
["hello", "hi"], 123)
assert game.guild_id == 123
assert game.name == "new_game"
assert game.base == 5
assert game.pool == 0
assert game.item_list == ["hello", "hi"]
assert game.creater_id == 123
assert game.status == Util.GamblingStatus.init
def test_updateGameStatus(self):
game: GamblingGame = GamblingUtil.create_game(123, "new_game", 5,
["hello", "hi"], 123)
GamblingUtil.update_game_status(game, Util.GamblingStatus.ready, 123,
456)
result_game = GamblingUtil.get_game(game.id)
assert game == result_game
def test_getActiveGameList(self):
game1: GamblingGame = GamblingUtil.create_game(123, "new_game1", 5,
["hello", "hi"], 123)
game2: GamblingGame = GamblingUtil.create_game(123, "new_game2", 5,
["hello", "hi"], 123)
gameList = GamblingUtil.get_active_games(123)
assert len(gameList) == 0
GamblingUtil.update_game_status(game1, Util.GamblingStatus.ready, 123,
321)
gameList = GamblingUtil.get_active_games(123)
assert len(gameList) == 1
GamblingUtil.update_game_status(game2, Util.GamblingStatus.ready, 123,
321)
gameList = GamblingUtil.get_active_games(123)
assert len(gameList) == 2
assert gameList == [game1, game2]
def test_getActiveGameInChannel(self):
original: GamblingGame = GamblingUtil.create_game(
123, "new_game", 5, ["hello", "hi"], 123)
GamblingUtil.update_game_status(original, Util.GamblingStatus.ready,
123, 456)
game = GamblingUtil.get_active_game_in_channel(123, 456)
assert len(game) == 0
game = GamblingUtil.get_active_game_in_channel(123, 123)
assert game == [original]
another = GamblingUtil.create_game(123, "new_game", 5, ["hello", "hi"],
123)
GamblingUtil.update_game_status(another, Util.GamblingStatus.ready,
123, 456)
game = GamblingUtil.get_active_game_in_channel(123, 123)
assert len(game) == 2
def test_addBetFailedNotReady(self):
original: GamblingGame = GamblingUtil.create_game(
123, "new_game", 5, ["hello", "hi"], 123)
assert not GamblingUtil.add_bet(original, 123, 2, 1)
def test_addBetSuccess(self):
original: GamblingGame = GamblingUtil.create_game(
123, "new_game", 5, ["hello", "hi"], 123)
GamblingUtil.update_game_status(original, Util.GamblingStatus.ready,
123, 456)
assert GamblingUtil.add_bet(original, 123, 10, 1)
original: GamblingGame = GamblingUtil.get_game(original.id)
assert original.pool == 10
assert GamblingUtil.add_bet(original, 123, 10, 1)
assert original.pool == 20
def test_getBets(self):
original: GamblingGame = GamblingUtil.create_game(
123, "new_game", 5, ["hello", "hi"], 123)
GamblingUtil.update_game_status(original, Util.GamblingStatus.ready,
123, 456)
assert GamblingUtil.add_bet(original, 123, 10, 1)
bets = GamblingUtil.get_bets(original)
assert len(bets) == 1
def test_addGamePoolAmount(self):
original: GamblingGame = GamblingUtil.create_game(
123, "new_game", 5, ["hello", "hi"], 123)
GamblingUtil.add_game_pool_amount(original, 300)
original = GamblingUtil.get_game(original.id)
assert original.pool == 300
def test_tokenRateDefault(self):
assert GamblingUtil.DEFAULT_RATE == GamblingUtil.get_token_rate()
def test_tokenRateCorrect(self):
self.database.add_member(123) # token: 100, coin: 0, rate: NAN
assert GamblingUtil.DEFAULT_RATE == GamblingUtil.get_token_rate()
self.database.add_coin(123, 10) # token: 100, coin: 10, rate: 0.1
assert GamblingUtil.DEFAULT_RATE == GamblingUtil.get_token_rate()
self.database.add_coin(123, 90) # token: 100, coin: 100, rate: 1
assert GamblingUtil.get_token_rate() == GamblingUtil.get_token_rate()
self.database.add_coin(123, 500) # token: 100, coin: 600, rate: 6
assert GamblingUtil.get_token_rate() == 6
self.database.add_member(321) # token: 200, coin: 600, rate: 3
assert GamblingUtil.get_token_rate() == GamblingUtil.get_token_rate()
self.database.add_coin(321, 600) # token: 200, coin: 1200, rate: 6
assert GamblingUtil.get_token_rate() == 6
def setup_method(self, method):
self.guild_id = 123
self.database = KfpDb(dbFile=":memory:")
self.database.add_member(default_user_id) # add a default member
class Gambling(commands.Cog):
def __init__(self, client: commands.Bot):
self.bot = client
self.database = KfpDb()
self.betting_permissions = self.database.load_permissions(
Util.ManagementType.Gambling)
@commands.Cog.listener('on_message')
async def profile_on_message(self, message: Message):
if isinstance(message.channel, DMChannel):
return
if self.database.is_channel_auto_clear(
message.guild.id,
message.channel.id) and not message.author.bot:
await message.delete()
@commands.Cog.listener('on_guild_role_delete')
async def betting_on_guild_role_delete(self, old_role: Role):
pass
# new_role = await old_role.guild.create_role(name= '賭盤權限狗(可以自由編輯這個身分組)')
# self.database.update_permission_role(old_role.id, new_role.id, old_role.guild.id, Util.ManagementType.Gambling)
@commands.Cog.listener('on_guild_join')
async def betting_guild_join(self, guild: Guild):
pass
# role = await guild.create_role(name= '賭盤權限狗(可以自由編輯這個身分組)')
# self.database.add_permission_role(guild, role, Util.ManagementType.Gambling)
@commands.command(
name='cheat',
description=
'argv: <@!member_id> token_numbers\n set tokens number that someone owns, this is cheating!'
)
async def betting_cheat_command(self, ctx: commands.Context, *argv):
if ctx.channel == None or ctx.guild == None:
await ctx.author.send('權限錯誤: 請在伺服器中做設置')
return
if len(argv) < 2 or len(argv[0]) < 3 or not argv[0][3:-1].isdigit(
) or not argv[1].isdigit():
await ctx.channel.send('參數錯誤: !cheat @成員 🍗量')
return
check_role = True
for member_role in ctx.author.roles:
if self.database.has_permission(ctx.guild.id, member_role.id,
Util.ManagementType.Gambling):
check_role = False
if check_role:
await ctx.channel.send('權限錯誤: 你沒有使用這個指令的權限')
return
if argv[0].startswith('<@!'):
target_id = argv[0][3:-1]
elif argv[0].startswith('<@'):
target_id = argv[0][2:-1]
else:
await ctx.channel.send('參數錯誤: !cheat @成員 🍗量')
return
target_member = ctx.guild.get_member(int(target_id))
if target_member == None:
await ctx.channel.send('權限錯誤: 無法獲得成員，id: {}'.format(argv[0][3:-1]))
return
self.database.add_token(target_member.id, int(argv[1]))
await ctx.channel.send('將成員: {}的🍗量設置為{}。'.format(
target_member.display_name, argv[1]))
await target_member.send('你的🍗量被{}設置為{}'.format(ctx.author.display_name,
argv[1]))
@commands.group(name='keep_clear', invoke_without_command=True)
async def betting_keep_clear_group(self, ctx: commands.Context, *argv):
if ctx.channel == None:
await ctx.author.send("請在頻道中設置這個指令")
return
if not self.database.has_channel(ctx.guild.id, ctx.channel.id,
Util.ChannelType.AUTO_DELETE):
result = ChannelUtil.setChannel(ctx.guild.id, ctx.channel.id,
Util.ChannelType.AUTO_DELETE)
if result:
await ctx.channel.send('這個頻道將開始自動刪除接下來的所有成員留言')
self.database.set_ignore_xp_channel(ctx.guild.id,
ctx.channel.id)
else:
await ctx.channel.send('這個頻道已經開啟自動刪除')
else:
await ctx.channel.send('這個頻道已經開啟自動刪除')
# 取消自動刪除留言功能
@betting_keep_clear_group.command(name='disable')
async def keep_clear_disable_command(self, ctx: commands.Context, *argv):
if ctx.channel == None:
await ctx.author.send("請在頻道中設置這個指令")
return
if ChannelUtil.hasChannel(ctx.guild.id, ctx.channel.id,
Util.ChannelType.AUTO_DELETE):
ChannelUtil.removeChannel(ctx.guild.id, ctx.channel.id,
Util.ChannelType.AUTO_DELETE)
await ctx.channel.send('取消這個頻道自動刪除成員留言功能')
self.database.remove_ignore_xp_channel(ctx.guild.id,
ctx.channel.id)
# 顯示所有啟動自動刪除留言功能的頻道
@betting_keep_clear_group.command(name='list')
async def keep_clear_list_command(self, ctx: commands.Context, *argv):
if ctx.guild == None:
await ctx.author.send("請在伺服器中呼叫這個指令")
return
result = ''
autoDeleteList = ChannelUtil.GetChannelWithGuild(
ctx.guild.id, Util.ChannelType.AUTO_DELETE)
channel: Channel
for channel in autoDeleteList:
if ctx.uild.get_channel(channel.channel_id) != None:
result += '<#{}>'.format(channel.channel_id)
await ctx.channel.send(result)
# 啟動下注, 格式為 "!bet 加注數量 下注編號 [賭局ID]"
@commands.command(name='bet')
async def betting_bte_command(self, ctx: commands.Context, *argv):
guild = ctx.guild
channel = ctx.channel
##
if channel == None:
await ctx.author.send('請到開啟賭牌的頻道下注!!')
return
if guild == None:
await ctx.author.send('無法處理的伺服器!')
return
if not self.database.has_channel(guild.id, channel.id,
Util.ChannelType.AUTO_DELETE):
await ctx.message.delete()
flag = False
if len(argv) < 2:
flag = True
elif not argv[0].isdigit() or not argv[1].isdigit():
flag = True
if flag:
await ctx.author.send('參數錯誤: `!bet 加注數量 下注編號 [賭局ID]`')
return
bet_amount = int(argv[0]) # 加注數量
choice_id = int(argv[1]) # 下注編號
_bettings = GamblingUtil.get_active_game_in_channel(
guild.id, ctx.channel.id)
ready_games = []
game: GamblingGame
for game in _bettings:
if game.status == Util.GamblingStatus.ready:
ready_games.append(game)
if len(ready_games) == 0:
await ctx.author.send('參數錯誤: 這個頻道沒有開啟的賭局!')
return
if len(ready_games) > 1:
if len(argv) <= 2:
tem_betting_list = ''
for game in ready_games:
tem_betting_list += '\n賭局名:{}, id: {}'.format(
game.name, game.id)
await ctx.author.send('這個頻道有複數賭局開啟中\n請指定賭局`!bet 下注數 賭局ID`' +
tem_betting_list)
return
if not argv[2].isdigit():
await ctx.author.send('參數錯誤: 賭局ID必須是數字')
betting_id = int(argv[2])
flag = True
for game in ready_games:
if betting_id == game.id:
flag = False
break
if flag:
ctx.author.send('參數錯誤: 這個<#{}>沒有ID為: {}的賭局'.format(
ctx.channel.id, betting_id))
ready_games = game
elif len(ready_games) == 1:
ready_games = ready_games[0]
else:
await ctx.channel.send('未預期的錯誤: <@!326752816238428164>快修阿!')
return
game: GamblingGame = ready_games
if game.status != Util.GamblingStatus.ready:
await ctx.author.send('權限錯誤: 現在的賭局狀態為: {}不能下注'.format(
Util.GamblingStatus(game.status).name))
return
if bet_amount < 1:
await ctx.author.send('參數錯誤: 下注🍗不能為此數: {}'.format(bet_amount))
return
# 所有可下注選項
betting_item_list = json.load(game.item_list)
if not choice_id < len(betting_item_list):
await ctx.author.send('參數錯誤: 不存在編號: {}'.format(choice_id))
return
member = self.database.get_member(ctx.author.id)
if member == None:
member = self.database.add_member(ctx.author.id)
require_amount = bet_amount * game.base
if member.coin < require_amount:
await ctx.author.send('道德錯誤: 你的🍗不夠啦! ...剩餘{}，下注{}'.format(
member.coin, require_amount))
return
self.database.add_coin(member, -1 * require_amount)
GamblingUtil.add_bet(game=game,
user_id=member.member_id,
amount=require_amount,
item_index=choice_id)
await ctx.author.send('你成功對{} 下注了{}點🍗。...餘額為: {}。'.format(
betting_item_list[choice_id], require_amount, member.coin))
@commands.group(name='betting', invoke_without_command=True)
async def betting_command_group(self, ctx: commands.Context, *attr):
# TODO print help commands
pass
# 顯示所有賭盤列表
@betting_command_group.command(name='list')
async def betting_list_command(self, ctx: commands.Context, *argv):
guild = ctx.guild
game_list = GamblingUtil.get_active_games(guild.id)
if len(game_list) == 0:
return
embed = Embed()
embed.title = '賭盤列表'
game: GamblingGame
for game in game_list:
channel = ctx.channel
embed.add_field(
name=game.name,
value='每注: {}, 獎金池: {}, 狀態: {}\n頻道: <#{}>, 伺服器:{}'.format(
game.base, game.pool, game.status.name, channel.id,
guild.name),
inline=False)
await ctx.channel.send(embed=embed)
@betting_command_group.command(name='紅包')
async def betting_red_command(self, ctx: commands.Context, *argv):
# 輸入參數檢查開始
if ctx.channel == None:
await ctx.author.send('權限錯誤: 請在頻道中使用這個指令!')
return
if len(argv) < 2 or len(argv) > 2:
await ctx.channel.send('參數錯誤: 請使用`!betitng 紅包 🍗數 紅包量`')
return
if not argv[0].isdigit() and not argv[1].isdigit():
await ctx.channel.send('參數錯誤: 請使用`!betitng 紅包 🍗數 紅包量`')
return
token_num = int(argv[0])
beg_num = int(argv[1])
if token_num < 1:
await ctx.channel.send('參數錯誤: 🍗數必須大於 0`')
return
if beg_num < 1:
await ctx.channel.send('參數錯誤: 紅包量必須大於 0`')
return
# 輸入參數檢查結束
member: Member = self.database.get_member(ctx.author.id)
if member == None:
member = self.database.add_member(ctx.author.id)
required_token = token_num * beg_num
if member.token < required_token:
await ctx.author.send(f"道德錯誤: 同志別裝大款，你只有{member.token}枚🍗。")
# 移除發紅包人的 🍗
self.database.add_token(member.id, -1 * required_token)
main_message = await ctx.channel.send("<@{}> 發紅包拉!!限時1分鐘!!!".format(
ctx.author.id))
await main_message.add_reaction('🤑')
def reaction_check(reaction, user):
if reaction.message == main_message and not user.bot:
return str(reaction.emoji) == '🤑'
else:
return False
temp_list = []
start_time = time.time()
while time.time() - start_time < 60 and beg_num > 0:
try:
reaction = await self.bot.wait_for('reaction_add',
timeout=3,
check=reaction_check)
except asyncio.TimeoutError:
continue
else:
if not reaction[1].id in temp_list:
temp_list.append(reaction[1].id)
member: Member = self.database.get_member(reaction[1].id)
if member == None:
member = self.database.add_member(reaction[1].id)
self.database.add_token(reaction[1].id, token_num)
beg_num -= 1
await ctx.channel.send('恭喜{}從{}的紅包獲得{}點🍗!'.format(
reaction[1].display_name, ctx.author.display_name,
token_num))
if beg_num < 1:
await main_message.edit(content='紅包搶光拉!')
else:
self.database.add_token(ctx.author.id, token_num * beg_num)
await main_message.edit(content='時間到!')
await ctx.channel.send('返還{} 給<@{}>。'.format(
token_num * beg_num, ctx.author.id))
@betting_command_group.command(name='info')
async def betting_info_command(self, ctx: commands.Context, *argv):
member: Member = self.database.get_member(ctx.author.id)
if member == None:
member = self.database.add_member(ctx.author.id)
if ctx.channel == None:
await ctx.author.send(f"您目前持有硬幣{member.coin}\n持有🍗{member.token}根")
else:
await ctx.channel.send(f"您目前持有硬幣{member.coin}\n持有🍗{member.token}根")
# 創立賭盤
@betting_command_group.command(name='create')
async def betting_create_command(self, ctx: commands.Context, *argv):
descript_base = '請<@{}>跟著指示完成創建\n'.format(ctx.author.id)
embed = Embed()
embed.title = '創建賭盤: 創建者<@!{}>'.format(ctx.author.id)
embed.description = descript_base
embed.add_field(name='設定賭盤名稱', value='請直接回覆賭局名稱', inline=False)
embed.add_field(name='設定賭注單位', value='請先回覆賭局名稱', inline=False)
main_message = await ctx.channel.send(embed=embed)
def reaction_check(reaction, user):
if user == ctx.author and reaction.message == main_message:
return str(reaction.emoji) == '⭕' or '❌'
else:
return False
betting_count = 0
bet_item_offset = 2
if not await GamblingUtil.create_loop(embed, main_message, ctx,
type(str()), '賭盤名稱', 0):
return
embed.set_field_at(1, name='設定賭注單位', value='請直接回覆每注單位', inline=False)
await main_message.edit(embed=embed)
if not await GamblingUtil.create_loop(embed, main_message, ctx,
type(int()), '賭注單位', 1):
return
add_flag = True
while add_flag or betting_count < 2:
embed.add_field(name='設定賭注項目-第{}項'.format(betting_count),
value='請先回覆賭注項目-第{}項'.format(betting_count),
inline=False)
await main_message.edit(embed=embed)
if not await GamblingUtil.create_loop(
embed, main_message, ctx, type(
str()), '賭品-第{}項'.format(betting_count),
betting_count + bet_item_offset):
return
if betting_count > 0:
embed.add_field(name='完成設定?',
value='完成設定⭕️繼續設定❌',
inline=False)
await main_message.edit(embed=embed)
await main_message.add_reaction('⭕')
await main_message.add_reaction('❌')
try:
get_reaction = await self.bot.wait_for(
'reaction_add', timeout=30.0, check=reaction_check)
except asyncio.TimeoutError:
embed.set_field_at(betting_count + bet_item_offset + 1,
name='完成設定?-等待反應超時',
value='error')
await main_message.clear_reactions()
await main_message.edit(embed=embed)
return False
else:
if get_reaction[0].emoji == '⭕':
tem_list = []
for i in embed.fields[2:-1]:
tem_list.append(i.value)
game: GamblingGame = GamblingUtil.create_game(
ctx.guild.id, embed.fields[0].value,
int(embed.fields[1].value), tem_list,
ctx.author.id)
embed.set_field_at(
betting_count + bet_item_offset + 1,
name='完成設定!!!',
value=
'設定完成!!!\n請<@{}> 到想要的頻道輸入\n`!betting start {}`\n開啟賭局!'
.format(ctx.author.id, game.id),
inline=False)
add_flag = False
else:
embed.remove_field(betting_count + bet_item_offset + 1)
await main_message.clear_reactions()
await main_message.edit(embed=embed)
betting_count += 1
# 開放賭盤
@betting_command_group.command(name='start')
async def betting_start_command(self, ctx: commands.Context, *argv):
if len(argv) != 1 or not argv[0].isdigit():
await ctx.channel.send('參數錯誤: 請使用`!betitng start 賭局id`')
return
game_id = int(argv[0])
game: GamblingGame = GamblingUtil.get_game(game_id)
if game == None:
await ctx.channel.send(
'參數錯誤: 無法找到id 為:{} 的賭盤。請使用`!betitng list`查詢。'.format(game_id))
return
if game.creater_id != ctx.author.id:
await ctx.channel.send('權限錯誤: 這個賭盤不是你創建的!')
return
if game.guild_id != ctx.guild.id:
await ctx.channel.send('權限錯誤: 這個賭盤不是在這裡創建的，創建的伺服為: {}'.format(
self.bot.get_guild(game.guild_id).name))
return
if game.status != Util.GamblingStatus.init:
await ctx.channel.send('權限錯誤: 這個賭盤的狀態為: {}，無法開始。'.format(
Util.GamblingStatus(game.status).name))
return
embed = GamblingEmbed.get_betting_embed(game)
msg = await ctx.channel.send(embed=embed)
await msg.pin()
GamblingUtil.update_game_status(game, Util.GamblingStatus.ready,
ctx.channel.id, msg.id)
@betting_command_group.command(name='lock')
async def betting_lock_command(self, ctx: commands.Context, *argv):
send_target = None
if ctx.channel == None:
send_target = ctx.author
else:
send_target = ctx.channel
if len(argv) != 1 or not argv[0].isdigit():
await send_target.send('參數錯誤: `!betting lock 賭盤id`')
return
game_id = int(argv[0])
game: GamblingGame = GamblingUtil.get_game(game_id)
if game == None:
await send_target.send('參數錯誤: 沒有ID為{}的賭盤'.format(game_id))
return
if game.creater_id != ctx.author.id:
await send_target.send('權限錯誤: 你不是創建這個賭盤的人')
return
if game.status != Util.GamblingStatus.ready:
await send_target.send('權限錯誤: 這個賭盤的狀態為:{}'.format(game.status.name)
)
return
GamblingUtil.update_game_status(game, Util.GamblingStatus.wait)
await send_target.send('更新賭盤狀態為: {}'.format(
Util.GamblingStatus.wait.name))
@betting_command_group.command(name='end')
async def betting_end_command(self, ctx: commands.Context, *argv):
if len(argv) != 2:
await ctx.channel.send('參數錯誤: `!betting lock 勝利編號 賭盤id`')
return
if not argv[1].isdigit() or not argv[0].isdigit():
await ctx.channel.send('參數錯誤: `!betting lock 勝利編號 賭盤id`')
return
game_id = int(argv[1])
option_index = int(argv[0])
game: GamblingGame = GamblingUtil.get_game(int(argv[1]))
if not game:
await ctx.channel.send('參數錯誤: 找不到id 為{}的賭盤'.format(game_id))
return
if game.creater_id != ctx.author.id:
await ctx.channel.send('權限錯誤: 你不是創建這個賭盤的人')
return
if game.status != Util.GamblingStatus.wait:
await ctx.channel.send('權限錯誤: 這個賭盤的狀態為:{}'.format(game.status.name)
)
return
betting_items = json.loads(game.item_list)
if option_index < 0 or option_index > len(betting_items):
await ctx.channel.send(f'參數錯誤: `勝利編號 {option_index} 為無效編號`')
return
#[第一項總注，第n項總注...,總項總注]
member_charge_sum = [0] * len(betting_items)
member_bet = {}
winning_item = betting_items[option_index]
bets = GamblingUtil.get_bets(game)
bet: GamblingBet
for bet in bets:
member_charge_sum[bet.item_index] += bet.charge
member_bet[bet.member_id] = member_bet.get(bet.member_id,
0) + bet.charge
for member_id in member_bet:
member: Member = self.database.get_member(member_id)
if member == None:
continue
token_spent = 0
if member_bet[member_id].get(winning_item, 0) != 0:
token_spent = member_bet[member_id][winning_item]
coin_won = 0
winning_sum = member_charge_sum[option_index]
if winning_sum != 0:
coin_won = int(token_spent / winning_sum * game.base *
game.pool)
user = await self.bot.fetch_user(member_id)
if user == None:
await ctx.channel.send('無法找到該id的用戶: {}，跳過!')
continue
self.database.add_token(member_id, coin_won)
member = self.database.get_member(member_id)
await user.send('恭喜獲得{}點🍗, ...結餘:{}'.format(coin_won, member.coin))
GamblingUtil.update_game_status(game,
Util.GamblingStatus.end,
winning_index=option_index)
channel = await self.bot.fetch_channel(game.channel_id)
if channel != None:
msg = await channel.fetch_message(game.message_id)
await msg.edit(embed=GamblingEmbed.get_betting_embed(
self.bot, self.database, game))
if msg.pinned:
await msg.unpin()
await ctx.channel.send('結算成功')
@tasks.loop(seconds=5.0)
async def refresh_betting_message(self):
for guild in self.bot.guilds:
games = GamblingUtil.get_active_games(guild.id)
game: GamblingGame
for game in games:
channel = game.guild_id
massage = await channel.fetch_message(game.message_id)
embed = self.get_betting_embed(game)
await massage.edit(embed=embed)
# 查詢兌換率
@betting_command_group.command(name='rate')
async def betting_exchange_rate_command(self, ctx: commands.Context,
*argv):
exchange_rate = GamblingUtil.get_token_rate()
await ctx.channel.send(f"目前🍗兌換率為 {exchange_rate} 硬幣:1隻🍗")
# 兌換🍗
@betting_command_group.command(name='exchange')
async def betting_exchange_command(self, ctx: commands.Context, *argv):
exchange_rate = GamblingUtil.get_token_rate()
if len(argv) != 1:
await ctx.channel.send('參數錯誤: 請使用`!betitng exchange 🍗數量`')
return
if not argv[0].isdigit():
await ctx.channel.send('參數錯誤: 🍗數量必須為數字')
return
desired_token = int(argv[0])
if desired_token < 1:
await ctx.channel.send('參數錯誤: 🍗數量不能低於1')
return
member: Member = self.database.get_member(ctx.author.id)
if member == None:
member = self.database.add_member(ctx.author.id)
required_coin = exchange_rate * desired_token
if member.coin < required_coin:
await ctx.channel.send(
f'參數錯誤: 您目前手持硬幣數量不夠 目前 {exchange_rate}硬幣兌換1🍗\n目前您擁有硬幣{member.coin}, 小於交換所需量 {required_coin}'
)
return
self.database.add_coin(member.member_id, -1 * required_coin)
self.database.add_token(member.member_id, desired_token)
member: Member = self.database.get_member(ctx.author.id)
msg = "```兌換完成! 兌換明細如下:\n"
msg += f"本次兌換匯率: {exchange_rate}\n"
msg += f"本次兌換消耗硬幣: {required_coin}\n"
msg += f"本次兌換獲得🍗: {desired_token}根\n"
msg += f"目前持有硬幣: {member.coin}\n"
msg += f"目前持有🍗: {member.token}根\n"
msg += "```"
await ctx.channel.send(msg)
# 重置所有人
@betting_command_group.command(name='reset_everyone')
async def reset_everyone_command(self, ctx: commands.Context, *argv):
self.database.reset_everyone_token()
class TestKfpDb():
def setup_method(self, method):
self.guild_id = 123
self.database = KfpDb(dbFile=":memory:")
self.database.add_member(default_user_id) # add a default member
def teardown_method(self, method):
self.database.teardown()
for file in os.listdir():
if file.endswith("test.db"):
os.remove(file)
def test_addMember(self):
self.database.add_member(12346)
member = self.database.get_member(12346)
assert member.member_id == 12346
def test_dataIntegrity(self):
self.database = KfpDb(dbFile="tmp_test.db")
self.database.add_member(12346)
self.database = KfpDb(dbFile="tmp_test.db")
member = self.database.get_member(12346)
assert member.member_id == 12346
def test_getMember_notExist(self):
assert not self.database.get_member(100) # user 100 does not exist
def test_addMultipleMembers(self):
member_ids = [1, 2, 3, 4, 5, 6, 7, 8]
self.database.add_members(member_ids)
for member_id in member_ids:
member = Member.get_by_id(member_id)
assert member.member_id == member_id
def test_memberHasUniqueId(self):
with pytest.raises(peewee.IntegrityError):
self.database.add_member(default_user_id)
def test_increaseExp_notExist(self):
assert -1 == self.database.increase_exp(0, 0, 100,
10) # user 100 does not exist
def test_increaseExp(self):
self.database.increase_exp(0, 0, default_user_id, 10)
member = Member.get_by_id(default_user_id)
assert member.exp == 10
def test_ignore_increaseExp(self):
self.database.set_ignore_xp_channel(3, 5)
self.database.increase_exp(3, 5, default_user_id, 10)
member = Member.get_by_id(default_user_id)
assert member.exp == 0
def test_after_remove_ignore_increaseExp(self):
self.database.remove_ignore_xp_channel(3, 5)
self.database.set_ignore_xp_channel(3, 5)
self.database.remove_ignore_xp_channel(3, 5)
self.database.increase_exp(3, 5, default_user_id, 10)
member = Member.get_by_id(default_user_id)
assert member.exp == 10
def test_rankUp(self):
member = Member.get_by_id(default_user_id)
assert member.rank == 0
self.database.increase_exp(0, 0, default_user_id, 100)
member = Member.get_by_id(default_user_id)
assert member.rank == 1
def test_addCoin_notExist(self):
assert not self.database.add_coin(100, 10) # user 100 does not exist
def test_addCoin(self):
self.database.add_coin(default_user_id, 10)
member = self.database.get_member(default_user_id)
assert member.coin == 10
def test_subtractCoin(self):
self.database.add_coin(default_user_id, 100)
self.database.add_coin(default_user_id, -10)
member = self.database.get_member(default_user_id)
assert member.coin == 90
def test_notEnoughMoney(self):
self.database.add_coin(default_user_id, 100)
assert not self.database.add_coin(default_user_id, -1000)
member = self.database.get_member(default_user_id)
assert member.coin == 100
def test_getMemberRankOrder_number1(self):
# 只有一人的時候那人是第一名
assert self.database.get_member_rank_order(0) == 1
def test_getMemberRankOrder_last(self):
self.database.add_member(1)
self.database.add_member(2)
self.database.increase_exp(0, 0, 1, 100)
self.database.increase_exp(0, 0, 2, 100)
assert self.database.get_member_rank_order(0) == 3
def test_getMemberRankOrder_first(self):
self.database.add_member(1)
self.database.add_member(2)
self.database.increase_exp(0, 0, default_user_id, 101)
self.database.increase_exp(0, 0, 1, 100)
self.database.increase_exp(0, 0, 2, 100)
assert self.database.get_member_rank_order(default_user_id) == 1
def test_sameMemberRankOrder_second(self):
self.database.add_member(1)
self.database.add_member(2)
self.database.add_member(3)
self.database.increase_exp(0, 0, default_user_id, 100)
self.database.increase_exp(0, 0, 1, 100)
self.database.increase_exp(0, 0, 2, 1000)
self.database.increase_exp(0, 0, 3, 10)
# 經驗相同時排名相同, 並列第二
assert self.database.get_member_rank_order(default_user_id) == 2
assert self.database.get_member_rank_order(1) == 2
def test_setRankupChannel_notExist(self):
assert not self.database.get_rankup_channel_id(11)
def test_setRankupChannel(self):
self.database.set_rankup_channel(11, 123)
assert self.database.get_rankup_channel_id(11) == 123
self.database.set_rankup_channel(11, 456)
assert self.database.get_rankup_channel_id(11) == 456
def test_resetEveryoneToken(self):
self.database.add_member(1)
self.database.add_member(2)
self.database.add_member(3)
self.database.reset_everyone_token()
assert self.database.get_member(1).token == 100
assert self.database.get_member(2).token == 100
assert self.database.get_member(3).token == 100
def test_dataIntegrity(self):
self.database = KfpDb(dbFile="tmp_test.db")
self.database.add_member(12346)
self.database = KfpDb(dbFile="tmp_test.db")
member = self.database.get_member(12346)
assert member.member_id == 12346
def __init__(self, client: commands.Bot):
self.bot = client
self.database = KfpDb()
self.betting_permissions = self.database.load_permissions(
Util.ManagementType.Gambling)
class TestInventoryUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
def teardown_method(self, method):
self.database.teardown()
def test_getInventory_empty(self):
inventory = InventoryUtil.getAllItemsBelongToUser(1, 1)
assert len(inventory) == 0
def test_buffTypeStorage_success(self):
item = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=30,
buff_round=3)
item_buff: Buff = item.buff
assert item_buff.buff_type == BuffType.ATTACK
assert item_buff.buff_value == 30
assert item_buff.buff_round == 3
def test_buffTypeStorageRead_success(self):
item = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=30,
buff_round=3)
item_from_db = ItemUtil.ListAllItem(guild_id=1)[0]
item_buff: Buff = item_from_db.buff
assert item_buff.buff_type == BuffType.ATTACK
assert item_buff.buff_value == 30
assert item_buff.buff_round == 3
def test_addItemToItemdb_success(self):
item1 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
item2 = ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
items = ItemUtil.ListAllItem(guild_id=1)
assert items == [item1, item2]
def test_addItemToItemdb_failed_samename(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
item2 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
items = ItemUtil.ListAllItem(guild_id=1)
assert len(items) == 1 and item2 == -1
def test_addItemTOShop_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hey",
amount=10)
result = InventoryUtil.ShopMenu(guild_id=1)
assert result == [shopItem1]
def test_addItemTOShop_failed_noitem(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
InventoryUtil.addItemToShop(guild_id=1, item_name="he", amount=10)
result = InventoryUtil.ShopMenu(guild_id=1)
assert result == []
def test_addItemTOShop_success_addAmount(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.ShopMenu(guild_id=1)
assert result[0].amount == 20 and len(result) == 1
def test_addItemTOShop_failed_addAmount(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
InventoryUtil.addItemToShop(guild_id=1, item_name="he", amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="he", amount=10)
result = InventoryUtil.ShopMenu(guild_id=1)
assert result == []
def test_addItemTOShop_failed_unlimitedSupply(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=-1)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.ShopMenu(guild_id=1)
assert result[0].amount == -1
def test_addItemTOShop_failed_limitedToUnlimited(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
shopitem = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=10)
result = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=-1)
assert result == -3 and shopitem.amount == 10
def test_findShopItem_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hey",
amount=10)
result = InventoryUtil.findShopItem(guild_id=1, item=shopItem1.item)
assert result == shopItem1
def test_findShopItem_failed(self):
item1 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
InventoryUtil.addItemToShop(guild_id=1, item_name="hey", amount=10)
result = InventoryUtil.findShopItem(guild_id=1, item=item1)
assert result == None
def test_deleteShopItem(self):
item1 = ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hey", amount=100)
list = InventoryUtil.ShopMenu(guild_id=1)
assert len(list) == 2
InventoryUtil.deleteShopItem(guild_id=1, item=item1)
list = InventoryUtil.ShopMenu(guild_id=1)
assert len(list) == 1
def test_deleteShopItems(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hey", amount=100)
list = InventoryUtil.ShopMenu(guild_id=1)
assert len(list) == 2
InventoryUtil.deleteShopItems(guild_id=1)
list = InventoryUtil.ShopMenu(guild_id=1)
assert len(list) == 0
def test_getAllItemsBelongToUser_empty(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=100)
itemList = InventoryUtil.getAllItemsBelongToUser(guild_id=1,
user_id=123)
assert len(itemList) == 0
def test_buyShopitem_success(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=100)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hey", amount=10)
InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=2)
itemList = InventoryUtil.getAllItemsBelongToUser(guild_id=1,
user_id=123)
assert itemList[0].user_id == 123 and itemList[0].item == shopItem1.item
def test_buyShopitem_success_shopItemchangeHidden(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=500)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=1)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=10)
InventoryUtil.checkZeroAmount(guild_id=1)
result = InventoryUtil.ShopMenu(guild_id=1)
assert len(result) == 0 and result == []
def test_buyShopitems_success(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=500)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3,
level_required=0,
price=10)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=10)
shopItem2 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hey",
amount=10)
InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=2)
InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hey",
count=4)
itemList = InventoryUtil.getAllItemsBelongToUser(guild_id=1,
user_id=123)
assert len(itemList) == 2 and itemList[0].user_id == 123 and itemList[
1].user_id == 123
def test_buyShopitem_failed_noProduct(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=100)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.addItemToShop(guild_id=1,
item_name="hey",
amount=10,
hidden=True)
result = InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="he",
count=2)
assert result == ErrorCode.CannotFindProduct
result == InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hey",
count=2)
assert result == ErrorCode.CannotFindProduct
def test_buyShopitem_failed_levelDoesNotRequired(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=10)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=100,
price=100)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=2)
assert result == ErrorCode.LevelDoesNotReach
def test_buyShopitem_failed_noMoney(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=10)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=100)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=2)
assert result == ErrorCode.TokenDoesNotEnough
def test_buyShopitem_failed_notEnoughSupply(self):
MemberUtil.add_member(member_id=123)
MemberUtil.add_token(member_id=123, amount=100)
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=1)
result = InventoryUtil.buyShopitem(guild_id=1,
user_id=123,
item_name="hello",
count=2)
assert result == ErrorCode.SupplyDoesNotEnough
def test_changeSupplyAmount_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=-1)
assert shopItem1.amount == -1
shopItem2 = InventoryUtil.changeSupplyAmount(guild_id=1,
item_name="hello",
newAmount=10)
assert shopItem2.amount == 10
def test_changeSupplyAmount_success_limitedToUnlimited(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
shopItem1 = InventoryUtil.addItemToShop(guild_id=1,
item_name="hello",
amount=10)
assert shopItem1.amount == 10
shopItem2 = InventoryUtil.changeSupplyAmount(guild_id=1,
item_name="hello",
newAmount=-1)
assert shopItem2.amount == -1
def test_changeSupplyAmount_failed(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
# item cannot find
result1 = InventoryUtil.changeSupplyAmount(guild_id=1,
item_name="he",
newAmount=-1)
assert result1 == -1
# shopitem cannot find
result2 = InventoryUtil.changeSupplyAmount(guild_id=1,
item_name="hey",
newAmount=-1)
assert result2 == -2
def test_changeShopitemHiddenStatus_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="hello",
hidden=True)
assert result.hidden == True
result2 = InventoryUtil.ShopMenu(guild_id=1)
assert len(result2) == 0
result3 = InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="hello",
hidden=False)
assert result3.hidden == False
def test_changeShopitemHiddenStatus_failed_noitem(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="he",
hidden=True)
assert result == -1
def test_changeShopitemHiddenStatus_failed_noShopitem(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="hey",
hidden=True)
assert result == -2
def test_checkShopitemStatus_success(self):
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
result = InventoryUtil.checkShopitemStatus(guild_id=1,
item_name="hello")
assert result.hidden == False
def test_checkShopitemStatus_failed(self):
# no item
result1 = InventoryUtil.checkShopitemStatus(guild_id=1,
item_name="hello")
assert result1 == -1
# no shopitem
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=10)
result2 = InventoryUtil.checkShopitemStatus(guild_id=1,
item_name="hello")
assert result2 == -2
def test_listHiddenitem_success(self):
result = InventoryUtil.listHiddenShopItem(guild_id=1)
assert len(result) == 0
ItemUtil.createItem(guild_id=1,
item_name="hello",
item_type=ItemType.ATTACK,
buff_type=BuffType.ATTACK,
buff_value=-1,
buff_round=2,
level_required=0,
price=1)
ItemUtil.createItem(guild_id=1,
item_name="hey",
item_type=ItemType.STATUS,
buff_type=BuffType.DEFENCE,
buff_value=10,
buff_round=3,
level_required=0,
price=1)
InventoryUtil.addItemToShop(guild_id=1, item_name="hello", amount=10)
InventoryUtil.addItemToShop(guild_id=1, item_name="hey", amount=10)
InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="hello",
hidden=True)
result = InventoryUtil.listHiddenShopItem(guild_id=1)
assert len(result) == 1
# items
InventoryUtil.changeShopitemHiddenStatus(guild_id=1,
item_name="hey",
hidden=True)
result = InventoryUtil.listHiddenShopItem(guild_id=1)
assert len(result) == 2
def setup_method(self, method):
self.database = KfpDb(":memory:")
class TestRoleUtil():
def setup_method(self, method):
self.database = KfpDb(":memory:")
pass
def teardown_method(self, method):
self.database.teardown()
pass
def test_updateRoleCreateNewOne(self):
role: KfpRole = RoleUtil.updateRole(123, 321, "testing", "0x000345")
assert role.guild_id == 123
assert role.role_id == 321
assert role.role_name == "testing"
assert role.color == "0x000345"
def test_updateRoleUpdateOldOne(self):
RoleUtil.updateRole(123, 321, "testing", "0x000345")
role: KfpRole = RoleUtil.updateRole(123, 321, "onetwothree",
"0x000543")
assert role.role_name == "onetwothree"
assert role.color == "0x000543"
def test_updateKfpRoleLevelZero(self):
RoleUtil.updateRole(123, 321, "testing", "0x000345")
role: KfpRole = RoleUtil.getRole(123, 321)
assert role.level == 0
def test_updateKfpRoleLevelSuccess(self):
RoleUtil.updateRole(123, 321, "testing", "0x000345")
role: KfpRole = RoleUtil.getRole(123, 321)
RoleUtil.updateKfpRoleLevel(role, 20)
assert role.level == 20
def test_getCurrentRolesEmpty(self):
roleList = RoleUtil.getCurrentRoles(123)
assert len(roleList) == 0
def test_getCurrentRolesSuccess(self):
role1 = RoleUtil.updateRole(123, 321, "testing", "0x000345")
role2 = RoleUtil.updateRole(123, 123, "onetwothree", "0x000543")
roleList = RoleUtil.getCurrentRoles(123)
assert len(roleList) == 2
assert roleList[0] == role1
assert roleList[1] == role2
def test_getCurrentRolesSuccess_level(self):
role1 = RoleUtil.updateRole(123, 100, "first", "0x000345",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role1, 0)
role2 = RoleUtil.updateRole(123, 101, "second", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role2, 20)
role3 = RoleUtil.updateRole(123, 102, "third", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role3, 40)
role4 = RoleUtil.updateRole(123, 103, "fourth", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role4, 60)
role5 = RoleUtil.updateRole(123, 104, "fifth", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role5, 80)
role6 = RoleUtil.updateRole(123, 105, "sixth", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role6, 100)
role = RoleUtil.getKfpRoleFromLevel(123, 80)
assert role == role5
def test_getRoleBeforeLevel(self):
role1 = RoleUtil.updateRole(123, 321, "testing", "0x000345",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role1, 10)
role2 = RoleUtil.updateRole(123, 123, "onetwothree", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role2, 20)
role = RoleUtil.getKfpRolesBeforeLevel(123, 9)
assert len(role) == 0
role = RoleUtil.getKfpRolesBeforeLevel(123, 11)
assert role[0] == role1
role = RoleUtil.getKfpRolesBeforeLevel(123, 20)
assert role[0] == role1
role = RoleUtil.getKfpRolesBeforeLevel(123, 21)
assert role[0] == role2
assert role[1] == role1
def test_getRoleFromLevel(self):
role0 = RoleUtil.updateRole(123, 100, "other roles", "0x000345")
RoleUtil.updateKfpRoleLevel(role0, 0)
role1 = RoleUtil.updateRole(123, 321, "testing", "0x000345",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role1, 10)
role2 = RoleUtil.updateRole(123, 123, "onetwothree", "0x000543",
Util.RoleCategory.KFP_DEFAULT)
RoleUtil.updateKfpRoleLevel(role2, 20)
role = RoleUtil.getKfpRoleFromLevel(123, 9)
assert not role
role = RoleUtil.getKfpRoleFromLevel(123, 11)
assert role == role1
role = RoleUtil.getKfpRoleFromLevel(123, 20)
assert role == role2