Source code for ace.centralize.models.document

"""
    Document models for Centralize

    Models for attachments and document management (REST API).
"""
from typing import Optional

from ace.centralize.models.base import ModelBase




class AdditionalInfo(ModelBase):


    def __init__(
            self,
            document_type_id=0,
            note="",
            verified=False
    ):
        self.documentTypeId = document_type_id
        self.note = note
        self.verified = verified






class DocumentUpdateInformation(ModelBase):


    def __init__(
            self,
            attachment_id: int,
            additional_info: Optional[AdditionalInfo] = None,
            entity_type_id: Optional[str] = None,
            entity_id: Optional[int] = None
    ):
        self.attachmentId = attachment_id
        if additional_info:
            self.additionalInfo = additional_info
        if entity_type_id:
            self.entityTypeId = entity_type_id
        if entity_id:
            self.entityId = entity_id






class Attachment(ModelBase):


    def __init__(
            self,
            file_path="",
            file_content="",
            file_name="",
            additional_info=None,
            metadata=None,
    ):
        self.file_path = file_path
        self.file_content = file_content
        self.file_name = file_name
        if additional_info:
            self.additionalInfo = additional_info
        if metadata:
            self.metadata = metadata