Added in_use and priority_id field for channels urls. Added priorities table. Setup sql alchemy migration. Generate first migration.

app/models/db.py

@@ -1,12 +1,19 @@
 from datetime import datetime, timezone
 import uuid
-from sqlalchemy import Column, String, JSON, DateTime, UniqueConstraint, ForeignKey
+from sqlalchemy import Column, String, JSON, DateTime, UniqueConstraint, ForeignKey, Boolean, Integer
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship
 
 Base = declarative_base()
 
+class Priority(Base):
+    """SQLAlchemy model for channel URL priorities"""
+    __tablename__ = "priorities"
+    
+    id = Column(Integer, primary_key=True)
+    description = Column(String, nullable=False)
+
 class ChannelDB(Base):
     """SQLAlchemy model for IPTV channels"""
     __tablename__ = "channels"
@@ -34,8 +41,11 @@ class ChannelURL(Base):
     id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
     channel_id = Column(UUID(as_uuid=True), ForeignKey('channels.id', ondelete='CASCADE'), nullable=False)
     url = Column(String, nullable=False)
+    in_use = Column(Boolean, default=False, nullable=False)
+    priority_id = Column(Integer, ForeignKey('priorities.id'), nullable=False)
     created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
     updated_at = Column(DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
 
-    # Relationship with ChannelDB
-    channel = relationship("ChannelDB", back_populates="urls")
+    # Relationships
+    channel = relationship("ChannelDB", back_populates="urls")
+    priority = relationship("Priority")

app/models/schemas.py

@@ -1,17 +1,36 @@
 from datetime import datetime
 from typing import List, Optional
 from uuid import UUID
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
+
+class PriorityBase(BaseModel):
+    """Base Pydantic model for priorities"""
+    id: int
+    description: str
+
+    class Config:
+        from_attributes = True
+
+class PriorityCreate(PriorityBase):
+    """Pydantic model for creating priorities"""
+    pass
+
+class PriorityResponse(PriorityBase):
+    """Pydantic model for priority responses"""
+    pass
 
 class ChannelURLCreate(BaseModel):
     """Pydantic model for creating channel URLs"""
     url: str
+    priority_id: int = Field(default=100, ge=100, le=300)  # Default to High, validate range
 
 class ChannelURLBase(ChannelURLCreate):
     """Base Pydantic model for channel URL responses"""
     id: UUID
+    in_use: bool
     created_at: datetime
     updated_at: datetime
+    priority_id: int
 
     class Config:
         from_attributes = True
@@ -22,20 +41,26 @@ class ChannelURLResponse(ChannelURLBase):
 
 class ChannelCreate(BaseModel):
     """Pydantic model for creating channels"""
-    urls: List[str]  # List of URLs to create with the channel
+    urls: List[ChannelURLCreate]  # List of URL objects with priority
     name: str
     group_title: str
     tvg_id: str
     tvg_logo: str
     tvg_name: str
 
+class ChannelURLUpdate(BaseModel):
+    """Pydantic model for updating channel URLs"""
+    url: Optional[str] = None
+    in_use: Optional[bool] = None
+    priority_id: Optional[int] = Field(default=None, ge=100, le=300)
+
 class ChannelUpdate(BaseModel):
-    """Pydantic model for updating channels"""
-    name: Optional[str] = None
-    group_title: Optional[str] = None
-    tvg_id: Optional[str] = None
+    """Pydantic model for updating channels (all fields optional)"""
+    name: Optional[str] = Field(None, min_length=1)
+    group_title: Optional[str] = Field(None, min_length=1)
+    tvg_id: Optional[str] = Field(None, min_length=1)
     tvg_logo: Optional[str] = None
-    tvg_name: Optional[str] = None
+    tvg_name: Optional[str] = Field(None, min_length=1)
 
 class ChannelResponse(BaseModel):
     """Pydantic model for channel responses"""