mirror of
https://github.com/KeksPirates/SoftwareManager.git
synced 2026-08-03 17:39:42 +02:00
Vxrtrauter
104 lines
2.6 KiB
Python
104 lines
2.6 KiB
Python
import time
|
|
from core.utils.general.wrappers import run_thread
|
|
import threading
|
|
import libtorrent as lt
|
|
from core.utils.data.state import state
|
|
from core.utils.general.logs import consoleLog
|
|
|
|
|
|
global loop_running
|
|
loop_running = False
|
|
|
|
def init_session():
|
|
if state.dl_session is not None:
|
|
return
|
|
|
|
state.dl_session = lt.session()
|
|
|
|
settings = {
|
|
"upload_rate_limit": state.up_speed_limit,
|
|
"download_rate_limit": state.down_speed_limit,
|
|
"enable_dht": True,
|
|
"enable_lsd": True,
|
|
"enable_upnp": True,
|
|
"enable_natpmp": True,
|
|
"dht_bootstrap_nodes": "router.bittorrent.com:6881,dht.transmissionbt.com:6881",
|
|
"connections_limit": state.max_connections,
|
|
"active_downloads": state.max_downloads
|
|
}
|
|
|
|
|
|
|
|
state.dl_session.apply_settings(settings)
|
|
|
|
consoleLog("Initialized Session")
|
|
|
|
|
|
def add_download(magnet_uri, dl_path=state.download_path):
|
|
|
|
if state.active_downloads is None:
|
|
state.active_downloads = {}
|
|
|
|
init_session()
|
|
|
|
if magnet_uri in state.active_downloads:
|
|
consoleLog("Skipping, download already running...")
|
|
return
|
|
|
|
|
|
magnetdl = lt.parse_magnet_uri(magnet_uri)
|
|
magnetdl.save_path = dl_path
|
|
|
|
download = state.dl_session.add_torrent(magnetdl)
|
|
state.active_downloads[magnet_uri] = download
|
|
consoleLog(f"Added {magnet_uri} to downloads")
|
|
|
|
run_thread(threading.Thread(target=dl_status_loop))
|
|
|
|
|
|
|
|
def dl_status_loop():
|
|
global loop_running
|
|
if loop_running == True:
|
|
return
|
|
|
|
loop_running = True
|
|
completed_set = set()
|
|
|
|
if not state.active_downloads:
|
|
consoleLog("No active downloads")
|
|
loop_running = False
|
|
return
|
|
|
|
while state.active_downloads:
|
|
for magnet_uri, magnetdl in list(state.active_downloads.items()):
|
|
status = magnetdl.status()
|
|
|
|
if status.state == lt.torrent_status.seeding and magnet_uri not in completed_set:
|
|
consoleLog(f"Download completed: {status.name}")
|
|
|
|
completed_set.add(magnet_uri)
|
|
|
|
if not state.active_downloads:
|
|
loop_running = False
|
|
break
|
|
|
|
time.sleep(1)
|
|
|
|
loop_running = False
|
|
|
|
def update_settings():
|
|
|
|
if state.dl_session is None:
|
|
return
|
|
|
|
settings = {
|
|
"upload_rate_limit": state.up_speed_limit,
|
|
"download_rate_limit": state.down_speed_limit,
|
|
"connections_limit": state.max_connections,
|
|
"active_downloads": state.max_downloads
|
|
}
|
|
|
|
state.dl_session.apply_settings(settings)
|
|
consoleLog("Updated Settings")
|