From 4920d244158486129c00dcf22ebeb96afdbfe45d Mon Sep 17 00:00:00 2001
From: Brage <bskjon@outlook.com>
Date: Tue, 18 Jul 2023 03:19:17 +0200
Subject: [PATCH] Updated meta sources

---
 pyMetadata/sources/anii.py | 20 +++++++++-----------
 pyMetadata/sources/imdb.py | 21 ++++++++-------------
 2 files changed, 17 insertions(+), 24 deletions(-)

diff --git a/pyMetadata/sources/anii.py b/pyMetadata/sources/anii.py
index 4a07e561..924b96e5 100644
--- a/pyMetadata/sources/anii.py
+++ b/pyMetadata/sources/anii.py
@@ -12,18 +12,16 @@ class metadata():
         """"""
         try:
             result = self.anilist.get_anime(self.name)
-            meta = Metadata()
-            meta.title = result.get("name_english", None)
-            meta.altTitle = result.get("name_romaji", None)
-            meta.cover = result.get("cover_image", None)
-            meta.summary = result.get("desc", None)
 
-            airing_format = result.get('airing_format', '').lower()
-            if airing_format == 'movie':
-                meta.type = 'movie'
-            else:
-                meta.type = 'serie'
-            meta.genres = result.get('genres', [])
+            meta = Metadata(
+                title = result.get("name_english", None),
+                altTitle = result.get("name_romaji", None),
+                cover = result.get("cover_image", None),
+                summary = result.get("desc", None),
+                type = 'movie' if result.get('airing_format', '').lower() == 'movie' else 'serie',
+                genres = result.get('genres', [])
+            )
+
             return DataResult("SUCCESS", None, meta)
 
         except IndexError as ingore:
diff --git a/pyMetadata/sources/imdb.py b/pyMetadata/sources/imdb.py
index 50dabf20..80223293 100644
--- a/pyMetadata/sources/imdb.py
+++ b/pyMetadata/sources/imdb.py
@@ -15,19 +15,14 @@ class metadata():
             query = self.imdb.search_movie(self.name)
             imdbId = query[0].movieID
             result = self.imdb.get_movie(imdbId)
-            meta = Metadata()
-            meta.title = result.get("title", None)
-            meta.altTitle = result.get("localized title", None)
-            meta.cover = result.get("cover url", None)
-            meta.summary = result.get("plot outline", None)
-
-            airing_format = result.get('kind', '').lower()
-            if airing_format == 'movie':
-                meta.type = 'movie'
-            else:
-                meta.type = 'serie'
-
-            meta.genres = result.get('genres', [])
+            meta = Metadata(
+                title = result.get("title", None),
+                altTitle = result.get("localized title", None),
+                cover = result.get("cover url", None),
+                summary = result.get("plot outline", None),
+                type = 'movie' if result.get('kind', '').lower() == 'movie' else 'serie',
+                genres = result.get('genres', [])
+            )
             
             return DataResult("SUCCESS", None, meta)
         except Exception as e: