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