Codeforces.py

import hashlib
import json
import random
import time
from collections import OrderedDict
from urllib.parse import urlencode

from ReqHandler import ReqHandler


def generate_random_number(length=6):
    letters = [chr(ord('0') + x) for x in range(10)]
    result_str = ''.join(random.choice(letters) for i in range(length))
    return result_str


class CodeforcesApi:
    req = ReqHandler()

    def __init__(self):
        with open("keys.txt", "r") as f:
            self.keys = json.loads(f.read())

    def request(self, method_name, params):
        params["apiKey"] = self.keys["apiKey"]
        params["time"] = int(time.time())
        params = {k: v for k, v in sorted(params.items(), key=lambda item: (item[0], item[1]))}
        random_string = generate_random_number()
        data = random_string + "/" + method_name + "?" + urlencode(params) + "#" + self.keys["apiSecret"]
        params['apiSig'] = random_string + hashlib.sha512(data.encode('utf-8')).hexdigest()
        result = json.loads(self.req.request("https://codeforces.com/api/" + method_name, parameters=params))
        if result["status"] != "OK":
            raise Exception("return status is not ok")
        return result['result']


cfApi = CodeforcesApi()
'''
sample input:
291445
100
'''

contestId = input()
total_score = int(input())
result = cfApi.request("contest.standings",
                       {"contestId": contestId, "showUnofficial": "false"})

problem_score = total_score / len(result['problems'])

users = {}
for user in result['rows']:
    users[user['party']['members'][0]['handle']] = {"rank": user['rank'], 'total_points': user['points'] * problem_score}
    users[user['party']['members'][0]['handle']]["problemResults"] = []
    for pres in user["problemResults"]:
        users[user['party']['members'][0]['handle']]["problemResults"].append(pres['points']*problem_score)
print(users)