main.py

import CodeChefScraper
import CodeForcesScraper
import AtcoderScraper
from time import sleep
import logging
from github import Github
from github.GithubException import UnknownObjectException
from UploadToGithub import upload_to_github
import inspect
# from multiprocessing import Process

EXTENSIONS = {
    'c++': 'cpp',
    'clang': 'cpp',
    'gcc': 'c',
    'py': 'py',
    'javascript': 'js',
    'java': 'java',
    'c#': 'cs',
    'go': 'go',
    'haskell': 'hs',
    'kotlin': 'kt',
    'delphi': 'dpr',
    'pascal': 'pas',
    'perl': 'pl',
    'php': 'php',
    'rust': 'rs',
    'scala': 'sc',
    'node': 'js',
}


def upload_solution(website, solution, repo):
    try:
        s = solution["language"].lower()

        extension = 'txt'
        for key, value in EXTENSIONS.items():
            if key in s:
                extension = value
                break

        if solution.get('problem_name', ''):
            path = f'{website}/{solution["language"]}/{solution["problem_code"]} | {solution["problem_name"]}/{solution["solution_id"]}.{extension}'
        else:
            path = f'{website}/{solution["language"]}/{solution["problem_code"]}/{solution["solution_id"]}.{extension}'

        problem_info = ''
        if solution.get('problem_link', ''):
            if not solution.get('problem_name', ''):
                problem_info = f"""
                # This text file was auto-generated by UpCode

                Link: {solution['problem_link']}
                """

            else:
                problem_info = f"""
                # This text file was auto-generated by UpCode

                Problem: {solution['problem_name']}
                Link: {solution['problem_link']}
                """

        upload_to_github(repo, path, solution['solution'], inspect.cleandoc(problem_info))
        return True

    except Exception as e:
        logging.error(f'{e} FOR {solution}')
        return False


def codeforces_uploader(codeforces_username, repo):
    failed_codeforces = []
    for solution in CodeForcesScraper.get_solutions(codeforces_username):
        if not upload_solution('CodeForces', solution, repo):
            failed_codeforces.append(solution)

    for _ in range(3):
        if failed_codeforces:
            sleep(180)

            new_failed_codeforces = []
            for solution in CodeForcesScraper.get_solutions(codeforces_username, failed_codeforces):
                if not upload_solution('CodeForces', solution, repo):
                    new_failed_codeforces.append(solution)

            failed_codeforces = new_failed_codeforces


def codechef_uploader(codechef_username, repo):
    for solution in CodeChefScraper.get_solutions(codechef_username):
        upload_solution('CodeChef', solution, repo)


def atcoder_uploader(atcoder_username, repo):
    for solution in AtcoderScraper.get_solutions(atcoder_username):
        upload_solution('Atcoder', solution, repo)


def main():
    codeforces_username = input('Enter codeforces username (Press enter if N/A): ')
    codechef_username = input('Enter codechef username (Press enter if N/A): ')
    atcoder_username = input('Enter atcoder username (Press enter if N/A): ')
    access_token = input('Enter github access token: ')

    repo_name = input('Enter repository name (Press enter to use "CP-Solutions"): ')
    if repo_name.isspace() or not repo_name:
        repo_name = 'CP-Solutions'

    g = Github(access_token)

    try:
        repo = g.get_user().get_repo(repo_name)

    except UnknownObjectException:
        repo = g.get_user().create_repo(repo_name, private=True)

    if atcoder_username:
        atcoder_uploader(atcoder_username, repo)
        # atcoder_process = Process(target=atcoder_uploader, args=(atcoder_username, repo))
        # atcoder_process.start()

    if codechef_username:
        codechef_uploader(codechef_username, repo)
        # codechef_process = Process(target=codechef_uploader, args=(codechef_username, repo))
        # codechef_process.start()

    if codeforces_username:
        codeforces_uploader(codeforces_username, repo)
        # codeforces_process = Process(target=codeforces_uploader, args=(codeforces_username, repo))
        # codeforces_process.start()


if __name__ == '__main__':
    main()