remove_db_data_field 397 B

1234567891011121314151617181920
  1. #!/usr/bin/python3
  2. import os
  3. import sys
  4. import yaml
  5. def migrate(values):
  6. if values["storage"].get("db_data"):
  7. values["storage"].pop("db_data", None)
  8. return values
  9. if __name__ == "__main__":
  10. if len(sys.argv) != 2:
  11. exit(1)
  12. if os.path.exists(sys.argv[1]):
  13. with open(sys.argv[1], "r") as f:
  14. print(yaml.dump(migrate(yaml.safe_load(f.read()))))