import requests as req
import logging
from datetime import datetime
from time import sleep

# Input your thunderpick token here after you login (available in HTTP headers)
THUNDERPICK_TOKEN = ''

def calculate_tp_odds(bet_amount, total_amount):
    n = bet_amount/total_amount
    return max(1.01, round(0.93/n, 2))

def get_live_matches():
    res = req.get('https://thunderpick.com/api/matches?gameIds=1&gameIds=2&gameIds=3&gameIds=4&gameIds=6&gameIds=7&gameIds=8&gameIds=9&gameIds=19&gameIds=20&gameIds=21&gameIds=22&gameIds=23', headers = {'thunderpick-token': THUNDERPICK_TOKEN}).json()
    return [r for r in res['data'] if (
            (r['hasInPlayMarkets'] == True) and (
            ((r['isLive'] == False) and (r['matchBet']['timeToLock'] <= 300000) and (r['matchBet']['timeToLock'] > 0))
            or ((r['isLive'] == True) and (r['matchBet']['timeToLock'] > 0))
            ))]

def get_match_detail(match_id):
    return req.get('https://thunderpick.com/api/matches/%i' % match_id).json()['data']

def get_match_inplays(match_id):
    return req.get('https://thunderpick.com/api/in-plays?matchId=%i' % match_id).json()['data']

def str_to_datetime(string):
    return datetime.strptime(string[:24], '%Y-%m-%dT%H:%M:%S.%f')

def calculate_min_odds_from_favorite(o):
    return 1./(1-1./o)

def place_outplay_bet(marketId, choice, stake):
    payload = {'amount': int(stake),
                'betId': int(marketId),
                'value': choice }
    res = req.post('https://thunderpick.com/api/bet/mutual',
        json = payload, headers = {'thunderpick-token': THUNDERPICK_TOKEN, 'content-type': 'application/json;charset=UTF-8'})
    return res.json()

def place_inplay_bet(marketId, choice, stake, odds):
    payload = {'amount': int(stake),
                'marketId': int(marketId),
                'selectionId': choice,
                'selectionOdds': odds}
    res = req.post('https://thunderpick.com/api/in-plays',
        json = payload, headers = {'thunderpick-token': THUNDERPICK_TOKEN, 'content-type': 'application/json;charset=UTF-8'})
    return res.json()

def find_outplay_info_from_name(bet, name):
    for s in bet['buckets']:
        if s['label'] == name:
            return {'name': s['label'], 'marketId': bet['id'], 'value': s['value']}

def find_inplay_info_from_name(bet, name):
    for s in bet['selections']:
        if s['name'] == name:
            return {'name': s['name'], 'marketId': bet['id'], 'selectionId': s['id'], 'selectionOdds': s['odds']}

def test_token():
    res = req.get('https://thunderpick.com/api/user', headers = {'thunderpick-token': THUNDERPICK_TOKEN})
    if res.json()['ok'] == True:
        return True
    else:
        raise Exception('BAD TOKEN')

if __name__ == '__main__':
    logger = logging.getLogger('thunderpick_arb')
    logger.setLevel(logging.DEBUG)
    fh = logging.FileHandler('eggs.log')
    fh.setLevel(logging.INFO)
    ch = logging.StreamHandler()
    ch.setLevel(logging.INFO)
    logger.addHandler(fh)
    logger.addHandler(ch)

    while True:
        logger.info(test_token())
        logger.info('Fetching...')
        live_matches = get_live_matches()
        for match in live_matches:
            if len(match['matchBet']['buckets']) != 2:
                continue

            if match['matchBet']['private']['userAmount'] > 0:
                continue

            total_pool_amt = sum([bucket['amount'] for bucket in match['matchBet']['buckets']])
            outplay_slip = {team['label']:  calculate_tp_odds(team['amount'], total_pool_amt) for team in match['matchBet']['buckets']}
            logger.info('Found Out-play odds - %s' % outplay_slip)
            match_inplays = get_match_inplays(match['id'])
            match_winner_bet = [m for m in match_inplays['inPlays'] if (m['name'] == 'Match Up Winner')]
            if (match_winner_bet and match_winner_bet[0]['status'] == 1):
                inplay_slip = {team['name']: team['odds'] for team in match_winner_bet[0]['selections']}
                logger.info('Found In-play odds - %s - %s' % (match_winner_bet[0]['name'], inplay_slip))
                best_odds = [max(outplay_slip[team], inplay_slip[team]) for team in outplay_slip.keys()]
                if min(best_odds) > (calculate_min_odds_from_favorite(max(best_odds)) + 0.1):
                    logger.info('ARB Opportunity Found! %.2f vs %.2f' % (min(best_odds), calculate_min_odds_from_favorite(max(best_odds))))

                    # place bets
                    ratio = max(best_odds) / min(best_odds)
                    if ratio < 6:
                        for t,o in outplay_slip.items():
                            slip = find_outplay_info_from_name(match['matchBet'], t)
                            if o == min(best_odds):
                                amt = 100*ratio # lowest sizing for testing
                                break
                            elif o == max(best_odds):
                                amt = 100/ratio
                                break
                        logger.info((slip, amt))
                        res = place_outplay_bet(slip['marketId'], slip['value'], round(amt))
                        logger.info(res)
                        for t,o in inplay_slip.items():
                            slip = find_inplay_info_from_name(match_winner_bet[0], t)
                            amt = 100
                            if o == min(best_odds):
                                break
                            elif o == max(best_odds):
                                break
                        logger.info((slip, amt))
                        res = place_inplay_bet(slip['marketId'], slip['selectionId'], round(amt), slip['selectionOdds'])
                        logger.info(res)
                    else:
                        logger.info('Ratio way too high - skipping')

        logger.handlers[0].flush()
        sleep(60)