This repository has been archived on 2025-08-26. You can view files and clone it, but cannot push or open issues or pull requests.
Files
enginex_bi_series-sherpa-onnx/flutter-examples/tts/generate-asset-list.py
2024-08-25 08:57:16 +08:00

99 lines
3.5 KiB
Python
Executable File

#!/usr/bin/env python3
"""
This file assumes that
assets:
is the last line in ./pubspec.yaml
It reads the file names of all files from the ./assets folder
and turns them as assets and writes them into ./pubspec.yaml
"""
import os
def main():
target = "./assets/"
space = " "
subfolders = []
patterns_to_skip = ["1.5x", "2.x", "3.x", "4.x"]
for root, dirs, files in os.walk(target):
for d in dirs:
path = os.path.join(root, d).replace("\\", "/")
if os.listdir(path):
path = path.lstrip('./')
if any(path.endswith(pattern) for pattern in patterns_to_skip):
continue
subfolders.append("{space}- {path}/".format(space=space, path=path))
assert subfolders, "The subfolders list is empty."
subfolders = sorted(subfolders)
loc_of_flutter = -1
loc_of_flutter_asset = -1
loc_of_end_flutter_asset = -1
loc_of_end_flutter = -1
with open("./pubspec.yaml", encoding="utf-8") as f:
lines = f.readlines()
for index, line in enumerate(lines):
if line == "flutter:\n":
loc_of_flutter = index + 1
if index == len(lines) - 1:
loc_of_end_flutter = index + 2
continue
if loc_of_flutter >= 0 and loc_of_flutter_asset < 0 and line == " assets:\n":
loc_of_flutter_asset = index + 1
continue
with open("./pubspec.yaml", encoding="utf-8") as f:
lines = f.readlines()
for index, line in enumerate(lines):
if index < loc_of_flutter:
continue
if loc_of_flutter_asset >= 0:
if line.startswith(" - assets/"):
loc_of_end_flutter_asset = index + 1
continue
else:
loc_of_end_flutter = index + 1
continue
else:
if line.startswith(" ") is False:
loc_of_end_flutter = index + 1
continue
else:
loc_of_end_flutter = index + 2
break
assert loc_of_flutter >= 0, "The 'flutter:' section is missing in the pubspec.yaml file."
with open("./pubspec.yaml", "w", encoding="utf-8") as f:
for index, line in enumerate(lines):
if loc_of_end_flutter_asset >= 0:
if index + 1 < loc_of_flutter_asset or index + 1 > loc_of_end_flutter_asset:
f.write(line)
if index + 1 == loc_of_flutter_asset:
f.write(" assets:\n")
for folder in subfolders:
f.write("{folder}\n".format(folder=folder))
else:
if index + 1 < loc_of_end_flutter or index + 1 > loc_of_end_flutter:
f.write(line)
if index + 1 == loc_of_end_flutter:
f.write(" assets:\n")
for indexOfFolder, folder in enumerate(subfolders):
f.write("{folder}\n".format(folder=folder))
if indexOfFolder == len(subfolders) - 1:
f.write("\n")
break
if loc_of_end_flutter == len(lines) + 1:
f.write("\n")
f.write(" assets:\n")
for folder in subfolders:
f.write("{folder}\n".format(folder=folder))
if __name__ == "__main__":
main()