__init__.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
# TODO use swagger based autogenerated library for implementing these functions
# TODO use colors for heading and contests
from ...models import Judge
from ...utils.logging import logger
import click
import requests
import json
from .utils import get_data, login_oauth, login_web
from beautifultable import BeautifulTable
import os

ALLOW_WEB = False


class Codechef(Judge):
    def __init__(self, session_data=None):
        # Init should not have any network requests
        # do them in login, logout, check_running_contest
        logger.debug("Initializing class Codechef with session_data:\n%s"
                     % session_data)
        self.name = "Codechef"
        self.url = "https://www.codechef.com"
        self.api_url = "https://api.codechef.com"

        # In case main api is down, use the env CODECHEF_WEB
        # as a feature switch to use the browser based Non
        # authenticated API. This is very helpful for testing
        self.CODECHEF_WEB = (
            'TERMICODER_CODECHEF_WEB' in os.environ and
            os.environ['TERMICODER_CODECHEF_WEB'].lower() == 'true'
            and ALLOW_WEB)
        if(self.CODECHEF_WEB):
            self.api_url = "https://www.codechef.com/api"

        self.session_data = session_data
        if(session_data is not None):
            self._update_session()

    def check_login(self):
        logger.debug("Checking Login")
        if(self.session is None):
            logger.debug("No session object initialized")
            return False
        path = 'user/me' if self.CODECHEF_WEB else 'users/me'
        me_url = self._make_url(path)
        r = self._request_api(me_url)

    def login(self):
        if self.CODECHEF_WEB:
            token = login_web(self)
        else:
            token = login_oauth()
        self.session_data = token
        self._update_session()
        self.check_login()

    def logout(self):
        logger.warning("Logout of CodeChef.")
        click.confirm("Are you sure?", default=True, abort=True)
        self.session_data = None

    def get_running_contests(self):
        logger.debug('get running contests')
        contests = get_data.running_contests(self)
        table = BeautifulTable()
        table.width_exceed_policy = BeautifulTable.WEP_WRAP
        # TODO: use map style.headers instead of str
        # requires change with beautifultable. we may try dev version
        table.column_headers = list(
            map(str, ['code', 'name', 'end', 'start']))
        for contest in contests:
            table.append_row(
                [
                    contest['code'], contest['name'],
                    str(contest['startDate']), str(contest['endDate'])
                ]
            )
        return table

    # This method serves both as a problem getter as well as kind of factory
    # for problem
    def get_problem(self, problem_name, contest_name, problem_data=None):
        # If problem data is passed, it should take precedence
        # Method should call the respective Problem.__init__ method to create a
        # problem instance and return it
        raise NotImplementedError

    def get_contest(self, contest_name, contest_data=None):
        # If contest data is passed, it should take precedence
        # Method should call the respective Problem.__init__ method to create a
        # problem instance and return it
        pass

    def _update_session(self):
        self.session = requests.Session()

        def debug_url(r, *args, **kwargs):
            logger.debug('Getting url %s' % r.url)

        def debug_data(r, *args, **kwargs):
            try:
                response = json.dumps(r.json(), indent=1)
            except json.JSONDecodeError:
                response = r.text
            logger.debug('Response %s' % response)

        self.session.hooks['response'].append(debug_url)
        self.session.hooks['response'].append(debug_data)
        if(self.CODECHEF_WEB and 'cookies' in self.session_data):
            logger.debug(self.session_data['cookies'])
            self.session.cookies = self.session_data['cookies']
        elif('data' in self.session_data):
            logger.debug('Token: ' + self.session_data['data']['access_token'])
            OAuth2_Header = {
                'Authorization': 'Bearer ' +
                self.session_data['data']['access_token']
            }
            self.session.headers.update(OAuth2_Header)

    def _make_url(self, rel_url):
        rel_url = rel_url.strip('/')
        api_url = self.api_url.strip('/')
        return "/".join([api_url, rel_url])

    def _request_api(self, url):
        logger.debug('fetching url %s' % url)
        with self.session as s:
            r = s.get(url)
            logger.debug(r)
            r.raise_for_status()
        return r.json()

    def _refresh_token(self):
        logger.debug('refreshing token')
        url = 'http://termicoder.diveshuttam.me/refresh_token/'
        raise NotImplementedError
        # TODO implement this on server side
        r = requests.get(url, data=self.session_data)
        logger.debug(r.json())