Update everything to work with new API + Reorganize file structurem functions and common logic

core/utils/data/state.py

@@ -1,14 +1,15 @@
 from dataclasses import dataclass
-from typing import Optional, Any
-
+from typing import Optional, Any, List, Dict
 
 @dataclass
 class AppState:
+    posts: List[Dict[str, Any]]
+    post_titles: List[str]
+    post_urls: List[str]
     debug: bool = False
     tracker: str = "rutracker"
     api_url: str = "https://api.michijackson.xyz"
     aria2process: Optional[Any] = None
     aria2_threads: int = 4
 
-
-state = AppState()
+state = AppState(posts=[], post_titles=[], post_urls=[])

core/utils/data/tracker.py

@@ -1,43 +1,39 @@
 import requests
 from bs4 import BeautifulSoup
-from core.network.aria2_wrapper import start_client
-from core.network.aria2_wrapper import add_magnet
 from core.utils.data.state import state
+from core.utils.network.jsonhandler import format_data
 
 
 
-def get_item_index(item, list, listlinks, debug):
-        position = list.index(item)
-        if 0 <= position < len(listlinks): # note for myself: py starts counting at 0; check if number is not negative, check if number is not more than list length.
+def get_item_url(item, posts, post_titles): # oftwarelist currentitem, post list (dict), post titles list
+        post_index = post_titles.index(item)
+        if 0 <= post_index < len(post_titles): 
             if state.tracker == "uztracker":
-                selected = "https://uztracker.net/" + listlinks[position].lstrip("./")
+                item = "https://uztracker.net/" + state.post_urls[post_index].lstrip("./")
+                return item
             if state.tracker == "rutracker":                
-                selected = listlinks[position]
-
-            if debug:
-                print("Selected URL: ", selected)
-            selected_magnet = get_magnet_link(selected, debug)
-            start_client()
-            add_magnet(selected_magnet)
+                item_dict = posts[post_index]
+                _, post_links = format_data([item_dict])
+                return post_links[0]
+        return None
 
 
 
-
-def get_magnet_link(post_url, debug):
+def get_magnet_link(post_url):
     try:
         response = requests.get(post_url)
         response.raise_for_status()
         soup = BeautifulSoup(response.text, 'html.parser')
         magnet_link = soup.find('a', href=lambda x: x and x.startswith('magnet:'))
         if magnet_link:
-            if debug:
+            if state.debug:
                 print("Magnet Link Retrieved: ", magnet_link['href'])
             return magnet_link['href']
         else:
-            if debug:
+            if state.debug:
                 print("Magnet Link not Found!")
     except requests.RequestException as e:
-        if debug:
+        if state.debug:
             print(f"Failed to fetch {post_url}: {e}")
         return None
     

core/utils/network/download.py

@@ -1,17 +1,30 @@
 from core.utils.data.state import state
-from core.utils.data.tracker import get_item_index
+from core.utils.data.tracker import get_item_url
+from core.utils.data.tracker import get_magnet_link
+from core.network.aria2_wrapper import start_client
+from core.network.aria2_wrapper import add_magnet
 from core.utils.wrappers import run_thread
 import threading
 
 
-def download_selected(item, ):
+def download_selected(item, posts, post_titles):
     if item is not None:
         if state.debug:
             print(f"network {item.text()}")
-        run_thread(threading.Thread(target=get_item_index, args=(item.text(), self.postnames, self.postlinks, state.debug)))
+        run_thread(threading.Thread(target=run_download, args=(item.text(), posts, post_titles)))
+        
+        
     else:
         if state.debug:
             print("No item selected for download.")
 
+def run_download(item, posts, post_titles):
+    post_url = get_item_url(item, posts, post_titles)
+    if state.debug:
+        print("Selected URL: ", post_url)
+    magnet_uri = get_magnet_link(post_url)
+    start_client()
+    add_magnet(magnet_uri)
+
 
 

core/utils/network/jsonhandler.py

@@ -1,10 +1,18 @@
-def split_data(data):
-    json = json.loads(data)
+import json
 
-    count = json["count"]
-    posts = json["data"]
-    query = json["query"]
-    success = json["success"]
+def split_data(data):
+    p_json = json.loads(data)
+
+    count = p_json["count"]
+    posts = p_json["data"]
+    query = p_json["query"]
+    success = p_json["success"]
 
     return count, posts, query, success
 
+def format_data(data):
+    post_titles = [post["title"] for post in data]
+    post_links = [post["url"] for post in data]
+
+    return post_titles, post_links
+