pyutilscripts 0.8.0__tar.gz → 0.9.0__tar.gz

{pyutilscripts-0.8.0/pyutilscripts.egg-info → pyutilscripts-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyutilscripts
-Version: 0.8.0
+Version: 0.9.0
 Summary: PyUtilScripts 是一个基于 Python 的通用小工具集合，目标是提供编写通用任务的辅助工具。
 Author-email: Zero Kwok <zero.kwok@foxmail.com>
 License: MIT License

{pyutilscripts-0.8.0 → pyutilscripts-0.9.0}/pyproject.toml

@@ -46,4 +46,5 @@ Issues = "https://github.com/ZeroKwok/pyutilscripts/issues"
 [project.scripts]
 "fcopy" = "pyutilscripts.fcopy:main"
 "prunedirs" = "pyutilscripts.prunedirs:main"
-"forward.tcp" = "pyutilscripts.forward_tcp:main"
+"forward.tcp" = "pyutilscripts.forward_tcp:main"
+"httpd" = "pyutilscripts.httpd:main"

{pyutilscripts-0.8.0 → pyutilscripts-0.9.0}/pyutilscripts/__init__.py

@@ -2,7 +2,7 @@
 PyUtilScripts 是一个基于 Python 的通用小工具集合，目标是提供编写通用任务的辅助工具。
 """
 
-__version__ = "0.8.0"
+__version__ = "0.9.0"
 __status__ = ""
 __author__ = "zero <zero.kwok@foxmail.com>"
 

pyutilscripts-0.9.0/pyutilscripts/httpd.py

@@ -0,0 +1,105 @@
+#! python
+# -*- coding: utf-8 -*-
+
+import time
+import argparse
+import sys
+import os
+from http.server import HTTPServer, SimpleHTTPRequestHandler
+from socketserver import ThreadingMixIn
+
+class RateLimitHandler(SimpleHTTPRequestHandler):
+    # Default value
+    SPEED_LIMIT = -1 
+
+    def copyfile(self, source, outputfile):
+        """Override copyfile to implement rate limiting. If unlimited, use the original efficient method."""
+        # If unlimited, call parent class method (which uses shutil.copyfileobj internally and is more efficient)
+        if self.SPEED_LIMIT <= 0:
+            return super().copyfile(source, outputfile)
+
+        chunk_size = 1024 * 64  # 64KB chunk size
+        try:
+            start_time = time.time()
+            bytes_sent = 0
+            while True:
+                data = source.read(chunk_size)
+                if not data:
+                    break
+                outputfile.write(data)
+                bytes_sent += len(data)
+
+                # Rate limiting calculation
+                elapsed = time.time() - start_time
+                expected_time = bytes_sent / self.SPEED_LIMIT
+
+                if elapsed < expected_time:
+                    # 最小睡眠时间设为 0.5 秒, 因此理论最低速度约为 0.5 * 64kb = 128KB/s
+                    time.sleep(min(expected_time - elapsed, 0.5)) 
+        except (ConnectionResetError, BrokenPipeError):
+            print(f"\n[!] Client {self.client_address[0]} disconnected")
+
+class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
+    daemon_threads = True
+
+def parse_speed(speed_str):
+    """Parse speed string; support negative for unlimited."""
+    s = speed_str.upper()
+    try:
+        # Handle unlimited cases
+        if s == '-1' or s == '0' or s == 'UNLIMITED':
+            return -1
+        
+        if s.endswith('MB'):
+            return int(float(s[:-2]) * 1024 * 1024)
+        elif s.endswith('KB'):
+            return int(float(s[:-2]) * 1024)
+        else:
+            return int(s)
+    except ValueError:
+        raise argparse.ArgumentTypeError("Invalid speed format. Use '1MB', '500KB', '-1' (unlimited), or a byte value.")
+
+def main():
+    parser = argparse.ArgumentParser(description="Multithreaded HTTP file server with rate limiting")
+    parser.add_argument('--port', '-p', type=int, default=8000, help='Port to listen on (default: 8000)')
+    parser.add_argument('--bind', '-b', default='0.0.0.0', help='Bind address (default: 0.0.0.0)')
+    parser.add_argument('--limit', '-l', default='-1', help="Rate limit (e.g., 1MB, 500KB). Use -1 for unlimited")
+    parser.add_argument('--dir', '-d', default='.', help='Root directory to serve (default: current directory)')
+
+    args = parser.parse_args()
+
+    # 1. Verify directory exists
+    if not os.path.isdir(args.dir):
+        print(f"[ERROR] Directory does not exist: {args.dir}")
+        sys.exit(1)
+    
+    # 2. Change to and record absolute path
+    abs_path = os.path.abspath(args.dir)
+    os.chdir(abs_path)
+
+    # 3. Parse rate limit
+    limit_bps = parse_speed(args.limit)
+    limit_display = f"{args.limit}/s ({limit_bps})" if limit_bps > 0 else "Unlimited"
+
+    # 4. Dynamic handler class
+    HandlerClass = type('CustomHandler', (RateLimitHandler,), {'SPEED_LIMIT': limit_bps})
+
+    server_address = (args.bind, args.port)
+    httpd = ThreadedHTTPServer(server_address, HandlerClass)
+
+    print(f"{'='*45}")
+    print(f"[*] HTTP server started")
+    print(f"[*] Listening on: http://{args.bind}:{args.port}")
+    print(f"[*] Serving directory: {abs_path}")
+    print(f"[*] Speed limit: {limit_display}")
+    print(f"{'='*45}")
+    print("[!] Press Ctrl+C to stop the server\n")
+
+    try:
+        httpd.serve_forever()
+    except KeyboardInterrupt:
+        print("\n[!] Shutting down server...")
+        httpd.server_close()
+
+if __name__ == "__main__":
+    main()

{pyutilscripts-0.8.0 → pyutilscripts-0.9.0/pyutilscripts.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyutilscripts
-Version: 0.8.0
+Version: 0.9.0
 Summary: PyUtilScripts 是一个基于 Python 的通用小工具集合，目标是提供编写通用任务的辅助工具。
 Author-email: Zero Kwok <zero.kwok@foxmail.com>
 License: MIT License

{pyutilscripts-0.8.0 → pyutilscripts-0.9.0}/pyutilscripts.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ pyproject.toml
 pyutilscripts/__init__.py
 pyutilscripts/fcopy.py
 pyutilscripts/forward_tcp.py
+pyutilscripts/httpd.py
 pyutilscripts/prunedirs.py
 pyutilscripts/toast.py
 pyutilscripts.egg-info/PKG-INFO

{pyutilscripts-0.8.0 → pyutilscripts-0.9.0}/pyutilscripts.egg-info/entry_points.txt

@@ -1,4 +1,5 @@
 [console_scripts]
 fcopy = pyutilscripts.fcopy:main
 forward.tcp = pyutilscripts.forward_tcp:main
+httpd = pyutilscripts.httpd:main
 prunedirs = pyutilscripts.prunedirs:main