125 lines
3.5 KiB
Python
125 lines
3.5 KiB
Python
import difflib
|
|
import orjson
|
|
import os
|
|
from datetime import date
|
|
|
|
import requests
|
|
|
|
from rare.lgndr.core import LegendaryCore
|
|
from rare.utils.paths import data_dir, cache_dir
|
|
|
|
replace_chars = ",;.:-_ "
|
|
url = "https://api.steampowered.com/ISteamApps/GetAppList/v2/"
|
|
|
|
|
|
__steam_ids_json = None
|
|
__grades_json = None
|
|
|
|
|
|
def get_rating(core: LegendaryCore, app_name: str):
|
|
global __grades_json
|
|
if __grades_json is None:
|
|
if os.path.exists(p := os.path.join(data_dir(), "steam_ids.json")):
|
|
grades = orjson.loads(open(p).read())
|
|
__grades_json = grades
|
|
else:
|
|
grades = {}
|
|
__grades_json = grades
|
|
else:
|
|
grades = __grades_json
|
|
|
|
if not grades.get(app_name):
|
|
game = core.get_game(app_name)
|
|
try:
|
|
steam_id = get_steam_id(game.app_title)
|
|
grade = get_grade(steam_id)
|
|
except:
|
|
return "fail"
|
|
grades[app_name] = {"steam_id": steam_id, "grade": grade}
|
|
with open(os.path.join(data_dir(), "steam_ids.json"), "w") as f:
|
|
f.write(orjson.dumps(grades).decode("utf-8"))
|
|
return grade
|
|
else:
|
|
return grades[app_name].get("grade")
|
|
|
|
|
|
# you should iniciate the module with the game's steam code
|
|
def get_grade(steam_code):
|
|
if steam_code == 0:
|
|
return "fail"
|
|
steam_code = str(steam_code)
|
|
url = "https://www.protondb.com/api/v1/reports/summaries/"
|
|
res = requests.get(f"{url}{steam_code}.json")
|
|
try:
|
|
lista = orjson.loads(res.text)
|
|
except orjson.JSONDecodeError:
|
|
return "fail"
|
|
|
|
return lista["tier"]
|
|
|
|
|
|
def load_json() -> dict:
|
|
file = os.path.join(cache_dir(), "game_list.json")
|
|
if not os.path.exists(file):
|
|
response = requests.get(url)
|
|
steam_ids = orjson.loads(response.text)["applist"]["apps"]
|
|
ids = {}
|
|
for game in steam_ids:
|
|
ids[game["name"]] = game["appid"]
|
|
|
|
with open(file, "w") as f:
|
|
f.write(orjson.dumps(ids).decode("utf-8"))
|
|
return ids
|
|
else:
|
|
return orjson.loads(open(file, "r").read())
|
|
|
|
|
|
def get_steam_id(title: str):
|
|
file = os.path.join(cache_dir(), "game_list.json")
|
|
# workarounds for satisfactory
|
|
title = title.replace("Early Access", "").replace("Experimental", "").strip()
|
|
global __steam_ids_json
|
|
if __steam_ids_json is None:
|
|
if not os.path.exists(file):
|
|
response = requests.get(url)
|
|
ids = {}
|
|
steam_ids = orjson.loads(response.text)["applist"]["apps"]
|
|
for game in steam_ids:
|
|
ids[game["name"]] = game["appid"]
|
|
__steam_ids_json = ids
|
|
|
|
with open(file, "w") as f:
|
|
f.write(orjson.dumps(ids).decode("utf-8"))
|
|
else:
|
|
ids = orjson.loads(open(file, "r").read())
|
|
__steam_ids_json = ids
|
|
else:
|
|
ids = __steam_ids_json
|
|
|
|
if title in ids.keys():
|
|
steam_name = [title]
|
|
|
|
else:
|
|
steam_name = difflib.get_close_matches(title, ids.keys(), n=1)
|
|
if steam_name:
|
|
return ids[steam_name[0]]
|
|
else:
|
|
return 0
|
|
|
|
|
|
def check_time(): # this function check if it's time to update
|
|
file = os.path.join(cache_dir(), "game_list.json")
|
|
json_table = orjson.loads(open(file, "r").read())
|
|
|
|
today = date.today()
|
|
day = 0 # it controls how many days it's necessary for an update
|
|
for i in "ymd":
|
|
if i == "d":
|
|
day = 7
|
|
else:
|
|
day = 0
|
|
if int(today.strftime("%" + i)) > int(json_table["data"][i]) + day:
|
|
return 1
|
|
else:
|
|
return 0
|