From 23c678e71e0d74bd8a791ee41f4db5e1cc5f4e0c Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Sat, 23 Mar 2024 09:42:11 +0800
Subject: [PATCH] fix mtorrent

---
 app/chain/site.py               |  3 ++-
 app/modules/indexer/mtorrent.py | 18 ++++++++++++++----
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/app/chain/site.py b/app/chain/site.py
index ab1f8060..9768d3cf 100644
--- a/app/chain/site.py
+++ b/app/chain/site.py
@@ -45,7 +45,8 @@ class SiteChain(ChainBase):
         # 特殊站点登录验证
         self.special_site_test = {
             "zhuque.in": self.__zhuque_test,
-            # "m-team.io": self.__mteam_test,
+            "m-team.io": self.__mteam_test,
+            "m-team.cc": self.__mteam_test,
         }
 
     @staticmethod
diff --git a/app/modules/indexer/mtorrent.py b/app/modules/indexer/mtorrent.py
index cb34de06..876ae974 100644
--- a/app/modules/indexer/mtorrent.py
+++ b/app/modules/indexer/mtorrent.py
@@ -30,9 +30,9 @@ class MTorrentSpider:
 
     # 标签
     _labels = {
-        0: "",
-        4: "中字",
-        6: "国配",
+        "0": "",
+        "4": "中字",
+        "6": "国配",
     }
 
     def __init__(self, indexer: CommentedMap):
@@ -74,6 +74,14 @@ class MTorrentSpider:
         if res and res.status_code == 200:
             results = res.json().get('data', {}).get("data") or []
             for result in results:
+                category_value = result.get('category')
+                if category_value in self._tv_category \
+                        and category_value not in self._movie_category:
+                    category = MediaType.TV.value
+                elif category_value in self._movie_category:
+                    category = MediaType.MOVIE.value
+                else:
+                    category = MediaType.UNKNOWN.value
                 torrent = {
                     'title': result.get('name'),
                     'description': result.get('smallDescr'),
@@ -87,7 +95,9 @@ class MTorrentSpider:
                     'uploadvolumefactor': self.__get_uploadvolumefactor(result.get('status', {}).get("discount")),
                     'page_url': self._pageurl % (self._domain, result.get('id')),
                     'imdbid': self.__find_imdbid(result.get('imdb')),
-                    'labels': [self._labels.get(result.get('labels') or 0)] if result.get('labels') else []
+                    'labels': [self._labels.get(result.get('labels'))] if result.get('labels') and result.get(
+                        'labels') != "0" else [],
+                    'category': category
                 }
                 torrents.append(torrent)
         elif res is not None: