snapshot

tormod.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3.11
+import argparse
+import io
+from pathlib import Path
+
+import torf
+
+
+def parse_args() -> dict:
+    arger = argparse.ArgumentParser()
+    arger.add_argument('torrent_path', type=Path, help='Path to torrent file')
+    arger.add_argument('--name', type=Path, help='New filename', required=True)
+    return dict(arger.parse_args().__dict__)
+
+
+def modify_torrent(
+    torrent_file: bytes,
+    name: str,
+    announce_url: str,
+    created_by: str = '',
+    comment: str = '',
+    private: bool = True,
+) -> bytes:
+    t = torf.Torrent.read_stream(io.BytesIO(torrent_file))
+
+    if len(t.files) > 1:
+        raise ValueError('Torrent contains multiple files')
+
+    t.name = name
+    t.private = private
+    t.comment = comment
+    t.created_by = created_by
+    t.trackers.clear()
+    t.trackers.insert(0, announce_url)
+
+    return t.dump()
+
+
+def main():
+    args = parse_args()
+    torrent_path: Path = args.pop('torrent_path')
+    modded = modify_torrent(
+        **args,
+        torrent_file=torrent_path.read_bytes(),
+        announce_url='http://tracker.empornium.sx:2710/tegqucis10uanp672qh6nhn393xlkncs/announce',
+        created_by='zzzp',
+        comment='created by zzzp',
+        private=True,
+    )
+    save_path = torrent_path.with_stem(f'{torrent_path.stem}.mod')
+    save_path.write_bytes(modded)
+
+
+if __name__ == '__main__':
+    main()