diff --git a/cmscontrib/RemoveTeam.py b/cmscontrib/RemoveTeam.py new file mode 100644 index 0000000000..a5a8ae3f0f --- /dev/null +++ b/cmscontrib/RemoveTeam.py @@ -0,0 +1,66 @@ +#!/usr/bin/env python3 + +# Contest Management System - http://cms-dev.github.io/ +# Copyright © 2026 Pasit Sangprachathanarak +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . + +"""Utility to remove a team. + +""" + +import argparse +import logging +import sys + +from cms import utf8_decoder +from cms.db import Participation, SessionGen, Team + + +logger = logging.getLogger(__name__) + + +def remove_team(code: str) -> bool: + with SessionGen() as session: + team = session.query(Team).filter(Team.code == code).first() + if team is None: + logger.error("Team %s does not exist.", code) + return False + + session.query(Participation).filter(Participation.team_id == team.id).update( + {Participation.team_id: None} + ) + + session.delete(team) + session.commit() + + return True + + +def main(): + """Parse arguments and launch process. + + """ + parser = argparse.ArgumentParser( + description="Remove a team from CMS.") + parser.add_argument("code", action="store", type=utf8_decoder, + help="the team code") + args = parser.parse_args() + + success = remove_team(code=args.code) + return 0 if success is True else 1 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/setup.py b/setup.py index 76ac8b01d4..6e66ef0e9b 100755 --- a/setup.py +++ b/setup.py @@ -144,6 +144,7 @@ class build_with_l10n(build): "cmsRemoveParticipation=cmscontrib.RemoveParticipation:main", "cmsRemoveSubmissions=cmscontrib.RemoveSubmissions:main", "cmsRemoveTask=cmscontrib.RemoveTask:main", + "cmsRemoveTeam=cmscontrib.RemoveTeam:main", "cmsRemoveUser=cmscontrib.RemoveUser:main", "cmsSolutionChecker=cmscontrib.SolutionChecker:main", "cmsSpoolExporter=cmscontrib.SpoolExporter:main",