download_results.py

import requests

from personal_info import (
    get_cookie_dict,
    update_and_save_cookie_to_disk_if_values_changed,
)
from utils import get_recommender_url, get_session_id, get_steam_id


def get_recommender_results_url(steam_id=None):
    if steam_id is None:
        steam_id = get_steam_id()

    recommender_results_url = get_recommender_url(steam_id) + "results"

    return recommender_results_url


def get_recommender_results_request_params(ignored_app_ids=None):
    if ignored_app_ids is None:
        ignored_app_ids = []

    params = {
        "sessionid": get_session_id(),
        "steamid": get_steam_id(),
        "include_played": 0,
        "algorithm": 0,
        "reinference": 0,
        "model_version": 0,
    }

    if len(ignored_app_ids) > 0:
        params["ignored"] = ",".join(str(app_id) for app_id in sorted(ignored_app_ids))

    return params


def download_recommender_results(ignored_app_ids=None):
    if ignored_app_ids is None:
        ignored_app_ids = []

    url = get_recommender_results_url()
    params = get_recommender_results_request_params(ignored_app_ids=ignored_app_ids)
    cookies = get_cookie_dict()

    response = requests.get(url, params=params, cookies=cookies)

    if response.status_code == 200:
        result = response.json()

        jar = dict(response.cookies)
        update_and_save_cookie_to_disk_if_values_changed(cookies, jar)
    else:
        print(
            'Download of results failed with status code {}.'.format(
                response.status_code,
            ),
        )
        result = None

    return result


def main():
    result = download_recommender_results()

    return


if __name__ == '__main__':
    main()