enforce HDR quality profile setting in processor

Author: lardbit

src/nefarious/migrations/0086_qualityprofile_is_hdr.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.0.2 on 2024-08-02 14:16
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('nefarious', '0085_auto_20240801_1323'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='qualityprofile',
+            name='is_hdr',
+            field=models.BooleanField(default=False),
+        ),
+    ]

src/nefarious/migrations/0087_auto_20240802_1457.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.0.2 on 2024-08-02 14:57
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('nefarious', '0086_qualityprofile_is_hdr'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='qualityprofile',
+            name='is_hdr',
+        ),
+        migrations.AddField(
+            model_name='qualityprofile',
+            name='hdr',
+            field=models.BooleanField(default=False, help_text='media must be in HDR (High Dynamic Range)'),
+        ),
+    ]

src/nefarious/models.py

@@ -25,6 +25,7 @@ class QualityProfile(models.Model):
         null=True, blank=True, max_digits=10, decimal_places=2, validators=[MinValueValidator(0)], help_text='minimum size (gb) to download')
     max_size_gb = models.DecimalField(
         null=True, blank=True, max_digits=10, decimal_places=2, validators=[MinValueValidator(0)], help_text='maximum size (gb) to download')
+    hdr = models.BooleanField(default=False, help_text='media must be in HDR (High Dynamic Range)')
 
     def __str__(self):
         if self.name == self.profile:

src/nefarious/parsers/base.py

@@ -4,6 +4,9 @@
 from nefarious.quality import Resolution, Profile
 
 
+# TODO add tags in the parsing: "5.1 dolby", release type ...
+
+
 # regex parsing taken from:
 # https://github.com/Sonarr/Sonarr/blob/537e4d7c39e839e75e7a7ad84e95cd582ec1d20e/src/NzbDrone.Core/Parser/QualityParser.cs
 # https://github.com/Radarr/Radarr/blob/e01900628f86696469875dcb79f60071278100ba/src/NzbDrone.Core/Parser/QualityParser.cs
@@ -57,6 +60,7 @@ class ParserBase:
     high_def_pdtv_regex = regex.compile(r"hr[-_. ]ws", regex.I)
     raw_hd_regex = regex.compile(r"\b(?<rawhd>RawHD|1080i[-_. ]HDTV|Raw[-_. ]HD|MPEG[-_. ]?2)\b", regex.I)
     hardcoded_subs_regex = regex.compile(r"\b(?<hc>hc|korsub)\b", regex.I)
+    hdr_regex = regex.compile(r"\bhdr\b", regex.I)
 
     def __init__(self, title):
         self.title_query = title
@@ -82,6 +86,13 @@ def parse(self):
             # hardcoded subs
             self.match['hc'] = self.parse_hardcoded_subs()
 
+            # hdr
+            self.match['hdr'] = self.parse_hdr()
+
+    def parse_hdr(self):
+        match = self.hdr_regex.search(self.title_query)
+        return True if match else False
+
     def parse_hardcoded_subs(self):
         match = self.hardcoded_subs_regex.search(self.title_query)
         return True if match else False
@@ -267,6 +278,9 @@ def is_match(self, title, *args, **kwargs) -> bool:
             return False
         return self._is_match(title, *args, **kwargs)
 
+    def is_hdr_match(self, needs_hdr = False):
+        return self.match['hdr'] if needs_hdr else True
+
     def is_quality_match(self, profile: Profile) -> bool:
         return self.match['quality'] in profile.qualities
 

src/nefarious/processors.py

@@ -7,6 +7,7 @@
 from transmissionrpc import Torrent
 
 from nefarious.models import WatchMovie, NefariousSettings, TorrentBlacklist, WatchTVEpisode, WatchTVSeason, QualityProfile
+from nefarious.parsers.base import ParserBase
 from nefarious.parsers.movie import MovieParser
 from nefarious.parsers.tv import TVParser
 from nefarious.quality import Profile
@@ -119,7 +120,7 @@ def is_match(self, title: str, size_kb: int) -> bool:
         size_gb = size_kb / (1024**2)
         mismatch_reason = ''
 
-        # TODO - test other profile attributes (min/max size, HDR, etc)
+        # TODO - test other profile attributes (dolby 5.1, etc)
 
         # title
         if not self._is_match(parser):
@@ -135,6 +136,10 @@ def is_match(self, title: str, size_kb: int) -> bool:
         # subs
         elif not parser.is_hardcoded_subs_match(self.nefarious_settings.allow_hardcoded_subs):
             mismatch_reason = f'hardcoded subs'
+        # hdr
+        elif not parser.is_hdr_match(quality_profile.hdr):
+            mismatch_reason = 'hdr'
+
         # keyword filters
         elif not parser.is_keyword_search_filter_match(
                 self.nefarious_settings.keyword_search_filters.keys() if self.nefarious_settings.keyword_search_filters else []
@@ -177,7 +182,7 @@ def _get_download_dir(self, transmission_session):
     def _get_tmdb_media(self):
         raise NotImplementedError
 
-    def _get_parser(self, title: str):
+    def _get_parser(self, title: str) -> ParserBase:
         raise NotImplementedError
 
     def _get_tmdb_title_key(self):
@@ -204,7 +209,7 @@ def _get_quality_profile(self) -> QualityProfile:
         # try custom quality profile then fallback to global setting
         return self.watch_media.quality_profile or self.nefarious_settings.quality_profile_movies
 
-    def _get_parser(self, title: str):
+    def _get_parser(self, title: str) -> MovieParser:
         return MovieParser(title)
 
     def _is_match(self, parser):
@@ -248,7 +253,7 @@ def _get_quality_profile(self) -> QualityProfile:
         watch_media = self.watch_media  # type: WatchTVEpisode|WatchTVSeason
         return watch_media.watch_tv_show.quality_profile or self.nefarious_settings.quality_profile_tv
 
-    def _get_parser(self, title: str):
+    def _get_parser(self, title: str) -> TVParser:
         return TVParser(title)
 
     def _get_media_type(self) -> str: