kris/tarc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

testing

Browse Source
This commit is contained in:
Kris Lamoureux 2025-03-15 23:29:39 -04:00
parent 613612415d
commit 43033572ad
1 changed files with 47 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
tarc/main.py
View File

@ -19,7 +19,7 @@ import qbittorrentapi
from sqlalchemy import create_engine, inspect from sqlalchemy import create_engine, inspect
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from .models import Base, SchemaVersion, Client from .models import Base, SchemaVersion, Client, Torrent, TorrentFile
# SCHEMA format is YYYYMMDDX # SCHEMA format is YYYYMMDDX
SCHEMA = 202503100 SCHEMA = 202503100
@ -101,15 +101,17 @@ def auth_qbittorrent(endpoint, username, password):
return qb return qb
def scan_torrents(qb_client, debug=False): def scan_torrents(qb_client, engine, debug=False):
""" """
Scan torrents using the provided qBittorrent client. Scan torrents using the provided qBittorrent client.
""" """
torrents = qb_client.torrents_info() torrents = qb_client.torrents_info()
print(f"[INFO]: There are {len(torrents)} torrents\n") print(f"[INFO]: There are {len(torrents)} torrents\n")
for torrent in torrents[:2]: for torrent in torrents[:2]:
files = qb_client.torrents_files(torrent.hash) files = qb_client.torrents_files(torrent.hash)
trackers = qb_client.torrents_trackers(torrent.hash) trackers = qb_client.torrents_trackers(torrent.hash)
print(f"[name]: {torrent.name}") print(f"[name]: {torrent.name}")
print(f"[infohash_v1]: {torrent.hash}") print(f"[infohash_v1]: {torrent.hash}")
print(f"[content_path]: {torrent.content_path}") print(f"[content_path]: {torrent.content_path}")
@ -117,11 +119,48 @@ def scan_torrents(qb_client, debug=False):
print(f"[completed_on]: {torrent.completed}\n") print(f"[completed_on]: {torrent.completed}\n")
print(f"[trackers]: {len(trackers)}") print(f"[trackers]: {len(trackers)}")
print(f"[file_count]: {len(files)}\n") print(f"[file_count]: {len(files)}\n")
if debug:
print(f"[DEBUG]: {repr(torrent)}") with Session(engine) as session:
for elem in trackers: completed_on = (
print(f"[DEBUG]: Tracker {repr(elem)}") datetime.fromtimestamp(torrent.completed) if torrent.completed else None
print("\n", end="") )
torrent_instance = Torrent(
info_hash_v1=torrent.hash,
file_count=len(files),
completed_on=completed_on,
)
if not session.query(Torrent).filter_by(info_hash_v1=torrent.hash).first():
session.add(torrent_instance)
session.commit()
print(f"[INFO]: Added torrent {torrent.name} to the database.")
for file in files:
if (
not session.query(TorrentFile)
.filter_by(file_path=file.name)
.first()
):
torrent_file_instance = TorrentFile(
torrent_id=torrent_instance.id,
file_id=file.id,
client_id=1,
file_index=file.index,
file_path=file.name,
is_downloaded=file.progress == 1,
last_checked=datetime.now(timezone.utc),
)
session.add(torrent_file_instance)
session.commit()
print(
f"[INFO]: Added {len(files)} files of torrent {torrent.name} to the database."
)
if debug:
print(f"[DEBUG]: {repr(torrent)}")
for elem in trackers:
print(f"[DEBUG]: Tracker {repr(elem)}")
print("\n", end="")
def scan(args, engine): def scan(args, engine):
@ -135,7 +174,7 @@ def scan(args, engine):
qb_client = auth_qbittorrent( qb_client = auth_qbittorrent(
client_info.endpoint, args.username, args.password client_info.endpoint, args.username, args.password
) )
scan_torrents(qb_client, debug=args.debug) scan_torrents(qb_client, engine, debug=args.debug)
elif len(clients) == 0: elif len(clients) == 0:
raise ValueError( raise ValueError(
f'Client with name "{args.name}" not found. ' f'Client with name "{args.name}" not found. '