escapefromtarkov.py

import json
import requests
import zlib
from functools import wraps
from urllib.parse import urlencode


class ApiResponse:
    def __init__(self, response: dict):
        self.data = response['data']
        self.errmsg = response['errmsg']
        self.err = response['err']

    def ok(self) -> bool:
        return self.err == 0

    def __repr__(self) -> str:
        return "<ApiResponse [{}]>".format("ok" if self.ok() else self.err)


def get(url: str) -> dict:
    content = requests.get(url).content
    return json.loads(zlib.decompress(content).decode('utf-8'))


def make_api_func(name: str, params):
    def api_func(*args: str):
        query_params = urlencode(dict(zip(params, args)))
        json_response = get("{}{}?{}".format(BASE_URL, name, query_params))
        return ApiResponse(json_response)
    return api_func


BASE_URL = "https://launcher.escapefromtarkov.com/launcher/"
ENDPOINTS = {
    "GetLauncherDistrib": [],
    "GetUpdatesFromVersion": ["version"],
    "GetUnpackedDistrib": ["version"],
    "GetDistrib": []
}
API = {
    name: make_api_func(name, params)
    for name, params in ENDPOINTS.items()
}