Bruno Charest
817f8b4ee7
Some checks failed
Tests / Backend Tests (Python) (3.10) (push) Has been cancelled
Tests / Backend Tests (Python) (3.11) (push) Has been cancelled
Tests / Backend Tests (Python) (3.12) (push) Has been cancelled
Tests / Frontend Tests (JS) (push) Has been cancelled
Tests / Integration Tests (push) Has been cancelled
Tests / All Tests Passed (push) Has been cancelled
50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
"""Docker Image model for Homelab Automation."""
|
|
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import BigInteger, DateTime, ForeignKey, Integer, String, JSON, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from sqlalchemy.sql import func
|
|
|
|
from .database import Base
|
|
|
|
|
|
class DockerImage(Base):
|
|
"""Model representing a Docker image on a host."""
|
|
__tablename__ = "docker_images"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
host_id: Mapped[str] = mapped_column(String(50), ForeignKey("hosts.id", ondelete="CASCADE"), nullable=False)
|
|
image_id: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
repo_tags: Mapped[list] = mapped_column(JSON, nullable=True) # ["nginx:latest", "nginx:1.25"]
|
|
size: Mapped[int] = mapped_column(BigInteger, nullable=True)
|
|
created: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
last_update_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False, server_default=func.now(), onupdate=func.now()
|
|
)
|
|
|
|
# Relationship to host
|
|
host: Mapped["Host"] = relationship("Host", back_populates="docker_images")
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint('host_id', 'image_id', name='uq_docker_images_host_image'),
|
|
{"sqlite_autoincrement": True},
|
|
)
|
|
|
|
def __repr__(self) -> str: # pragma: no cover
|
|
tags = self.repo_tags[0] if self.repo_tags else self.image_id[:12]
|
|
return f"<DockerImage id={self.id} tags={tags}>"
|
|
|
|
def to_dict(self) -> dict:
|
|
"""Convert to dictionary for API responses."""
|
|
return {
|
|
"id": self.id,
|
|
"host_id": self.host_id,
|
|
"image_id": self.image_id,
|
|
"repo_tags": self.repo_tags,
|
|
"size": self.size,
|
|
"created": self.created.isoformat() if self.created else None,
|
|
"last_update_at": self.last_update_at.isoformat() if self.last_update_at else None,
|
|
}
|