from tornado_swagger_ui.tornado_swagger import swagger
from handlers import GenericApiHandler
from common.constants import HTTP_FORBIDDEN
from project_models import Project


class GenericProjectHandler(GenericApiHandler):
    def __init__(self, application, request, **kwargs):
        super(GenericProjectHandler, self).__init__(application,
                                                    request,
                                                    **kwargs)
        self.table = 'projects'
        self.table_cls = Project


class ProjectCLHandler(GenericProjectHandler):
    @swagger.operation(nickname="list-all")
    def get(self):
        """
            @description: list all projects
            @return 200: return all projects, empty list is no project exist
            @rtype: L{Projects}
        """
        self._list()

    @swagger.operation(nickname="create")
    def post(self):
        """
            @description: create a project
            @param body: project to be created
            @type body: L{ProjectCreateRequest}
            @in body: body
            @rtype: L{Project}
            @return 200: project is created.
            @raise 403: project already exists
            @raise 400: post without body
        """
        def query(data):
            return {'name': data.name}

        def error(data):
            message = '{} already exists as a project'.format(data.name)
            return HTTP_FORBIDDEN, message

        db_check = [(self.table, False, query, error)]
        self._create(db_check)


class ProjectGURHandler(GenericProjectHandler):
    @swagger.operation(nickname='get-one')
    def get(self, project_name):
        """
            @description: get a single project by project_name
            @rtype: L{Project}
            @return 200: project exist
            @raise 404: project not exist
        """
        self._get_one({'name': project_name})

    @swagger.operation(nickname="update")
    def put(self, project_name):
        """
            @description: update a single project by project_name
            @param body: project to be updated
            @type body: L{ProjectUpdateRequest}
            @in body: body
            @rtype: L{Project}
            @return 200: update success
            @raise 404: project not exist
            @raise 403: new project name already exist or nothing to update
        """
        query = {'name': project_name}
        db_keys = ['name']
        self._update(query, db_keys)

    @swagger.operation(nickname='delete')
    def delete(self, project_name):
        """
            @description: delete a project by project_name
            @return 200: delete success
            @raise 404: project not exist
        """
        self._delete({'name': project_name})