138 lines
3.2 KiB
Python
138 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import pymysql
|
|
import bencodepy
|
|
from openpyxl import Workbook
|
|
from openpyxl.utils import get_column_letter
|
|
|
|
# ============================================================
|
|
# DB CONFIG
|
|
# ============================================================
|
|
|
|
DB_CONFIG = {
|
|
"host": "192.168.1.50",
|
|
"user": "root",
|
|
"password": "Vlado9674+",
|
|
"database": "torrents",
|
|
"charset": "utf8mb4",
|
|
"cursorclass": pymysql.cursors.SSCursor
|
|
}
|
|
|
|
OUTPUT_FILE = "torrent_report.xlsx"
|
|
|
|
# ============================================================
|
|
# HELPERS
|
|
# ============================================================
|
|
|
|
def decode_if_bytes(value):
|
|
if isinstance(value, bytes):
|
|
return value.decode("utf-8", errors="replace")
|
|
return value
|
|
|
|
|
|
def get_root_name(info):
|
|
# prefer UTF-8 variant
|
|
if b"name.utf-8" in info:
|
|
return decode_if_bytes(info[b"name.utf-8"])
|
|
return decode_if_bytes(info[b"name"])
|
|
|
|
|
|
def get_file_parts(file_entry):
|
|
# prefer UTF-8 variant
|
|
if b"path.utf-8" in file_entry:
|
|
return [decode_if_bytes(p) for p in file_entry[b"path.utf-8"]]
|
|
return [decode_if_bytes(p) for p in file_entry[b"path"]]
|
|
|
|
|
|
def parse_torrent(blob):
|
|
|
|
data = bencodepy.decode(blob)
|
|
info = data[b"info"]
|
|
|
|
root_name = get_root_name(info)
|
|
|
|
files = []
|
|
|
|
# =====================
|
|
# MULTI FILE TORRENT
|
|
# =====================
|
|
if b"files" in info:
|
|
|
|
for f in info[b"files"]:
|
|
|
|
parts = get_file_parts(f)
|
|
|
|
# ochrana proti root/root duplicite
|
|
if parts and parts[0] == root_name:
|
|
full_path = "/".join(parts)
|
|
else:
|
|
full_path = root_name + "/" + "/".join(parts)
|
|
|
|
files.append(full_path)
|
|
|
|
# =====================
|
|
# SINGLE FILE TORRENT
|
|
# =====================
|
|
else:
|
|
files.append(root_name)
|
|
|
|
return files
|
|
|
|
|
|
# ============================================================
|
|
# MAIN
|
|
# ============================================================
|
|
|
|
def main():
|
|
|
|
conn = pymysql.connect(**DB_CONFIG)
|
|
|
|
wb = Workbook()
|
|
ws = wb.active
|
|
ws.title = "Torrent report"
|
|
|
|
ws.append(["QB Status", "Title", "Torrent Path"])
|
|
|
|
with conn.cursor() as cursor:
|
|
|
|
cursor.execute("""
|
|
SELECT qb_state, title_visible, torrent_content
|
|
FROM torrents
|
|
WHERE torrent_content IS NOT NULL
|
|
""")
|
|
|
|
for qb_state, title_visible, blob in cursor:
|
|
|
|
qb_state = qb_state or "UNKNOWN"
|
|
title_visible = title_visible or ""
|
|
|
|
try:
|
|
files = parse_torrent(blob)
|
|
|
|
for f in files:
|
|
ws.append([qb_state, title_visible, f])
|
|
|
|
except Exception as e:
|
|
ws.append([qb_state, title_visible, f"ERROR: {e}"])
|
|
|
|
# autosize
|
|
for col in ws.columns:
|
|
max_len = 0
|
|
col_letter = get_column_letter(col[0].column)
|
|
|
|
for cell in col:
|
|
if cell.value:
|
|
max_len = max(max_len, len(str(cell.value)))
|
|
|
|
ws.column_dimensions[col_letter].width = min(max_len + 2, 90)
|
|
|
|
wb.save(OUTPUT_FILE)
|
|
conn.close()
|
|
|
|
print("DONE ->", OUTPUT_FILE)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|