##############################################################################
# Copyright (c) 2015 Orange
# guyrodrigue.koffi@orange.com / koffirodrigue@gmail.com
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import re

from opnfv_testapi.handlers import base_handlers
from opnfv_testapi.models import pod_models
from opnfv_testapi.tornado_swagger import swagger


class GenericPodHandler(base_handlers.GenericApiHandler):
    def __init__(self, application, request, **kwargs):
        super(GenericPodHandler, self).__init__(application, request, **kwargs)
        self.table = 'pods'
        self.table_cls = pod_models.Pod

    def set_query(self):
        query = dict()
        for k in self.request.query_arguments.keys():
            v = self.get_query_argument(k)
            if k == 'name':
                query['name'] = re.compile(v, re.IGNORECASE)
        return query


class PodCLHandler(GenericPodHandler):
    @swagger.operation(nickname='listAllPods')
    def get(self):
        """
            @description: list all pods
            @return 200: list all pods, empty list is no pod exist
            @rtype: L{Pods}
            @param name: pod name
            @type name: L{string}
            @in name: query
            @required name: False
        """
        self._list(query=self.set_query())

    @swagger.operation(nickname='createPod')
    def post(self):
        """
            @description: create a pod
            @param body: pod to be created
            @type body: L{PodCreateRequest}
            @in body: body
            @rtype: L{CreateResponse}
            @return 200: pod is created.
            @raise 403: pod already exists
            @raise 400: body or name not provided
        """
        def query():
            return {'name': self.json_args.get('name')}
        miss_fields = ['name']
        self._create(miss_fields=miss_fields, query=query)


class PodGURHandler(GenericPodHandler):
    @swagger.operation(nickname='getPodByName')
    def get(self, pod_name):
        """
            @description: get a single pod by pod_name
            @rtype: L{Pod}
            @return 200: pod exist
            @raise 404: pod not exist
        """
        self._get_one(query={'name': pod_name})

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