30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
from __future__ import annotations
|
|
import enum
|
|
import uuid
|
|
from datetime import datetime, timezone
|
|
from sqlalchemy import Boolean, DateTime, Enum, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from .base import Base
|
|
|
|
|
|
class ProbeType(str, enum.Enum):
|
|
tcp = "tcp"
|
|
icmp = "icmp"
|
|
|
|
|
|
class Host(Base):
|
|
__tablename__ = "hosts"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
name: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
address: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
probe_type: Mapped[ProbeType] = mapped_column(Enum(ProbeType), nullable=False, default=ProbeType.tcp)
|
|
probe_port: Mapped[int] = mapped_column(Integer, nullable=False, default=80)
|
|
ping_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
|
dns_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
|
dns_expected_ip: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
poll_interval_seconds: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc)
|
|
)
|