async def birth(self, ctx):
message = ctx.message
user_id = Users.get_or_none((Users.discord_id == message.author.id) & (Users.server_id == message.guild.id)).id;
if user_id is not None:
num_egg = random.randint(0,10)
colony_id = Colony.get(Colony.user == user_id).id
#TODO: logic to assign role
eggs = [(colony_id, names.get_full_name(), AntRole.WORKER, 0) for x in range(num_egg)]
Ants.insert_many(eggs, fields=[Ants.colony, Ants.name, Ants.role, Ants.life_stage]).execute()
await message.channel.send("You have spawned %s egg(s)." % num_egg)
else:
await message.channel.send("No queen found.")
def login(self):
self.private_data["error_msg"] = None
if web.ctx.method == "GET":
return self.display("admin/login")
else:
inputs = self.get_input()
username = inputs.get("username", None)
password = inputs.get("password", None)
res = (not username) or (not password)
if res:
self.private_data["error_msg"] = "用户名和密码不能为空！"
return self.display("admin/login")
user = Users.get_or_none(Users.name == username)
if not user:
self.private_data["error_msg"] = "用户名或密码错误！"
return self.display("admin/login")
if user.password != admin_pwd_digest(password):
self.private_data["error_msg"] = "用户名或密码错误！"
return self.display("admin/login")
self.set_login(username, user.id)
self.private_data['user'] = user
return web.seeother(self.make_url('home'))
