Use human readble names for release type

Flexget · Dec 16, 2022 · 0e8416a · 0e8416a
1 parent e8bb38e
commit 0e8416a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/flexget/components/tmdb/api_tmdb.py b/flexget/components/tmdb/api_tmdb.py
@@ -93,6 +93,15 @@ def upgrade(ver, session):
 )
 Base.register_table(genres_table)
 
+RELEASE_DATE_TYPE_MAPPING = {
+    1: 'premiere',
+    2: 'theatrical_limited',
+    3: 'theatrical',
+    4: 'digital',
+    5: 'physical',
+    6: 'tv',
+}
+
 
 class TMDBMovie(Base):
     __tablename__ = 'tmdb_movies'
@@ -193,7 +202,13 @@ def release_dates(self):
                 'release dates for movie {} not found in DB, fetching from TMDB', self.name
             )
             try:
-                self._release_dates = tmdb_request('movie/{}/release_dates'.format(self.id))
+                results = tmdb_request('movie/{}/release_dates'.format(self.id))['results']
+                for iso in results:
+                    for release in iso['release_dates']:
+                        release['type'] = RELEASE_DATE_TYPE_MAPPING[release['type']]
+
+                self._release_dates = results
+
             except requests.RequestException as e:
                 raise LookupError('Error updating data from tmdb: %s' % e)
         return self._release_dates

diff --git a/flexget/tests/test_tmdb.py b/flexget/tests/test_tmdb.py
@@ -25,6 +25,7 @@ def test_tmdb_lookup(self, execute_task):
             tmdb_name='The Matrix Resurrections', tmdb_year=2021
         ), 'Didn\'t populate tmdb info for The Matrix Resurrections'
 
+
 @pytest.mark.online
 class TestTmdbUnicodeLookup:
     config = """