update_37.py 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. #!/usr/bin/env python3
  2. # Contest Management System - http://cms-dev.github.io/
  3. # Copyright © 2018 Stefano Maggiolo <s.maggiolo@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. Add the feedback_level field for tasks. The default for new tasks is
  20. "restricted", but the old behavior is "full", so when updating we use "full".
  21. """
  22. from cms import FEEDBACK_LEVEL_FULL
  23. class Updater:
  24. def __init__(self, data):
  25. assert data["_version"] == 36
  26. self.objs = data
  27. def run(self):
  28. for k, v in self.objs.items():
  29. if k.startswith("_"):
  30. continue
  31. if v["_class"] == "Task":
  32. v["feedback_level"] = FEEDBACK_LEVEL_FULL
  33. return self.objs