58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
|
import re
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from uuid import UUID, uuid4
|
||
|
|
|
||
|
|
from api.domain import DomainValidationError
|
||
|
|
from api.domain.entity import DomainEntity
|
||
|
|
from api.domain.user.model import User, UserId
|
||
|
|
from api.domain.value_obj import DomainValueObject
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class CompanyEmail(DomainValueObject):
|
||
|
|
value: str
|
||
|
|
|
||
|
|
def __post_init__(self) -> None:
|
||
|
|
pattern = r"^[\w\.-]+@[a-zA-Z\d\.-]+\.[a-zA-Z]{2,}$"
|
||
|
|
|
||
|
|
if not re.match(pattern, self.value):
|
||
|
|
raise DomainValidationError(
|
||
|
|
"Invalid email format. Email must be in the format 'example@example.com'."
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class CompanyName(DomainValueObject):
|
||
|
|
value: str
|
||
|
|
|
||
|
|
def __post_init__(self) -> None:
|
||
|
|
if len(self.value) < 1:
|
||
|
|
raise DomainValidationError("First name must be at least 1 character long.")
|
||
|
|
if len(self.value) > 100:
|
||
|
|
raise DomainValidationError(
|
||
|
|
"First name must be at most 100 characters long."
|
||
|
|
)
|
||
|
|
if not self.value.isalpha():
|
||
|
|
raise DomainValidationError("First name must only contain letters.")
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class CompanyId(DomainValueObject):
|
||
|
|
value: UUID
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Company(DomainEntity[CompanyId]):
|
||
|
|
name: CompanyName
|
||
|
|
email: CompanyEmail
|
||
|
|
owner: User
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def create(name: str, email: str, owner: User) -> "Company":
|
||
|
|
return Company(
|
||
|
|
id=CompanyId(uuid4()),
|
||
|
|
name=CompanyName(name),
|
||
|
|
email=CompanyEmail(email),
|
||
|
|
owner=owner,
|
||
|
|
)
|