forked from MarseyWorld/MarseyWorld
128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
import time
|
|
from random import choice
|
|
from sqlalchemy import *
|
|
from files.helpers.alerts import *
|
|
from files.helpers.wrappers import *
|
|
from files.helpers.actions import badge_grant
|
|
from flask import g
|
|
from .const import *
|
|
|
|
LOTTERY_WINNER_BADGE_ID = 137
|
|
|
|
def get_active_lottery():
|
|
return g.db.query(Lottery).order_by(Lottery.id.desc()).filter(Lottery.is_active).one_or_none()
|
|
|
|
|
|
def get_users_participating_in_lottery():
|
|
return g.db.query(User) \
|
|
.filter(User.currently_held_lottery_tickets > 0) \
|
|
.order_by(User.currently_held_lottery_tickets.desc()).all()
|
|
|
|
|
|
def get_active_lottery_stats():
|
|
active_lottery = get_active_lottery()
|
|
participating_users = get_users_participating_in_lottery()
|
|
|
|
return None if active_lottery is None else active_lottery.stats, len(participating_users)
|
|
|
|
|
|
def end_lottery_session():
|
|
active_lottery = get_active_lottery()
|
|
|
|
if (active_lottery is None):
|
|
return False, "There is no active lottery."
|
|
|
|
participating_users = get_users_participating_in_lottery()
|
|
raffle = []
|
|
for user in participating_users:
|
|
for _ in range(user.currently_held_lottery_tickets):
|
|
raffle.append(user.id)
|
|
|
|
winner = choice(raffle)
|
|
active_lottery.winner_id = winner
|
|
winning_user = next(filter(lambda x: x.id == winner, participating_users))
|
|
winning_user.coins += active_lottery.prize
|
|
winning_user.total_lottery_winnings += active_lottery.prize
|
|
badge_grant(winner, LOTTERY_WINNER_BADGE_ID)
|
|
|
|
for user in participating_users:
|
|
chance_to_win = user.currently_held_lottery_tickets / len(raffle) * 100
|
|
if user.id == winner:
|
|
notification_text = f'You won {active_lottery.prize} dramacoins in the lottery! ' \
|
|
+ f'Congratulations!\nOdds of winning: {chance_to_win}%'
|
|
else:
|
|
notification_text = f'You did not win the lottery. Better luck next time!\n' \
|
|
+ f'Odds of winning: {chance_to_win}%\nWinner: @{winning_user.username}'
|
|
send_repeatable_notification(user.id, notification_text)
|
|
user.currently_held_lottery_tickets = 0
|
|
|
|
active_lottery.is_active = False
|
|
|
|
g.db.commit()
|
|
|
|
return True, f'{winning_user.username} won {active_lottery.prize} dramacoins!'
|
|
|
|
|
|
def start_new_lottery_session():
|
|
end_lottery_session()
|
|
|
|
lottery = Lottery()
|
|
epoch_time = int(time.time())
|
|
# Subtract 4 minutes from one week so cronjob interval doesn't cause the
|
|
# time to drift toward over multiple weeks.
|
|
one_week_from_now = epoch_time + 60 * 60 * 24 * 7 - (4 * 60)
|
|
lottery.ends_at = one_week_from_now
|
|
lottery.is_active = True
|
|
|
|
g.db.add(lottery)
|
|
g.db.commit()
|
|
|
|
|
|
def check_if_end_lottery_task():
|
|
active_lottery = get_active_lottery()
|
|
|
|
if active_lottery is None:
|
|
return False
|
|
elif active_lottery.timeleft > 0:
|
|
return False
|
|
|
|
start_new_lottery_session()
|
|
return True
|
|
|
|
def lottery_ticket_net_value():
|
|
return LOTTERY_TICKET_COST - LOTTERY_SINK_RATE
|
|
|
|
def purchase_lottery_tickets(v, quantity=1):
|
|
if quantity < 1:
|
|
return False, "Must purchase one or more lottery tickets."
|
|
elif (v.coins < LOTTERY_TICKET_COST * quantity):
|
|
return False, f'Lottery tickets cost {LOTTERY_TICKET_COST} dramacoins each.'
|
|
|
|
most_recent_lottery = get_active_lottery()
|
|
if (most_recent_lottery is None):
|
|
return False, "There is no active lottery."
|
|
|
|
v.coins -= LOTTERY_TICKET_COST * quantity
|
|
v.currently_held_lottery_tickets += quantity
|
|
v.total_held_lottery_tickets += quantity
|
|
|
|
net_ticket_value = lottery_ticket_net_value() * quantity
|
|
most_recent_lottery.prize += net_ticket_value
|
|
most_recent_lottery.tickets_sold += quantity
|
|
|
|
g.db.commit()
|
|
|
|
return True, f'Successfully purchased {quantity} lottery tickets!'
|
|
|
|
def grant_lottery_tickets_to_user(v, quantity):
|
|
active_lottery = get_active_lottery()
|
|
prize_value = lottery_ticket_net_value() * quantity
|
|
|
|
if active_lottery:
|
|
v.currently_held_lottery_tickets += quantity
|
|
v.total_held_lottery_tickets += quantity
|
|
|
|
active_lottery.prize += prize_value
|
|
active_lottery.tickets_sold += quantity
|
|
|
|
g.db.commit() |