from lib.Ticket import Ticket
from .Feed import Feed
from .winning_structure import get_winnings_bb, get_winnings_no_bb


class TwoStep(Ticket):

    def __init__(self, my_picks, my_special_ball, mult):
        super().__init__(my_picks, my_special_ball)
        self.special_ball_name = 'Bonus Ball'
        self.feed = Feed()
        self.get_winning_numbers()
        self.multiplier = mult

    def get_winnings_with_special_ball(self, jp):
        return get_winnings_bb(self.jackpot)

    def get_winnings_without_special_ball(self):
        return get_winnings_no_bb()

    def get_winning_numbers(self):
        winners = self.feed.get_winning_numbers()
        self.winning_numbers = winners['numbers']
        self.special_ball = winners['special_ball']