fga-eps-mds/2019.1-unbrake

'''
    Schema to use graphene framework to requirement on db
'''

import graphene
from graphene_django.types import DjangoObjectType
from graphql_jwt.decorators import login_required, superuser_required
from configuration.models import Config

# pylint: disable = too-few-public-methods


class ConfigType(DjangoObjectType):
    '''
        Defining the CyclesConfig Type
    '''
    class Meta:
        '''
            Defining the CyclesConfig Type
        '''
        model = Config


class CreateConfig(graphene.Mutation):
    # pylint: disable =  unused-argument, no-self-use, too-many-arguments
    # pylint: disable = too-many-locals
    '''
    Class to create a new Config object on db
    '''
    config = graphene.Field(ConfigType)

    class Arguments:
        '''
        Arguments required to create a new config
        '''
        name = graphene.String()
        number = graphene.Int()
        time_between_cycles = graphene.Int()
        upper_limit = graphene.Int()
        inferior_limit = graphene.Int()
        upper_time = graphene.Int()
        inferior_time = graphene.Int()
        disable_shutdown = graphene.Boolean()
        enable_output = graphene.Boolean()
        temperature = graphene.Float()
        time = graphene.Float()

    @login_required
    def mutate(
            self,
            info,
            name,
            number,
            time_between_cycles,
            upper_limit,
            inferior_limit,
            upper_time,
            inferior_time,
            disable_shutdown,
            enable_output,
            temperature,
            time):
        '''
        Create the Config with the given parameters end add to db
        '''
        config = Config(
            name=name,
            is_default=False,
            number=number,
            time_between_cycles=time_between_cycles,
            upper_limit=upper_limit,
            inferior_limit=inferior_limit,
            upper_time=upper_time,
            inferior_time=inferior_time,
            disable_shutdown=disable_shutdown,
            enable_output=enable_output,
            temperature=temperature,
            time=time,
        )
        config.save()

        return CreateConfig(config=config)


class CreateDefaultConfig(graphene.Mutation):
    # pylint: disable =  unused-argument, no-self-use, too-many-arguments
    # pylint: disable = too-many-locals
    '''
    Class to create a new default Config object on db
    '''
    config = graphene.Field(ConfigType)

    class Arguments:
        '''
        Arguments required to create a new config
        '''
        name = graphene.String()
        is_default = graphene.Boolean()
        number = graphene.Int()
        time_between_cycles = graphene.Int()
        upper_limit = graphene.Int()
        inferior_limit = graphene.Int()
        upper_time = graphene.Int()
        inferior_time = graphene.Int()
        disable_shutdown = graphene.Boolean()
        enable_output = graphene.Boolean()
        temperature = graphene.Float()
        time = graphene.Float()

    @superuser_required
    def mutate(
            self,
            info,
            number,
            name,
            time_between_cycles,
            upper_limit,
            inferior_limit,
            upper_time,
            inferior_time,
            disable_shutdown,
            enable_output,
            temperature,
            time):
        '''
            Set the othes default config as a ordinary Config
            and define a new default Config
        '''
        config = Config(
            name=name,
            is_default=True,
            number=number,
            time_between_cycles=time_between_cycles,
            upper_limit=upper_limit,
            inferior_limit=inferior_limit,
            upper_time=upper_time,
            inferior_time=inferior_time,
            disable_shutdown=disable_shutdown,
            enable_output=enable_output,
            temperature=temperature,
            time=time,
        )
        config.save()

        return CreateDefaultConfig(config=config)


class Mutation(graphene.ObjectType):
    '''
    GraphQL class to declare all the mutations
    '''
    create_config = CreateConfig.Field()
    create_default_config = CreateDefaultConfig.Field()


class Query:
    # pylint: disable =  unused-argument, no-self-use
    '''
        The Query list all the types created above
    '''

    all_config = graphene.List(ConfigType)

    config_at = graphene.Field(
        ConfigType,
        id=graphene.Int(),
        name=graphene.String(),
        is_default=graphene.Boolean(),
        number=graphene.Int(),
        time_between_cycles=graphene.Int(),
        upper_limit=graphene.Int(),
        inferior_limit=graphene.Int(),
        upper_time=graphene.Int(),
        inferior_time=graphene.Int(),
        disable_shutdown=graphene.Boolean(),
        enable_output=graphene.Boolean(),
        temperature=graphene.Float(),
        time=graphene.Float())

    config = graphene.Field(
        ConfigType,
        id=graphene.Int(),
        name=graphene.String(),
        is_default=graphene.Boolean(),
        number=graphene.Int(),
        time_between_cycles=graphene.Int(),
        upper_limit=graphene.Int(),
        inferior_limit=graphene.Int(),
        upper_time=graphene.Int(),
        inferior_time=graphene.Int(),
        disable_shutdown=graphene.Boolean(),
        enable_output=graphene.Boolean(),
        temperature=graphene.Float(),
        time=graphene.Float())

    config_default = graphene.List(ConfigType)

    config_not_default = graphene.List(ConfigType)

    @login_required
    def resolve_all_config(self, info, **kwargs):
        '''
            Returning all CyclesConfig on db
        '''
        return Config.objects.all()

    @login_required
    def resolve_config_at(self, info, **kwargs):
        '''
            Returning only one Config by id
        '''
        pk = kwargs.get('id')

        return Config.objects.get(pk=pk)

    @login_required
    def resolve_config(self, info, **kwargs):
        '''
            Return one config by name
        '''
        name = kwargs.get('name')

        return Config.objects.get(name=name)

    @login_required
    def resolve_config_default(self, info):
        '''
        Return a list with all the Config objects with is_default equal true
        '''
        return Config.objects.exclude(is_default=False)

    @login_required
    def resolve_config_not_default(self, info):
        '''
        Return a list with all the Config objects with is_default equal false
        '''
        return Config.objects.exclude(is_default=True)