from crc import session
from crc.api.common import ApiError
from crc.models.file import FileModel
from crc.scripts.script import Script
from crc.services.document_service import DocumentService
from crc.services.file_service import FileService


class DeleteFile(Script):

    @staticmethod
    def process_document_deletion(doc_code, workflow_id, task):
        if DocumentService.is_allowed_document(doc_code):
            result = session.query(FileModel).filter(
                FileModel.workflow_id == workflow_id, FileModel.irb_doc_code == doc_code).all()
            if isinstance(result, list) and len(result) > 0 and isinstance(result[0], FileModel):
                for file in result:
                    FileService.delete_file(file.id)
            else:
                raise ApiError.from_task(code='no_document_found',
                                         message=f'No document of type {doc_code} was found for this workflow.',
                                         task=task)
        else:
            raise ApiError.from_task(code='invalid_document_code',
                                     message=f'{doc_code} is not a valid document code',
                                     task=task)

    def get_codes(self, task, args, kwargs):
        if 'code' in kwargs:
            if isinstance(kwargs['code'], list):
                codes = kwargs['code']
            else:
                codes = [kwargs['code']]
        else:
            codes = []
            for arg in args:
                if isinstance(arg, list):
                    codes.extend(arg)
                else:
                    codes.append(arg)

        if codes is None or len(codes) == 0:
            raise ApiError.from_task("invalid_argument", "Please provide a valid document code to delete.  "
                                                         "No valid arguments found.", task=task)
        return codes

    def get_description(self):
        return """Delete an IRB document from a workflow"""

    def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
        doc_codes = self.get_codes(task, args, kwargs)
        for code in doc_codes:
            result = session.query(FileModel).filter(
                FileModel.workflow_id == workflow_id, FileModel.irb_doc_code == code).all()
            if not result:
                return False
        return True

    def do_task(self, task, study_id, workflow_id, *args, **kwargs):
        doc_codes = self.get_codes(task, args, kwargs)
        for doc_code in doc_codes:
            self.process_document_deletion(doc_code, workflow_id, task)