from crc import session
from crc.models.task_log import TaskLogModel, TaskLogModelSchema, TaskLogQuery
from crc.scripts.script import Script
from crc.services.task_logging_service import TaskLoggingService


class GetLogsByWorkflow(Script):

    def get_description(self):
        return """Script to retrieve logs for the current workflow. 
        Accepts an optional `code` argument that is used to filter the DB query.
        Accepts an optional 'size' otherwise will return the most recent 10 records.
        """

    def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
        log_model = TaskLogModel(level='info',
                                 code='mocked_code',
                                 message='This is my logging message',
                                 study_id=study_id,
                                 workflow_id=workflow_id,
                                 task=task.get_name())
        TaskLogModelSchema(many=True).dump([log_model])

    def do_task(self, task, study_id, workflow_id, *args, **kwargs):
        code = None
        size = 10
        if 'code' in kwargs:
            code = kwargs['code']
        elif len(args) > 0:
            code = args[0]
        if 'size' in kwargs:
            size = kwargs['size']
        elif len(args) > 1:
            size = args[1]

        query = TaskLogQuery(code=code, per_page=size)
        log_models = TaskLoggingService.get_logs_for_workflow(workflow_id, query).items
        return TaskLogModelSchema(many=True).dump(log_models)