From e17e6d8271930ddeea6abc86a833e2b51ae39197 Mon Sep 17 00:00:00 2001
From: maxDorninger <97409287+maxDorninger@users.noreply.github.com>
Date: Sun, 6 Jul 2025 15:17:16 +0200
Subject: [PATCH] add usenet bool to various schemas/models

---
 alembic/env.py                     | 2 ++
 media_manager/config.py            | 1 +
 media_manager/movies/repository.py | 1 +
 media_manager/movies/schemas.py    | 1 +
 media_manager/torrent/models.py    | 1 +
 media_manager/torrent/schemas.py   | 1 +
 media_manager/tv/schemas.py        | 1 +
 media_manager/tv/service.py        | 1 +
 8 files changed, 9 insertions(+)

diff --git a/alembic/env.py b/alembic/env.py
index 3183f03..f5730ed 100644
--- a/alembic/env.py
+++ b/alembic/env.py
@@ -30,6 +30,7 @@ from media_manager.torrent.models import Torrent  # noqa: E402
 from media_manager.tv.models import Show, Season, Episode, SeasonFile, SeasonRequest  # noqa: E402
 from media_manager.movies.models import Movie, MovieFile, MovieRequest  # noqa: E402
 from media_manager.notification.models import Notification  # noqa: E402
+from media_manager.usenet.models import UsenetDownload  # noqa: E402
 
 from media_manager.database import Base  # noqa: E402
 
@@ -51,6 +52,7 @@ target_metadata = Base.metadata
     MovieFile,
     MovieRequest,
     Notification,
+    UsenetDownload
 )
 
 
diff --git a/media_manager/config.py b/media_manager/config.py
index 05f1913..5064b33 100644
--- a/media_manager/config.py
+++ b/media_manager/config.py
@@ -9,6 +9,7 @@ class BasicConfig(BaseSettings):
     tv_directory: Path = "/data/tv"
     movie_directory: Path = "/data/movies"
     torrent_directory: Path = "/data/torrents"
+    usenet_directory: Path = "/data/usenet"
     FRONTEND_URL: AnyHttpUrl = "http://localhost:3000/"
     CORS_URLS: list[str] = []
     DEVELOPMENT: bool = False
diff --git a/media_manager/movies/repository.py b/media_manager/movies/repository.py
index 30ddd13..1f5c445 100644
--- a/media_manager/movies/repository.py
+++ b/media_manager/movies/repository.py
@@ -362,6 +362,7 @@ class MovieRepository:
                     quality=torrent.quality,
                     imported=torrent.imported,
                     file_path_suffix=file_path_suffix,
+                    usenet=torrent.usenet
                 )
                 formatted_results.append(movie_torrent)
             return formatted_results
diff --git a/media_manager/movies/schemas.py b/media_manager/movies/schemas.py
index 2c8c59e..de0433a 100644
--- a/media_manager/movies/schemas.py
+++ b/media_manager/movies/schemas.py
@@ -83,6 +83,7 @@ class MovieTorrent(BaseModel):
     quality: Quality
     imported: bool
     file_path_suffix: str
+    usenet: bool
 
 
 class RichMovieTorrent(BaseModel):
diff --git a/media_manager/torrent/models.py b/media_manager/torrent/models.py
index 61eeeda..296c879 100644
--- a/media_manager/torrent/models.py
+++ b/media_manager/torrent/models.py
@@ -14,6 +14,7 @@ class Torrent(Base):
     quality: Mapped[Quality]
     imported: Mapped[bool]
     hash: Mapped[str]
+    usenet: Mapped[bool]
 
     season_files = relationship("SeasonFile", back_populates="torrent")
     movie_files = relationship("MovieFile", back_populates="torrent")
diff --git a/media_manager/torrent/schemas.py b/media_manager/torrent/schemas.py
index 73ad3f4..dfcca71 100644
--- a/media_manager/torrent/schemas.py
+++ b/media_manager/torrent/schemas.py
@@ -39,3 +39,4 @@ class Torrent(BaseModel):
     quality: Quality
     imported: bool
     hash: str
+    usenet: bool = False
diff --git a/media_manager/tv/schemas.py b/media_manager/tv/schemas.py
index f1ba5d7..7defca3 100644
--- a/media_manager/tv/schemas.py
+++ b/media_manager/tv/schemas.py
@@ -117,6 +117,7 @@ class RichSeasonTorrent(BaseModel):
     status: TorrentStatus
     quality: Quality
     imported: bool
+    usenet: bool
 
     file_path_suffix: str
     seasons: list[SeasonNumber]
diff --git a/media_manager/tv/service.py b/media_manager/tv/service.py
index 12f96fe..22456e4 100644
--- a/media_manager/tv/service.py
+++ b/media_manager/tv/service.py
@@ -364,6 +364,7 @@ class TvService:
                 imported=show_torrent.imported,
                 seasons=seasons,
                 file_path_suffix=file_path_suffix,
+                usenet=show_torrent.usenet,
             )
             rich_season_torrents.append(season_torrent)
         return RichShowTorrent(