64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import re
|
|
import glob
|
|
import requests
|
|
import lxml.etree as ET # builtin library doesn't preserve comments
|
|
|
|
|
|
SETTINGS_PATH = "app/src/main/java/com/lagradost/cloudstream3/ui/settings/SettingsGeneral.kt"
|
|
START_MARKER = "/* begin language list */"
|
|
END_MARKER = "/* end language list */"
|
|
XML_NAME = "app/src/main/res/values-"
|
|
ISO_MAP_URL = "https://raw.githubusercontent.com/haliaeetus/iso-639/master/data/iso_639-1.min.json"
|
|
INDENT = " "*4
|
|
|
|
iso_map = requests.get(ISO_MAP_URL, timeout=300).json()
|
|
|
|
# Load settings file
|
|
src = open(SETTINGS_PATH, "r", encoding='utf-8').read()
|
|
before_src, rest = src.split(START_MARKER)
|
|
rest, after_src = rest.split(END_MARKER)
|
|
|
|
# Load already added langs
|
|
languages = {}
|
|
for lang in re.finditer(r'Triple\("(.*)", "(.*)", "(.*)"\)', rest):
|
|
flag, name, iso = lang.groups()
|
|
languages[iso] = (flag, name)
|
|
|
|
# Add not yet added langs
|
|
for folder in glob.glob(f"{XML_NAME}*"):
|
|
iso = folder[len(XML_NAME):]
|
|
if iso not in languages.keys():
|
|
entry = iso_map.get(iso.lower(),{'nativeName':iso})
|
|
languages[iso] = ("", entry['nativeName'].split(',')[0])
|
|
|
|
# Create triples
|
|
triples = []
|
|
for iso in sorted(languages.keys()):
|
|
flag, name = languages[iso]
|
|
triples.append(f'{INDENT}Triple("{flag}", "{name}", "{iso}"),')
|
|
|
|
# Update settings file
|
|
open(SETTINGS_PATH, "w+",encoding='utf-8').write(
|
|
before_src +
|
|
START_MARKER +
|
|
"\n" +
|
|
"\n".join(triples) +
|
|
"\n" +
|
|
END_MARKER +
|
|
after_src
|
|
)
|
|
|
|
# Go through each values.xml file and fix escaped \@string
|
|
for file in glob.glob(f"{XML_NAME}*/strings.xml"):
|
|
try:
|
|
tree = ET.parse(file)
|
|
for child in tree.getroot():
|
|
if child.text.startswith("\\@string/"):
|
|
print(f"[{file}] fixing {child.attrib['name']}")
|
|
child.text = child.text.replace("\\@string/", "@string/")
|
|
with open(file, 'wb') as fp:
|
|
fp.write(b'<?xml version="1.0" encoding="utf-8"?>\n')
|
|
tree.write(fp, encoding="utf-8", method="xml", pretty_print=True, xml_declaration=False)
|
|
except ET.ParseError as ex:
|
|
print(f"[{file}] {ex}")
|