mirror of
https://github.com/truenas/charts.git
synced 2026-02-03 02:23:49 +08:00
94 lines
3.1 KiB
Python
Executable File
94 lines
3.1 KiB
Python
Executable File
#!/usr/bin/python3
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def storage_migrate(storage):
|
|
delete_keys = []
|
|
if storage['type'] == 'hostPath':
|
|
# Check if the key exists, if not we have already migrated
|
|
if not storage.get('hostPath'):
|
|
return storage
|
|
|
|
storage['hostPathConfig'] = {'hostPath': storage['hostPath']}
|
|
delete_keys.append('hostPath')
|
|
|
|
elif storage['type'] == 'ixVolume':
|
|
# Check if the key exists, if not we have already migrated
|
|
if not storage.get('datasetName'):
|
|
return storage
|
|
|
|
storage['ixVolumeConfig'] = {'datasetName': storage['datasetName']}
|
|
delete_keys.append('datasetName')
|
|
|
|
# Clean up for some older versions.
|
|
if storage.get('hostPath'):
|
|
delete_keys.append('hostPath')
|
|
|
|
elif storage['type'] == 'emptyDir':
|
|
# Check if the key exists, if not we have already migrated
|
|
if not 'medium' in storage.keys():
|
|
return storage
|
|
|
|
if storage['size'] and (not storage['size'].endswith('Gi')):
|
|
raise Exception('In order to proceed with the update, ' \
|
|
'please change the size for the storage using ' \
|
|
'"emptyDir" in the format xGi. eg. "2Gi"')
|
|
|
|
storage['emptyDirConfig'] = {
|
|
'medium': storage['medium'],
|
|
'size': int(storage['size'].replace('Gi', '')) if storage['size'] else None,
|
|
}
|
|
delete_keys.extend(['medium', 'size'])
|
|
|
|
elif storage['type'] == 'smb-pv-pvc':
|
|
# Check if the key exists, if not we have already migrated
|
|
if not storage.get('server'):
|
|
return storage
|
|
|
|
storage['smbConfig'] = {
|
|
'server': storage['server'],
|
|
'share': storage['share'],
|
|
'domain': storage['domain'],
|
|
'username': storage['username'],
|
|
'password': storage['password'],
|
|
'size': storage['size'],
|
|
}
|
|
delete_keys.extend(['server', 'share', 'domain', 'username', 'password', 'size'])
|
|
|
|
for key in delete_keys:
|
|
storage.pop(key, None)
|
|
|
|
return storage
|
|
|
|
|
|
def migrate(values):
|
|
storage_key = 'tdarrStorage'
|
|
storages = ['server', 'configs', 'logs', 'transcodes']
|
|
|
|
for storage in storages:
|
|
check_val = values.get(storage_key, {}).get(storage, {})
|
|
if not isinstance(check_val, dict) or not check_val:
|
|
raise Exception(f'Storage section {storage} is malformed')
|
|
|
|
values[storage_key][storage] = storage_migrate(check_val)
|
|
|
|
additionalStorages = values.get(storage_key, {}).get('additionalStorages', [])
|
|
for idx, storage in enumerate(additionalStorages):
|
|
if not isinstance(storage, dict) or not storage:
|
|
raise Exception(f'Item {idx} in additionalStorages is malformed')
|
|
|
|
values[storage_key]['additionalStorages'][idx] = storage_migrate(storage)
|
|
|
|
return values
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) != 2:
|
|
exit(1)
|
|
|
|
if os.path.exists(sys.argv[1]):
|
|
with open(sys.argv[1], 'r') as f:
|
|
print(json.dumps(migrate(json.loads(f.read()))))
|