insert_redis_password_field 498 B

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