Anjay

Torrent.py

@@ -1,65 +1,91 @@
 import libtorrent as lt
+import os
 import time
 
 ses = lt.session()
 ses.listen_on(6881, 6891)
+downloads = []
 
-params = {
-    'save_path': '/home/user/app/Torrent',
-    'storage_mode': lt.storage_mode_t(2),
-    'paused': False,
-    'auto_managed': True,
-    'duplicate_is_error': True
-}
+# Path folder Hasil Download
+download_folder = '/home/user/app/Torrent'
 
-def torrent_info(link, extension):
+# Membuat folder hasil download jika belum ada
+if not os.path.exists(download_folder):
+    os.makedirs(download_folder)
+    print("Folder 'Hasil Download' berhasil dibuat")
+else:
+    print("Folder 'Hasil Download' sudah ada")
+
+def torrent_info(link, search):
+    params = {
+        'save_path': download_folder,
+        'storage_mode': lt.storage_mode_t.storage_mode_sparse,
+        'paused': False,
+        'auto_managed': True,
+        'duplicate_is_error': True
+    }
     handle = lt.add_magnet_uri(ses, link, params)
+    ses.start_dht()
 
-    print('Mendownload metadata...')
+    print("Mendapatkan metadata dari magnet link...")
     while (not handle.has_metadata()):
         time.sleep(1)
-    print('Metadata didownload, memulai download torrent...')
+    print("Metadata diterima.")
 
-    torinfo = handle.get_torrent_info()
-    file_priorities = [0] * torinfo.num_files()
-    torrent_info = []
-    for idx, file in enumerate(torinfo.files()):
-        if file.path.endswith(extension):
-            print(f'{idx}. {file.path}')
-            torrent_info.append(file.path)
-            file_priorities[idx] = 1
-    handle.prioritize_files(file_priorities)
-    return torrent_info
+    info = handle.get_torrent_info()
+    file_list = info.files()
+    info_array = []
+    for i, file in enumerate(file_list):
+        if file.path.endswith(search):
+            info_array.append(f"{i}. {file.path}")
+    return info_array
 
 def torrent_download(link, selected_files):
+    params = {
+        'save_path': download_folder,
+        'storage_mode': lt.storage_mode_t.storage_mode_sparse,
+        'paused': False,
+        'auto_managed': True,
+        'duplicate_is_error': True
+    }
     handle = lt.add_magnet_uri(ses, link, params)
+    ses.start_dht()
 
-    # Menunggu metadata selesai didownload
+    print("Mendapatkan metadata dari magnet link...")
     while (not handle.has_metadata()):
         time.sleep(1)
+    print("Metadata diterima.")
 
-    # Mengatur prioritas file sesuai dengan selected_files
-    torinfo = handle.get_torrent_info()
-    file_priorities = [0] * torinfo.num_files()
+    info = handle.get_torrent_info()
+    file_list = info.files()
+    
+    file_priorities = [0] * len(file_list)
     for index in selected_files:
         file_priorities[index] = 1
+
     handle.prioritize_files(file_priorities)
 
-    # Mendownload file .mp4 yang dipilih
-    while (handle.status().state != lt.torrent_status.seeding):
+    while (not handle.is_seed()):
         s = handle.status()
         state_str = ['queued', 'checking', 'downloading metadata',
-                     'downloading', 'finished', 'seeding', 'allocating']
-        print('%.2f%% selesai (download: %.1f kb/s upload: %.1f kB/s peers: %d) %s' %
+                     'downloading', 'finished', 'seeding', 'allocating', 'checking fastresume']
+        print('%.2f%% complete (down: %.1f kb/s up: %.1f kB/s peers: %d) %s' %
               (s.progress * 100, s.download_rate / 1000, s.upload_rate / 1000,
                s.num_peers, state_str[s.state]))
-        time.sleep(5)
 
-    # Mencari file yang telah selesai didownload dan mengembalikan path-nya
+        # Status bar
+        progress_bar = '#' * int(s.progress * 100)
+        remaining_space = ' ' * (100 - len(progress_bar))
+        status_bar = '[%s%s] %.2f%%' % (progress_bar, remaining_space, s.progress * 100)
+        print(status_bar)
+
+        if state_str[s.state] == 'finished':
+            break
+
+        time.sleep(1)
+
+    # Return the path of the downloaded files
     downloaded_files = []
     for index in selected_files:
-        file_path = torinfo.files().file_path(index)
-        downloaded_files.append(file_path)
+        downloaded_files.append(os.path.join(download_folder, file_list[index].path))
     return downloaded_files
-
-

app.py

@@ -1,3 +1,11 @@
+import os
+
+os.system('pip install jedi')
+os.system('python -m pip install --upgrade pip setuptools wheel')
+os.system('python -m pip install lbry-libtorrent')
+os.system('apt-get install ffmpeg -y')
+os.system('apt install python3-libtorrent')
+
 import streamlit as st
 from streamlit_option_menu import option_menu
 from Torrent import torrent_info, torrent_download