import time
import requests

class TwoCaptcha(object):
    def __init__(self, api_key, sleep_time=5):
        self.api_key = api_key
        self.sleep_time = sleep_time

    def __get_captcha_id(self, site_key, page_url):
        params = {
            'googlekey': site_key,
            'pageurl': page_url,
            'method': 'userrecaptcha',
            'key': self.api_key
        }
        endpoint = 'http://2captcha.com/in.php'
        response = requests.post(endpoint, params=params)
        captcha_id = response.text.split('|')[1]
        return captcha_id

    def __get_captcha_token(self, captcha_id):
        params = {
            'id': captcha_id,
            'action': 'get',
            'key': self.api_key
        }
        endpoint = 'http://2captcha.com/res.php'
        response = requests.get(endpoint, params=params)
        while 'CAPCHA_NOT_READY' in response.text:
            time.sleep(self.sleep_time)
            response = requests.get(endpoint, params=params)
        captcha_token = response.text.split('|')[1]
        return captcha_token

    def solve_captcha(self, site_key, page_url):
        captcha_id = self.__get_captcha_id(site_key, page_url)
        captcha_token = self.__get_captcha_token(captcha_id)
        return captcha_token

    def get_balance(self):
        params = {
            'action': 'getbalance',
            'json': 1,
            'key': self.api_key
        }
        endpoint = 'http://2captcha.com/res.php'
        response = requests.get(endpoint, params=params)
        response_json = response.json()

        balance = None
        if response_json.get('status', 0) == 1:
            balance = response_json.get('request', None)

        if balance:
            return balance
        else:
            raise ValueError('2Captcha returned invalid balance')