allocate-torrents: limit to specific computers

chapmanjacobd · chapmanjacobd · commit 9e39b04f3443 · 2024-12-29T21:27:32.000-06:00
diff --git a/library/multidb/allocate_torrents.py b/library/multidb/allocate_torrents.py
@@ -42,6 +42,8 @@ def parse_args():
     )
     parser.add_argument("--up-limit", "--ul-limit", "--upload-limit", type=str, help="Upload limit")
 
+    parser.add_argument("--hosts", action=argparse_utils.ArgparseList, help="Limit to specific computers")
+
     arggroups.debug(parser)
 
     parser.add_argument("computer_database")
@@ -91,6 +93,8 @@ def allocate_torrents():
             },
         )
     )
+    disks = [d for d in disks if d["host"] in args.hosts]
+
     total_available = sum(d["free"] - args.min_free_space for d in disks)
     print(f"{len(disks)} disks matched. {strings.file_size(total_available)} available space")
 

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,8 @@ def parse_args():`
`42`	`42`	`)`
`43`	`43`	`parser.add_argument("--up-limit", "--ul-limit", "--upload-limit", type=str, help="Upload limit")`
`44`	`44`
	`45`	`+ parser.add_argument("--hosts", action=argparse_utils.ArgparseList, help="Limit to specific computers")`
	`46`	`+`
`45`	`47`	`arggroups.debug(parser)`
`46`	`48`
`47`	`49`	`parser.add_argument("computer_database")`
`@@ -91,6 +93,8 @@ def allocate_torrents():`
`91`	`93`	`},`
`92`	`94`	`)`
`93`	`95`	`)`
	`96`	`+ disks = [d for d in disks if d["host"] in args.hosts]`
	`97`	`+`
`94`	`98`	`total_available = sum(d["free"] - args.min_free_space for d in disks)`
`95`	`99`	`print(f"{len(disks)} disks matched. {strings.file_size(total_available)} available space")`
`96`	`100`