# -----------------------------------------------------------------------------
# Copyright (c) 2014--, The Qiita Development Team.
#
# Distributed under the terms of the BSD 3-clause License.
#
# The full license is in the file LICENSE, distributed with this software.
# -----------------------------------------------------------------------------
from qiita_db.study import Study
from qiita_db.exceptions import QiitaDBUnknownIDError
from qiita_pet.handlers.util import to_int
from qiita_pet.handlers.base_handlers import BaseHandler


class RESTHandler(BaseHandler):
    def fail(self, msg, status, **kwargs):
        out = {'message': msg}
        out.update(kwargs)

        self.write(out)
        self.set_status(status)
        self.finish()

    def safe_get_study(self, study_id):
        study_id = to_int(study_id)
        s = None
        try:
            s = Study(study_id)
        except QiitaDBUnknownIDError:
            self.fail('Study not found', 404)
        finally:
            return s