fix context

2023-07-07 15:34:54 +08:00 · 2023-07-07 15:34:54 +08:00 · 89f2b60e7c
commit 89f2b60e7c
parent 1e1bba93de
1 changed files with 2 additions and 6 deletions
--- a/app/core/context.py
+++ b/app/core/context.py
@ -165,13 +165,9 @@ class MediaInfo:
        从字典中初始化
        """
        for key, value in data.items():
-            if key == "type":
-                if value == MediaType.MOVIE.value:
-                    setattr(self, key, MediaType.MOVIE)
-                elif value == MediaType.TV.value:
-                    setattr(self, key, MediaType.TV)
-                continue
            setattr(self, key, value)
+        if isinstance(self.type, str):
+            self.type = MediaType(self.type)

    def set_image(self, name: str, image: str):
        """