Last active
February 1, 2024 21:46
-
-
Save NooNameR/4c998dad0f462d387a9946b6540cdf10 to your computer and use it in GitHub Desktop.
Mover script which checking cache mount
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| # This standalone script is used to pause torrents older than last x days, | |
| # run mover (in Unraid) and start torrents again once completed | |
| import argparse | |
| import os | |
| import sys | |
| import time | |
| from datetime import datetime | |
| from datetime import timedelta | |
| parser = argparse.ArgumentParser(prog="Qbit Mover", description="Stop torrents and kick off Unraid mover process") | |
| parser.add_argument("--host", help="qbittorrent host including port", required="localhost:8080") | |
| parser.add_argument("-u", "--user", help="qbittorrent user", default="admin") | |
| parser.add_argument("-p", "--password", help="qbittorrent password", default="adminadmin") | |
| parser.add_argument("--cache-mount", "--cache_mount", help="Cache mount point in Unraid", default=None) | |
| parser.add_argument("--days-from", "--days_from", help="Set Number of Days to stop torrents between two offsets", type=int, default=0) | |
| parser.add_argument("--days-to", "--days_to", help="Set Number of Days to stop torrents between two offsets", type=int, default=2) | |
| # --START SCRIPT--# | |
| try: | |
| from qbittorrentapi import Client, LoginFailed, APIConnectionError | |
| except ModuleNotFoundError: | |
| print('Requirements Error: qbittorrent-api not installed. Please install using the command "pip install qbittorrent-api"') | |
| sys.exit(1) | |
| def filter_torrents(torrent_list, timeoffset_from, timeoffset_to, cache_mount): | |
| result = [] | |
| for torrent in torrent_list: | |
| if torrent.added_on >= timeoffset_to and torrent.added_on <= timeoffset_from: | |
| if not cache_mount or exists_in_cache(cache_mount, torrent.content_path): | |
| result.append(torrent) | |
| elif torrent.added_on < timeoffset_to: | |
| break | |
| return result | |
| def exists_in_cache(cache_mount, content_path): | |
| cache_path = os.path.join(cache_mount, content_path.lstrip('/')) | |
| return os.path.exists(cache_path) | |
| def stop_start_torrents(torrent_list, pause=True): | |
| for torrent in torrent_list: | |
| if pause: | |
| print(f"Pausing: {torrent.name} [{torrent.added_on}]") | |
| torrent.pause() | |
| else: | |
| print(f"Resuming: {torrent.name} [{torrent.added_on}]") | |
| torrent.resume() | |
| if __name__ == "__main__": | |
| current = datetime.now() | |
| args = parser.parse_args() | |
| if args.days_from > args.days_to: | |
| raise ("Config Error: days_from must be set lower than days_to") | |
| try: | |
| client = Client(host=args.host, username=args.user, password=args.password) | |
| except LoginFailed: | |
| raise ("Qbittorrent Error: Failed to login. Invalid username/password.") | |
| except APIConnectionError: | |
| raise ("Qbittorrent Error: Unable to connect to the client.") | |
| except Exception: | |
| raise ("Qbittorrent Error: Unable to connect to the client.") | |
| timeoffset_from = current - timedelta(days=args.days_from) | |
| timeoffset_to = current - timedelta(days=args.days_to) | |
| torrent_list = client.torrents.info(sort="added_on", reverse=True) | |
| torrents = filter_torrents(torrent_list, timeoffset_from.timestamp(), timeoffset_to.timestamp(), args.cache_mount) | |
| # Pause Torrents | |
| print(f"Pausing [{len(torrents)}] torrents from {args.days_from} - {args.days_to} days ago") | |
| stop_start_torrents(torrents, True) | |
| time.sleep(10) | |
| # Start mover | |
| print("Starting Mover") | |
| # Or using mover tunning | |
| #os.system('/usr/local/sbin/mover start') | |
| os.system("/usr/local/sbin/mover.old start") | |
| # Start Torrents | |
| print(f"Resuming [{len(torrents)}] paused torrents from {args.days_from} - {args.days_to} days ago") | |
| stop_start_torrents(torrents, False) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment