mirror of
https://github.com/thegeneralist01/twitter-openapi
synced 2026-01-11 15:40:26 +01:00
116 lines
3.5 KiB
Python
116 lines
3.5 KiB
Python
import os
|
|
import glob
|
|
import json
|
|
import yaml
|
|
import shutil
|
|
import copy
|
|
import re
|
|
|
|
|
|
class placeholder_manager:
|
|
data: dict
|
|
config: str = "src/config/placeholder.json"
|
|
|
|
def __init__(self):
|
|
with open(self.config, mode="r", encoding="utf-8") as f:
|
|
self.data = json.load(f)
|
|
|
|
def __replace(self, file, old: str, new) -> str:
|
|
if type(new) is dict:
|
|
dump = f"'{json.dumps(new)}'"
|
|
return file.replace(f'"{old}"', dump)
|
|
else:
|
|
return file.replace(f"{old}", new)
|
|
|
|
def replace(self, file: str) -> str:
|
|
for prefix in self.data.keys():
|
|
for key in self.data[prefix]:
|
|
value = self.data[prefix][key]
|
|
file = self.__replace(file, f"{{{{{prefix}{key}}}}}", value)
|
|
return file
|
|
|
|
def replace_file(self, file: str, prefix: str) -> str:
|
|
if self.data.get(prefix, None) is None:
|
|
return file
|
|
for key in self.data[prefix].keys():
|
|
value = self.data[prefix][key]
|
|
file = self.__replace(file, f"{{{{{key}}}}}", value)
|
|
return file
|
|
|
|
|
|
OUTPUT_DIR = "dist/{0}"
|
|
INPUT_DIR = "src/openapi"
|
|
|
|
try:
|
|
shutil.rmtree("dist")
|
|
except:
|
|
pass
|
|
|
|
with open("src/config/variable.json", mode="r", encoding="utf-8") as f:
|
|
variable = json.load(f)
|
|
|
|
|
|
for lang in variable.keys():
|
|
|
|
def read(file: str):
|
|
with open(file, mode="r", encoding="utf-8") as f:
|
|
return remove(f.read())
|
|
|
|
def write(file: str, data: str) -> None:
|
|
with open(
|
|
file.replace(INPUT_DIR, OUTPUT_DIR.format(lang), 1),
|
|
mode="w+",
|
|
encoding="utf-8",
|
|
) as f:
|
|
f.write(data)
|
|
|
|
def get_yaml(data, key):
|
|
return yaml.safe_load(placeholder.replace_file(str(data), key))
|
|
|
|
def remove(data):
|
|
for match in re.findall(r"(\{% (.*?) %\})", data):
|
|
equation = match[1].split(" ")
|
|
if equation[0] == "if" and equation[2] == "==":
|
|
if equation[3] != variable[lang][equation[1]]:
|
|
data = re.sub(
|
|
re.escape(match[0]) + "[\s\S]*?" + re.escape("{% endif %}"),
|
|
"",
|
|
data,
|
|
)
|
|
|
|
return data
|
|
|
|
for dir in glob.glob(os.path.join(INPUT_DIR, "**/")):
|
|
os.makedirs(dir.replace(INPUT_DIR, OUTPUT_DIR.format(lang), 1), exist_ok=True)
|
|
|
|
placeholder = placeholder_manager()
|
|
parameters = read("src/config/parameters.yaml")
|
|
header = read("src/config/header.yaml")
|
|
|
|
paths = {}
|
|
for file in glob.glob(os.path.join(INPUT_DIR, "**/*.yaml")):
|
|
file = file.replace(os.path.sep, "/")
|
|
relative = file.replace(INPUT_DIR, "", 1)
|
|
|
|
load = yaml.safe_load(placeholder.replace(read(file)))
|
|
|
|
for key in load["paths"].keys():
|
|
append = get_yaml(parameters, key.split("/")[-1])
|
|
req = load["paths"][key]["get"]
|
|
req["parameters"] = append["paths"]["/parameters"]["get"]["parameters"]
|
|
|
|
append = get_yaml(header, key.split("/")[-1])
|
|
req = load["paths"][key]["get"]
|
|
req["responses"]["200"]["headers"] = append["components"]["headers"]
|
|
|
|
escape = key.replace("/", "~1")
|
|
paths.update({key: {"$ref": f".{relative}#/paths/{escape}"}})
|
|
write(file, yaml.dump(load))
|
|
|
|
file = "src/openapi/openapi-3.0.yaml"
|
|
data = read(file)
|
|
for path in paths:
|
|
load = yaml.safe_load(placeholder.replace(data))
|
|
load["paths"] = paths
|
|
|
|
write(file, yaml.dump(load))
|