AddStatement.py

#!/usr/bin/env python3

# Contest Management System - http://cms-dev.github.io/
# Copyright © 2016 Myungwoo Chun <mc.tamaki@gmail.com>
# Copyright © 2016 Stefano Maggiolo <s.maggiolo@gmail.com>
#
# 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 <http://www.gnu.org/licenses/>.

"""This script adds a statement to a specific task in the database.

"""

import argparse
import logging
import os
import sys

from cms import utf8_decoder
from cms.db import SessionGen, Statement, Task
from cms.db.filecacher import FileCacher


logger = logging.getLogger(__name__)


def add_statement(task_name, language_code, statement_file, overwrite, default):
    logger.info("Adding the statement(language: %s) of task %s "
                "in the database.", language_code, task_name)

    if not os.path.exists(statement_file):
        logger.error("Statement file (path: %s) does not exist.",
                     statement_file)
        return False
    if not statement_file.endswith(".pdf"):
        logger.error("Statement file should be a pdf file.")
        return False

    with SessionGen() as session:
        task = session.query(Task)\
            .filter(Task.name == task_name).first()
        if not task:
            logger.error("No task named %s", task_name)
            return False
        try:
            file_cacher = FileCacher()
            digest = file_cacher.put_file_from_path(
                statement_file,
                "Statement for task %s (lang: %s)" %
                (task_name, language_code))
        except Exception:
            logger.error("Task statement storage failed.", exc_info=True)
        arr = session.query(Statement)\
            .filter(Statement.language == language_code)\
            .filter(Statement.task == task)\
            .all()
        if arr:  # Statement already exists
            if overwrite:
                logger.info("Overwriting already existing statement.")
                session.delete(arr[0])
                session.commit()
            else:
                logger.error("A statement with given language already exists. "
                             "Not overwriting.")
                return False
        statement = Statement(language_code, digest, task=task)

        # create a new list instance to force ORM object update
        primary_statements = task.primary_statements.copy()

        if default and language_code not in task.primary_statements:
            primary_statements.append(language_code)
        elif not default and language_code in task.primary_statements:
            primary_statements.remove(language_code)

        task.primary_statements = primary_statements

        session.add(statement)
        session.commit()

    logger.info("Statement added.")
    return True


def main():
    """Parse arguments and launch process.

    """
    parser = argparse.ArgumentParser(description="Add a statement to CMS.")
    parser.add_argument("task_name", action="store", type=utf8_decoder,
                        help="short name of the task")
    parser.add_argument("language_code", action="store", type=utf8_decoder,
                        help="language code of statement, e.g. en")
    parser.add_argument("statement_file", action="store", type=utf8_decoder,
                        help="absolute/relative path of statement file")
    parser.add_argument("-o", "--overwrite", dest="overwrite",
                        action="store_true",
                        help="overwrite existing statement")
    parser.add_argument("-d", "--default", dest="default",
                        action="store_true",
                        help="set statement as default")
    parser.set_defaults(overwrite=False, default=False)

    args = parser.parse_args()

    success = add_statement(args.task_name, args.language_code,
                            args.statement_file, args.overwrite, args.default)
    return 0 if success is True else 1


if __name__ == "__main__":
    sys.exit(main())