external/mailinabox
1
0
Fork 0
mirror of https://github.com/mail-in-a-box/mailinabox.git synced 2025-04-04 00:17:06 +00:00
Code Issues Releases Wiki Activity

Backup configuration conversion in setup/migrate.py is more robust.

Browse Source
This commit is contained in:
pappapisshu 2023-02-14 00:36:04 +01:00
parent 607ad27941
commit c2636f68f8
1 changed files with 62 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
setup/migrate.py
View File

@ -197,55 +197,74 @@ def migration_15(env):
import yaml import yaml
with open(backup_configuration_file, 'r') as f: with open(backup_configuration_file, 'r') as f:
custom_config = yaml.load(f, Loader=yaml.SafeLoader) old_configuration = yaml.load(f, Loader=yaml.SafeLoader)
if not isinstance(custom_config, dict): raise ValueError()
# Converting the previous configuration (which was not very clear) # Converting the previous configuration (which was not very clear)
# into the new configuration format which also provides # into the new configuration format which also provides
# a "type" attribute to distinguish the type of backup. # a "type" attribute to distinguish the type of backup.
scheme = custom_config["target"].split(":")[0] conversion_succeded = True
if not isinstance(old_configuration, dict):
conversion_succeded = False
print("Exception when converting to the new configuration format: the previous configuration has an unexpected format.")
elif "type" in old_configuration:
print("The previous configuration appears to have already been converted.")
return
else:
try:
scheme = old_configuration["target"].split(":")[0]
if scheme == "off": if scheme == "off":
custom_config = { new_configuration = {
"type": "off" "type": "off"
} }
elif scheme == "file": elif scheme == "file":
custom_config = { new_configuration = {
"type": "local", "type": "local",
"min_age_in_days": custom_config["min_age_in_days"] "min_age_in_days": old_configuration["min_age_in_days"]
} }
elif scheme == "rsync": elif scheme == "rsync":
custom_config = { new_configuration = {
"type": "rsync", "type": "rsync",
"target_url": custom_config["target"], "target_url": old_configuration["target"],
"min_age_in_days": custom_config["min_age_in_days"] "min_age_in_days": old_configuration["min_age_in_days"]
} }
elif scheme == "s3": elif scheme == "s3":
import urllib.parse import urllib.parse
url = urllib.parse.urlparse(custom_config["target"]) url = urllib.parse.urlparse(old_configuration["target"])
target_url = url.scheme + ":/" + url.path target_url = url.scheme + ":/" + url.path
s3_endpoint_url = "https://" + url.netloc s3_endpoint_url = "https://" + url.netloc
s3_access_key_id = custom_config["target_user"] new_configuration = {
s3_secret_access_key = custom_config["target_pass"]
custom_config = {
"type": "s3", "type": "s3",
"target_url": target_url, "target_url": target_url,
"s3_endpoint_url": s3_endpoint_url, "s3_endpoint_url": s3_endpoint_url,
"s3_access_key_id": custom_config["target_user"], "s3_access_key_id": old_configuration["target_user"],
"s3_secret_access_key": custom_config["target_pass"], "s3_secret_access_key": old_configuration["target_pass"],
"min_age_in_days": custom_config["min_age_in_days"] "min_age_in_days": old_configuration["min_age_in_days"]
} }
elif scheme == "b2": elif scheme == "b2":
custom_config = { new_configuration = {
"type": "b2", "type": "b2",
"target_url": custom_config["target"], "target_url": old_configuration["target"],
"min_age_in_days": custom_config["min_age_in_days"] "min_age_in_days": old_configuration["min_age_in_days"]
} }
else: else:
raise ValueError("Unexpected scheme during the conversion of the previous config to the new format.") conversion_succeded = False
print("Exception when converting to the new configuration format: the previous configuration has an unexpected scheme.")
except KeyError:
conversion_succeded = False
print("Exception when converting to the new configuration format: the previous configuration is missing one or more keys.")
except:
conversion_succeded = False
print("Exception when converting to the new configuration format: generic exception.")
if conversion_succeded:
with open(backup_configuration_file, "w") as f: with open(backup_configuration_file, "w") as f:
f.write(yaml.dump(custom_config)) f.write(yaml.dump(new_configuration))
else:
with open(backup_configuration_file, "r") as f:
lines = list(map(lambda line:"# " + line, f.readlines()))
with open(backup_configuration_file, "w") as f:
f.writelines(lines)
print("The previous configuration inside the", backup_configuration_file, "file has been commented out.")
########################################################### ###########################################################