import feedparser


class Feed:

    def __init__(self, feed):
        self.feed = 'https://www.texaslottery.com/export/sites/lottery/rss/tlc_latest.xml'
        self.parsed_feed = feedparser.parse(self.feed)
        self.game_entries = feed
        self.modifiers = {
            'Billion': 1000000000,
            'Million': 1000000
        }

    def get_winning_numbers(self):
        entry = self.game_entries
        nums_pre = self.parsed_feed.entries[entry['winning_numbers']].summary
        nums_pre = nums_pre.replace(entry['special_ball'], '-').replace(f' {entry["multiplier"]}', '-')
        nums = nums_pre.split(' - ')

        winning_numbers = nums[:5]

        for idx, num in enumerate(nums[:5]):
            winning_numbers[idx] = int(num)

        special_ball = int(nums[5])
        multiplier = int(nums[6])
        return {'numbers': winning_numbers, 'special_ball': special_ball, 'multiplier': multiplier}

    def get_jackpot(self):
        entry = self.game_entries
        jp_pre = self.parsed_feed.entries[entry['jackpot']].summary
        jp_pre = jp_pre[jp_pre.index('Cash'):]
        jp_pre = jp_pre[jp_pre.index(':') + 3:]
        jp_data = jp_pre.split()
        return float(jp_data[0]) * self.modifiers[jp_data[1]]