from pydantic import validator, Field

from typing import Optional

from sqlalchemy import Column, Integer, ForeignKey

from dispatch.database.core import Base

from dispatch.messaging.strings import STORAGE_DESCRIPTION

from dispatch.models import ResourceBase, ResourceMixin

class Storage(Base, ResourceMixin):

id = Column(Integer, primary_key=True)

incident_id = Column(Integer, ForeignKey("incident.id", ondelete="CASCADE"))

case_id = Column(Integer, ForeignKey("case.id", ondelete="CASCADE"))

# Pydantic models...

class StorageBase(ResourceBase):

pass

class StorageCreate(StorageBase):

pass

class StorageUpdate(StorageBase):

pass

class StorageRead(StorageBase):

description: Optional[str] = Field(None, nullable=True)

@validator("description", pre=True, always=True)

def set_description(cls, v):

"""Sets the description"""

return STORAGE_DESCRIPTION