From d6822c2fe13b5508d5bace20d01e65074b6e9996 Mon Sep 17 00:00:00 2001
From: ghost <localhost>
Date: Sat, 4 Nov 2023 10:46:05 +0200
Subject: [PATCH] update yggtracker api, add search by category support

---
 yggtracker.py | 40 ++++++++++++++++++++++------------------
 1 file changed, 22 insertions(+), 18 deletions(-)

diff --git a/yggtracker.py b/yggtracker.py
index 050dcac..bf15c6a 100644
--- a/yggtracker.py
+++ b/yggtracker.py
@@ -12,6 +12,17 @@ class yggtracker(object):
 
   name = 'YGGtracker'
   url = 'https://github.com/YGGverse/YGGtracker'
+  supported_categories = [
+      'all',
+      'anime',
+      'books',
+      'games',
+      'movies',
+      'music',
+      'pictures',
+      'software',
+      'tv'
+  ]
 
   def __init__(self):
     pass
@@ -32,27 +43,20 @@ class yggtracker(object):
       # apply query request
       what = unquote(what)
       params = {
-        'query': what
+        'query': what,
+        'filter': 'true'
       }
 
-      # apply locales filter
-      if len(node['locales']) > 0:
-        params['locales'] = '|'.join(node['locales'])
+      # apply categories filter
+      categories = []
+      for category in node['categories'][cat]:
+        categories.append(category)
 
-      # apply sensitive filter
-      if node['sensitive'] is True:
-        params['sensitive'] = '1'
-      if node['sensitive'] is False:
-        params['sensitive'] = '0'
-
-      # apply yggdrasil filter
-      if node['yggdrasil'] is True:
-        params['yggdrasil'] = '1'
-      if node['yggdrasil'] is False:
-        params['yggdrasil'] = '0'
+      if len(categories) > 0:
+        params['categories'] = '|'.join(categories)
 
       # send api request
-      response = retrieve_url(node['api'] % urlencode(params))
+      response = retrieve_url(node['url'] % urlencode(params))
       response_json = json.loads(response)
 
       # check empty response
@@ -67,7 +71,7 @@ class yggtracker(object):
           'size': str(item['torrent']['file']['size']) + " B",
           'seeds': item['torrent']['scrape']['seeders'],
           'leech': item['torrent']['scrape']['leechers'],
-          'engine_url': node['url'],
-          'desc_link': item['torrent']['url'][node['locale']]
+          'engine_url': response_json['tracker']['url'],
+          'desc_link': item['torrent']['url']
         }
         prettyPrinter(res)
\ No newline at end of file