Add code

Author: NeverMendel

.gitignore

@@ -127,3 +127,7 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# IDE folders
+.vscode
+.idea

Dockerfile

@@ -0,0 +1,19 @@
+FROM alpine:latest
+
+RUN apk add --no-cache \
+      nss \
+      ca-certificates \
+      python3 \
+      py3-pip
+
+# Copy current directory to /usr/src/app
+ADD . /usr/src/app
+WORKDIR /usr/src/app
+
+# Create out folder
+RUN mkdir -p out
+
+# Install dependencies
+RUN pip3 install -r requirements.txt
+
+ENTRYPOINT ["python3", "./main.py", "--folder", "out"]

LICENSE

@@ -186,7 +186,7 @@
       same "printed page" as the copyright notice for easier
       identification within third-party archives.
 
-   Copyright [yyyy] [name of copyright owner]
+   Copyright 2022 Davide Cazzin
 
    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.

README.md

@@ -1,2 +1,52 @@
-# leetcode-export
-Python script and Docker image to export your LeetCode solutions
+# LeetCode Export
+
+> Python script and Docker image to export your LeetCode solutions
+
+## DISCLAIMER
+
+All the problems hosted on leetcode.com are intellectual propriety of LeetCode, LLC. **DO NOT UPLOAD
+THE DESCRIPTION OF LEETCODE PROBLEMS ON GITHUB OR ON ANY OTHER WEBSITE** or you might receive ad DMCA Takedown notice.
+
+Read [LeetCode Terms of Service here](https://leetcode.com/terms/).
+
+To avoid committing the problem description on git, you can add `*.txt` to your `.gitignore` file.
+
+## How it works
+
+This script uses `selenium` and LeetCode APIs to download all your LeetCode submitted solutions.
+
+Before running the script, make sure that python3, chrome and `chromedriver` are installed in your system.
+
+If you do not want to configure and install all the required dependencies, you can download the Docker image. For
+further instructions read the section [Docker Image here](#docker-image).
+
+`chromedriver` is used to get the cookies needed to download your LeetCode submissions. Alternatively, you can provide
+the cookies using the flag `--cookies` in the Python script.
+
+## How to use
+
+### Run locally
+
+First install all the needed dependencies by executing:
+
+```bash
+pip install -r requirements.txt
+```
+
+### Docker Image
+
+To use the Docker Image you first need to download it from the Docker repository by running:
+
+```bash
+docker pull nevermendel/leetcode-export
+```
+
+Now you can download all your LeetCode submission in the current folder by executing:
+
+```bash
+docker run -it -v $(pwd):/usr/app/out --rm nevermendel/leetcode-export
+```
+
+## License
+
+[Apache License 2.0](LICENSE)

leetcode.py

@@ -0,0 +1,128 @@
+from dataclasses import dataclass
+from time import sleep
+from typing import Dict, List
+
+import requests
+from dataclasses_json import dataclass_json
+from selenium import webdriver
+
+LOGIN_URL = 'https://leetcode.com/accounts/login/'
+SUBMISSIONS_API_URL = 'https://leetcode.com/api/submissions/?offset={}&limit={}'
+PROBLEM_URL = 'https://leetcode.com/problems/'
+GRAPHQL_URL = 'https://leetcode.com/graphql'
+
+FILE_EXTENSION = {
+    "python": 'py',
+    "python3": 'py',
+    "c": 'c',
+    "cpp": 'cpp',
+    "csharp": 'cs',
+    "java": 'java',
+    "kotlin": 'kt',
+    "mysql": 'sql',
+    "mssql": 'sql',
+    "oraclesql": 'sql',
+    "javascript": 'js',
+    "html": 'html',
+    "php": 'php',
+    "golang": 'go',
+    "scala": 'scala',
+    "pythonml": 'py',
+    "rust": 'rs',
+    "ruby": 'rb',
+    "bash": 'sh',
+    "swift": 'swift'
+}
+
+
+@dataclass_json
+@dataclass
+class Submission:
+    id: int
+    lang: str
+    time: str
+    timestamp: int
+    status_display: str
+    runtime: str
+    url: str
+    is_pending: str
+    title: str
+    memory: str
+    code: str
+    compare_result: str
+    title_slug: str
+
+
+@dataclass
+class Problem:
+    problem_id: int
+    title: str
+    slug: str
+    difficulty: str
+    description: str
+    test_cases: str
+
+
+def question_data(slug):
+    return {
+        "operationName": "questionData",
+        "variables": {
+            "titleSlug": slug
+        },
+        "query": """query questionData($titleSlug: String!) {
+            question(titleSlug: $titleSlug) {
+                questionId
+                questionFrontendId
+                boundTopicId
+                title
+                titleSlug
+                content
+                difficulty
+                sampleTestCase
+            }
+        }"""
+    }
+
+
+def get_problem_info(cookies: str, slug: str):
+    response = requests.post(
+        GRAPHQL_URL,
+        json=question_data(slug),
+        headers={'Cookie': cookies})
+    return response
+
+
+def valid_cookies(cookies: str) -> bool:
+    return True
+
+
+def get_cookies(username: str, password: str) -> str:
+    driver = webdriver.Chrome()
+    driver.get(LOGIN_URL)
+
+    return ''
+
+
+def get_submissions(cookies: str) -> Dict[str, List[Submission]]:
+    dictionary: Dict[str, List[Submission]] = {}
+    batch_size = 20
+    offset = 0
+    while True:
+        print(f"getting batch #{offset + 1}")
+        response = requests.get(
+            SUBMISSIONS_API_URL.format(offset, batch_size),
+            headers={'Cookie': cookies})
+        json_response = response.json()
+        if 'detail' in json_response:
+            print(json_response['detail'])
+        if 'submissions_dump' in json_response:
+            for submission_dict in json_response['submissions_dump']:
+                submission = Submission.from_dict(submission_dict)
+                if submission.title_slug not in dictionary:
+                    dictionary[submission.title_slug] = []
+                dictionary[submission.title_slug].append(submission)
+        if 'has_next' not in json_response or not json_response['has_next']:
+            break
+        offset += batch_size
+        sleep(2)
+    return dictionary

main.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+import argparse
+import os.path
+from string import Template
+
+import leetcode
+
+PROBLEM_INFO_TEMPLATE = Template('''${problem_id} - ${title}
+${difficulty} - https://leetcode.com/problems/${slug}/"
+
+${description}
+
+${test_cases}
+''')
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Export LeetCode solutions')
+    parser.add_argument('--username', type=str, required=False, help='LeetCode username')
+    parser.add_argument('--password', type=str, required=False, help='LeetCode password')
+    parser.add_argument('--folder', type=str, required=False, help='Output folder', default='out')
+    parser.add_argument('--cookies', type=str, required=False, help='LeetCode cookies')
+
+    return parser.parse_args()
+
+
+if __name__ == '__main__':
+    args = parse_args()
+
+    cookies = args.cookies
+
+    print(leetcode.get_submissions(cookies))
+    # if not cookies or not leetcode.valid_cookies(cookies):
+    #     if not args.username:
+    #         print("Insert LeetCode username: ")
+    #         args.username = input()
+    #
+    #     if not args.password:
+    #         print("Insert LeetCode username: ")
+    #         args.password = input()
+    #
+    #     if not os.path.exists(args.folder):
+    #         os.mkdir(args.folder)
+    #     os.chdir(args.folder)
+    #     cookies = leetcode.get_cookies(args.username, args.password)
+    #
+    # solved_problems = leetcode.get_solved_problem_slugs(cookies)
+    #
+    # for slug in solved_problems:
+    #     if not os.path.exists(slug):
+    #         os.mkdir(slug)
+    #     os.chdir(slug)
+    #     problem_info = leetcode.get_problem(cookies, slug)
+    #     info_filename = f"{problem_info.slug}.txt"
+    #     if not os.path.exists(info_filename):
+    #         info_file = open(info_filename, 'w+')
+    #         info_file.write(PROBLEM_INFO_TEMPLATE.substitute(**problem_info.__dict__))
+    #         info_file.close()
+    #
+    #     submissions = leetcode.get_submissions(cookies, slug)
+    #
+    #     for sub in submissions:
+    #         sub_filename = f"{sub.submission_date.strftime('%Y-%m-%d %H-%M-%S')} - {sub.status} - runtime: {sub.runtime} - memory: {sub.memory}.{sub.extension}"
+    #         if not os.path.exists(sub_filename):
+    #             sub_file = open(sub_filename, 'w+')
+    #             sub_file.write(sub.code)
+    #             sub_file.close()

requirements.txt

@@ -0,0 +1,3 @@
+selenium
+dataclasses_json
+requests