async def scramble(ctx):
guild = ctx.message.guild
logging.info(guild)
players = models.Player_list()
players.load_all()
for p in players.items:
old_team_id = p.team_id
# Pick a number & update player
team_id = randint(1, 3)
if old_team_id != team_id:
p.team_id = team_id
p.update()
old_team = models.Team()
old_team.load(old_team_id)
team = models.Team()
team.load(team_id)
# Clear/Assign role
old_team_role = get(guild.roles, name=old_team.name)
team_role = get(guild.roles, name=team.name)
member = guild.get_member(int(p.discord_id))
if member is None:
logging.info(
"Player w/ id {} not found. Maybe they left the server.".
format(p.discord_id))
continue
await member.remove_roles(old_team_role,
reason="Automated Team Scramble")
await member.add_roles(team_role, reason="Automated Team Scramble")
logging.info("{} (ID:{}) moved from {} to {}".format(
member.nick, p.discord_id, old_team.name, team.name))
await asyncio.sleep(1)
models.save()
logging.info("Scramble complete")
def pay_players():
teams = models.Team_list()
teams.custom_load("team_id > ?", (0, ))
for t in teams.items:
# load all the players
players = models.Player_list()
players.custom_load("team_id = ?",
(t.team_id, )) # Still the dumb Loadall HACK
total_team_shares = sum([p.shares for p in players.items])
share_value = config['daily_team_coins'] // total_team_shares
# pay 'em
for player in players.items:
# add daily salary
player.coins += share_value * player.shares
# get THIS PLAYER's payments from TODAY for THIS PLAYER's TEAM that
# WEREN'T put toward the current location (for refunds)
models.c.execute(
"SELECT SUM(amount) FROM Payment p LEFT JOIN LocationEdge le ON(p.location_edge = le.edge_id) WHERE player_id = ? AND team_id = ? AND time = ? AND end_location_id != ?",
(player.player_id, player.team_id, helpers.get_game_day() - 1,
t.current_location_id))
refund = models.c.fetchone()[0]
if refund is None:
refund = 0
player.coins += int(refund * (config['refund_percentage'] / 100))
player.update()
models.save()
def update_shares():
players = models.Player_list()
players.load_all()
for player in players.items:
days_since_active = helpers.get_game_day() - player.last_active_day
if days_since_active < len(config['afk_falloff']):
player.shares = config['afk_falloff'][days_since_active]
else:
player.shares = 0
player.update()
models.save()
async def validateteams(ctx):
bad = 0
guild = ctx.message.guild
logging.info(guild)
players = models.Player_list()
players.load_all()
for p in players.items:
member = guild.get_member(int(p.discord_id))
t = models.Team()
t.load(p.team_id)
if member is None:
logging.info(
"Player w/ id {} not found. Maybe they left the server.".
format(p.discord_id))
else:
if member.top_role.name != t.name:
logging.error(
"{} (ID:{})'s role is {}, but it should be {}".format(
member.nick, p.discord_id, member.top_role.name,
t.name))
bad += 1
logging.info("Validation done. {}/{} roles are correct.".format(
len(players.items) - bad, len(players.items)))
async def startgame(ctx):
logging.info("The king has signalled to begin the game!")
# Reset every player's coins
logging.info("Resetting player coins and last_active_day...")
ps = models.Player_list()
ps.custom_load("player_id > ?", (0, )) # HACK to select all
for p in ps.items:
p.coins = config["starting_coins"]
p.last_active_day = 0
p.update()
# Reset every team's starting location
logging.info("Resetting team locations...")
ts = models.Team_list()
ts.custom_load("team_id > ?", (0, )) # SAME HACK to select all
for t in ts.items:
t.current_location_id = 1
t.update()
# Set time to -1 for all payments (i'm sorry reality)
logging.info("Invalidating previous payments...")
pmts = models.Payment_list()
pmts.custom_load("time >= ?", (0, ))
for p in pmts.items:
p.time = -1
p.update()
models.save()
# Update config
config['game_ongoing'] = 1
d = datetime.now()
d = d.replace(minute=0, second=0, microsecond=0)
config['start_date'] = str(d)
with open('config.json', 'w') as f:
json.dump(config, f)
scheduledjobs.config = config
helpers.config = config
scheduledjobs.on_new_day()