mirror of
https://github.com/truenas/charts.git
synced 2026-04-13 13:09:52 +08:00
56 lines
1.3 KiB
Python
Executable File
56 lines
1.3 KiB
Python
Executable File
#!/usr/bin/python3
|
|
import json
|
|
import re
|
|
import sys
|
|
|
|
from catalog_update.upgrade_strategy import semantic_versioning
|
|
|
|
|
|
RE_STABLE_VERSION_BASE = r'\d+\.\d+\.\d+'
|
|
ENUMS = {
|
|
'image': {
|
|
'RE_STABLE_VERSION': re.compile(RE_STABLE_VERSION_BASE),
|
|
'STRIP_TEXT': ''
|
|
},
|
|
'tensorrtImage': {
|
|
'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-tensorrt'),
|
|
'STRIP_TEXT': '-tensorrt'
|
|
},
|
|
}
|
|
|
|
|
|
def newer_mapping(image_tags):
|
|
output = {
|
|
"tags": {},
|
|
"app_version": ""
|
|
}
|
|
|
|
for key in image_tags.keys():
|
|
STRIP_TEXT = ENUMS[key].get('STRIP_TEXT', None) if key in ENUMS else None
|
|
RE_STABLE_VERSION = ENUMS[key].get('RE_STABLE_VERSION', None) if key in ENUMS else None
|
|
|
|
if (STRIP_TEXT is None) or (RE_STABLE_VERSION is None):
|
|
continue
|
|
|
|
tags = {t.strip(STRIP_TEXT): t for t in image_tags[key] if RE_STABLE_VERSION.fullmatch(t)}
|
|
version = semantic_versioning(list(tags))
|
|
|
|
if not version:
|
|
continue
|
|
|
|
if key == 'image':
|
|
output['app_version'] = version
|
|
|
|
output['tags'][key] = tags[version]
|
|
|
|
return output
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
versions_json = json.loads(sys.stdin.read())
|
|
except ValueError:
|
|
raise ValueError('Invalid json specified')
|
|
|
|
print(json.dumps(newer_mapping(versions_json)))
|