update_3.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. #!/usr/bin/env python3
  2. # Contest Management System - http://cms-dev.github.io/
  3. # Copyright © 2013 Luca Wehrstedt <luca.wehrstedt@gmail.com>
  4. #
  5. # This program is free software: you can redistribute it and/or modify
  6. # it under the terms of the GNU Affero General Public License as
  7. # published by the Free Software Foundation, either version 3 of the
  8. # License, or (at your option) any later version.
  9. #
  10. # This program is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. # GNU Affero General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU Affero General Public License
  16. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. """A class to update a dump created by CMS.
  18. Used by DumpImporter and DumpUpdater.
  19. This adapts the dump to some changes in the model introduced in commits
  20. 483ee85965f527cbc459ebe7e010c6854661b6eb
  21. e051ee4d2667ba381a24c7b1764a6d9c3d792b45
  22. d66951d3149a954fb0b81b6015e8e0b060020152
  23. """
  24. class Updater:
  25. def __init__(self, data):
  26. assert data["_version"] == 2
  27. self.objs = data
  28. def run(self):
  29. for k, v in self.objs.items():
  30. if k.startswith("_"):
  31. continue
  32. if v["_class"] == "User":
  33. if v["email"] == "":
  34. v["email"] = None
  35. if v["ip"] == "0.0.0.0":
  36. v["ip"] = None
  37. return self.objs