class BetsManager(object): def __init__(self, session): self.session = session self.games = NFLGameManager(self.session) self.teams = NFLTeamManager(self.session) self.odds = NFLOddsManager(self.session) self.accounts = AccountingManager(self.session) # FIXME: put this in config or database self.max_bet = 500 def query(self): return self.session.query(UserBet) def get(self, bet_id): return self.query().get(bet_id) def get_current_bet(self, user_id): q = self.session.query(CurrentBet) return q.get(user_id) def _get_odds(self, game_id): odds = self.odds.get_odds(game_id) return odds def _copy_current_odds(self, bet, odds): for field in ['favored_id', 'underdog_id', 'total', 'spread']: setattr(bet, field, getattr(odds, field)) def _check_amount(self, user_id, amount): if amount % 10 != 0: raise BetNotInTensError, "Bad amount %d" % amount if amount < 10: raise MinimumBetError, "Bet must be at least ten: %d" % amount if amount > self.max_bet: msg = "Bet must be no more than %d: amount %d" raise MaximumBetError, msg % (self.max_bet, amount) acct = self.accounts.get(user_id) balance = acct.balance.balance max_bet = determine_max_bet(balance) if amount > max_bet: juice_insurance = amount / 10 total = amount + juice_insurance msg = "Total needed %d, current balance %d" % (total, balance) raise InsufficientFundsError, msg return True def check_bet_amount(self, user_id, amount): return self._check_amount(user_id, amount) # pick is either an NFLTeam object, 'over', or 'under' def request_bet(self, user_id, game_id, amount, bettype, pick): # here we sanitize the bettype and # amount arguments if bettype not in ['line', 'underover']: raise RuntimeError, "Bad bettype %s:" % bettype # check amount # self.accounts.check_bet_amount(user_id, amount) self._check_amount(user_id, amount) if self.get_current_bet(user_id) is not None: raise UnconfirmedBetExistsError, "Betting in progress" with transaction.manager: now = datetime.now() curbet = CurrentBet() curbet.user_id = user_id curbet.game_id = game_id curbet.created = now curbet.amount = amount curbet.bet_type = bettype if bettype == 'line': # the pick is a team curbet.team_id = pick.id else: # the pick is under/over curbet.underover = pick self.session.add(curbet) return self.session.merge(curbet) def show_requested_bet(self, user_id): current = self.get_current_bet(user_id) if current is None: raise NoCurrentBetError, "No current bet" odds = self._get_odds(current.game_id) odata = self.odds.get_data(odds) return current, odata def cancel_requested_bet(self, user_id): with transaction.manager: q = self.session.query(CurrentBet) q = q.filter_by(user_id=user_id) q.delete() def place_requested_bet(self, user_id): current, odata = self.show_requested_bet(user_id) game_id = current.game_id amount = current.amount bettype = current.bet_type underover = current.underover team_id = current.team_id account = self.accounts.get(user_id) txn = self.accounts.place_bet(account.id, amount) with transaction.manager: now = datetime.now() bet = UserBet() bet.user_id = user_id bet.game_id = current.game_id bet.txn_id = txn.id bet.created = current.created bet.amount = current.amount bet.bet_type = current.bet_type bet.underover = current.underover bet.team_id = current.team_id for field in odata: setattr(bet, field, odata[field]) self.session.add(bet) dq = self.session.query(CurrentBet).filter_by(user_id=user_id) dq.delete() return self.session.merge(bet) def get_bets(self, user_id=None, game_id=None): pass def get_game_bets(self, game_id): q = self.query() q = q.filter_by(game_id=game_id) return q.all() def get_user_bets(self, user_id, game_id=None): q = self.query() q = q.filter_by(user_id=user_id) if game_id is not None: q = q.filter_by(game_id=game_id) return q.all() def get_user_closed_bets(self, user_id, game_id=None): q = self.session.query(ClosedBet) q = q.filter_by(user_id=user_id) if game_id is not None: q = q.filter_by(game_id=game_id) return q.all() def _get_score(self, bet): "returns score as favored, underdog" game = bet.game score = game.score if bet.favored_id == game.away_id: favored_score = score.away_score elif bet.favored_id == game.home_id: favored_score = score.home_score else: raise RuntimeError, "Team/score mismatch game %d" % game.id if bet.underdog_id == game.away_id: underdog_score = score.away_score elif bet.underdog_id == game.home_id: underdog_score = score.home_score else: raise RuntimeError, "Team/score mismatch game %d" % game.id return favored_score, underdog_score def _determine_bet(self, bet_id): bet = self.query().get(bet_id) if bet is None: raise RuntimeError, "Bad bet id %d" % bet_id if bet.game.score is None: return fscore, uscore = self._get_score(bet) result = determine_bet(bet, fscore, uscore) acct = self.accounts.get(bet.user_id) if result == 'push': txn = self.accounts.push_bet(acct.id, bet.amount) elif result == 'win': txn = self.accounts.win_bet(acct.id, bet.amount) elif result == 'lose': txn = self.accounts.lose_bet(bet.amount) else: raise RuntimeError, "bad result %s" % result with transaction.manager: cb = make_closed_bet(bet, result, txn) self.session.add(cb) self.session.delete(bet) return self.session.merge(cb) def determine_bet(self, bet_id): return self._determine_bet(bet_id) def determine_bets(self): closed = list() for bet in self.get_all_bets(): cb = self.determine_bet(bet.id) closed.append(cb) return closed def get_all_bets(self): return self.query().all() def get_all_closed_bets(self): return self.session.query(ClosedBet).all()