Add rate limiting to download

docs/commands/download.md

@@ -79,6 +79,11 @@ twitch-dl download <video> [FLAGS] [OPTIONS]
     <td class="code">-o, --output</td>
     <td>Output file name template. See docs for details.</td>
 </tr>
+
+<tr>
+    <td class="code">-r, --rate-limit</td>
+    <td>Limit the maximum download speed in bytes per second. Use &#x27;k&#x27; and &#x27;m&#x27; suffixes for kbps and mbps.</td>
+</tr>
 </tbody>
 </table>
 
@@ -172,4 +177,4 @@ download command:
 
 ```
 twitch-dl download 221837124 --auth-token iduetx4i107rn4b9wrgctf590aiktv
-```
+```

twitchdl/commands/download.py

@@ -305,7 +305,7 @@ def _download_video(video_id, args):
         len(vod_paths), args.max_workers, target_dir))
     sources = [base_uri + path for path in vod_paths]
     targets = [os.path.join(target_dir, "{:05d}.ts".format(k)) for k, _ in enumerate(vod_paths)]
-    asyncio.run(download_all(sources, targets, args.max_workers))
+    asyncio.run(download_all(sources, targets, args.max_workers, rate_limit=args.rate_limit))
 
     # Make a modified playlist which references downloaded VODs
     # Keep only the downloaded segments and skip the rest

twitchdl/console.py

@@ -2,6 +2,7 @@
 
 import logging
 import sys
+import re
 
 from argparse import ArgumentParser, ArgumentTypeError
 from collections import namedtuple
@@ -46,6 +47,24 @@ def pos_integer(value):
     return value
 
 
+def rate(value):
+    match = re.search(r"^([0-9]+)(k|m|)$", value, flags=re.IGNORECASE)
+
+    if not match:
+        raise ArgumentTypeError("must be an integer, followed by an optional 'k' or 'm'")
+
+    amount = int(match.group(1))
+    unit = match.group(2)
+
+    if unit == "k":
+        return amount * 1024
+
+    if unit == "m":
+        return amount * 1024 * 1024
+
+    return amount
+
+
 COMMANDS = [
     Command(
         name="videos",
@@ -197,7 +216,12 @@ COMMANDS = [
                 "help": "Output file name template. See docs for details.",
                 "type": str,
                 "default": "{date}_{id}_{channel_login}_{title_slug}.{format}"
-            })
+            }),
+            (["-r", "--rate-limit"], {
+                "help": "Limit the maximum download speed in bytes per second. "
+                        "Use 'k' and 'm' suffixes for kbps and mbps.",
+                "type": rate,
+            }),
         ],
     ),
     Command(