LoaderSpot/update_version.py

78 lines
2.7 KiB
Python

#!/usr/bin/env python3
import json
import re
import sys
import os
def update_version(data_input, build_type: str = None, version_file: str = "versions.json") -> bool:
try:
# Если это путь к файлу
if os.path.isfile(data_input):
with open(data_input, 'r', encoding='utf-8') as f:
data = json.load(f)
# Если это JSON строка
else:
data = json.loads(data_input)
version_key = list(data.keys())[0]
version_data = data[version_key]
if build_type and build_type.lower() != 'false':
version_data = {'buildType': build_type, **version_data}
with open(version_file, 'r', encoding='utf-8') as f:
file_data = json.load(f)
file_data[version_key] = version_data
sorted_data = dict(
sorted(
file_data.items(),
key=lambda x: [
int(part) if part.isdigit() else part
for part in re.split(r'(\d+)', x[0])
],
reverse=True
)
)
with open(version_file, 'w', encoding='utf-8') as f:
json.dump(sorted_data, f, indent=2)
print(f"Version {version_key} added to {version_file}")
return True
except json.JSONDecodeError as e:
print(f"JSON parse error: {e}", file=sys.stderr)
return False
except FileNotFoundError as e:
print(f"File not found: {e}", file=sys.stderr)
return False
except Exception as e:
print(f"Error: {e}", file=sys.stderr)
return False
def main():
# Проверяем переменную окружения DATA_JSON_ENV (приоритет)
if 'DATA_JSON_ENV' in os.environ:
data_input = os.environ['DATA_JSON_ENV']
build_type = sys.argv[1] if len(sys.argv) > 1 else None
version_file = sys.argv[2] if len(sys.argv) > 2 else "versions.json"
# Иначе используем аргументы командной строки
elif len(sys.argv) >= 2:
data_input = sys.argv[1] # Может быть путь к файлу или JSON строка
build_type = sys.argv[2] if len(sys.argv) > 2 else None
version_file = sys.argv[3] if len(sys.argv) > 3 else "versions.json"
else:
print("Usage: python update_version.py <json_file_or_string> [build_type] [version_file]", file=sys.stderr)
print("Or set DATA_JSON_ENV environment variable", file=sys.stderr)
sys.exit(1)
success = update_version(data_input, build_type, version_file)
sys.exit(0 if success else 1)
if __name__ == "__main__":
main()