Linted and formatted all files

app/models/__init__.py

@@ -1,4 +1,19 @@
 from .db import Base, ChannelDB, ChannelURL
-from .schemas import ChannelCreate, ChannelUpdate, ChannelResponse, ChannelURLCreate, ChannelURLResponse
+from .schemas import (
+    ChannelCreate,
+    ChannelResponse,
+    ChannelUpdate,
+    ChannelURLCreate,
+    ChannelURLResponse,
+)
 
-__all__ = ["Base", "ChannelDB", "ChannelCreate", "ChannelUpdate", "ChannelResponse", "ChannelURL", "ChannelURLCreate", "ChannelURLResponse"]
+__all__ = [
+    "Base",
+    "ChannelDB",
+    "ChannelCreate",
+    "ChannelUpdate",
+    "ChannelResponse",
+    "ChannelURL",
+    "ChannelURLCreate",
+    "ChannelURLResponse",
+]

app/models/auth.py

@@ -1,20 +1,26 @@
-from typing import List, Optional
+from typing import Optional
+
 from pydantic import BaseModel, Field
 
+
 class SigninRequest(BaseModel):
     """Request model for the signin endpoint."""
+
     username: str = Field(..., description="The user's username")
     password: str = Field(..., description="The user's password")
 
+
 class TokenResponse(BaseModel):
     """Response model for successful authentication."""
+
     access_token: str = Field(..., description="Access JWT token from Cognito")
     id_token: str = Field(..., description="ID JWT token from Cognito")
-    refresh_token: Optional[str] = Field(
-        None, description="Refresh token from Cognito")
+    refresh_token: Optional[str] = Field(None, description="Refresh token from Cognito")
     token_type: str = Field(..., description="Type of the token returned")
 
+
 class CognitoUser(BaseModel):
     """Model representing the user returned from token verification."""
+
     username: str
-    roles: List[str]
+    roles: list[str]

app/models/db.py

@@ -1,21 +1,33 @@
-from datetime import datetime, timezone
 import uuid
-from sqlalchemy import Column, String, JSON, DateTime, UniqueConstraint, ForeignKey, Boolean, Integer
+from datetime import datetime, timezone
+
+from sqlalchemy import (
+    Boolean,
+    Column,
+    DateTime,
+    ForeignKey,
+    Integer,
+    String,
+    UniqueConstraint,
+)
 from sqlalchemy.dialects.postgresql import UUID
-from sqlalchemy.orm import declarative_base
-from sqlalchemy.orm import relationship
+from sqlalchemy.orm import declarative_base, 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"
 
     id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
@@ -25,27 +37,43 @@ class ChannelDB(Base):
     tvg_name = Column(String)
 
     __table_args__ = (
-        UniqueConstraint('group_title', 'name', name='uix_group_title_name'),
+        UniqueConstraint("group_title", "name", name="uix_group_title_name"),
     )
     tvg_logo = Column(String)
     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))
-    
+    updated_at = Column(
+        DateTime,
+        default=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(timezone.utc),
+    )
+
     # Relationship with ChannelURL
-    urls = relationship("ChannelURL", back_populates="channel", cascade="all, delete-orphan")
+    urls = relationship(
+        "ChannelURL", back_populates="channel", cascade="all, delete-orphan"
+    )
+
 
 class ChannelURL(Base):
     """SQLAlchemy model for channel URLs"""
+
     __tablename__ = "channels_urls"
 
     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)
+    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)
+    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))
+    updated_at = Column(
+        DateTime,
+        default=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(timezone.utc),
+    )
 
     # Relationships
     channel = relationship("ChannelDB", back_populates="urls")
-    priority = relationship("Priority")
+    priority = relationship("Priority")

app/models/schemas.py

@@ -1,30 +1,43 @@
 from datetime import datetime
-from typing import List, Optional
+from typing import Optional
 from uuid import UUID
-from pydantic import BaseModel, Field, ConfigDict
+
+from pydantic import BaseModel, ConfigDict, Field
+
 
 class PriorityBase(BaseModel):
     """Base Pydantic model for priorities"""
+
     id: int
     description: str
 
     model_config = ConfigDict(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
+    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
@@ -33,43 +46,53 @@ class ChannelURLBase(ChannelURLCreate):
 
     model_config = ConfigDict(from_attributes=True)
 
+
 class ChannelURLResponse(ChannelURLBase):
     """Pydantic model for channel URL responses"""
+
     pass
 
+
 class ChannelCreate(BaseModel):
     """Pydantic model for creating channels"""
-    urls: List[ChannelURLCreate]  # List of URL objects with priority
+
+    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 (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] = Field(None, min_length=1)
 
+
 class ChannelResponse(BaseModel):
     """Pydantic model for channel responses"""
+
     id: UUID
     name: str
     group_title: str
     tvg_id: str
     tvg_logo: str
     tvg_name: str
-    urls: List[ChannelURLResponse]  # List of URL objects without channel_id
+    urls: list[ChannelURLResponse]  # List of URL objects without channel_id
     created_at: datetime
     updated_at: datetime
 
-    model_config = ConfigDict(from_attributes=True)
+    model_config = ConfigDict(from_attributes=True)