libentry 1.18__py3-none-any.whl → 1.19__py3-none-any.whl

libentry/test_api.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+
+__author__ = "xi"
+
+import json
+from argparse import ArgumentParser
+from concurrent.futures import ThreadPoolExecutor
+from threading import Lock
+from time import time
+from typing import Literal, Optional
+
+from pydantic import BaseModel
+
+from libentry.api import APIClient
+
+
+class TestRequest(BaseModel):
+    url: str
+    method: Literal["GET", "POST"] = "GET"
+    data: Optional[dict] = None
+    timeout: float = 15
+    num_threads: int = 1
+    num_calls: int = 1
+    quiet: bool = False
+    max_resp_len: int = 100
+
+
+class TestResponse(BaseModel):
+    num_success: int = 0
+    num_failed: int = 0
+    avg_time: float = 0
+    max_time: float = 0
+    min_time: float = 0
+
+
+def test(request: TestRequest):
+    time_list = []
+    lock = Lock()
+
+    def _worker(tid):
+        for cid in range(request.num_calls):
+            try:
+                kwargs = dict(
+                    on_error=lambda err: print(f"[{tid}:{cid}:RETRY] {err}"),
+                    timeout=request.timeout
+                )
+                t = time()
+                if request.method == "GET":
+                    response = APIClient().get(
+                        request.url,
+                        **kwargs
+                    )
+                else:
+                    response = APIClient().post(
+                        request.url,
+                        request.data,
+                        **kwargs
+                    )
+                t = time() - t
+                with lock:
+                    time_list.append(t)
+                if not request.quiet:
+                    content = str(response).replace("\n", " ")
+                    print(f"[{tid}:{cid}:SUCCESS] Time={t:.04f} Response={content:.{request.max_resp_len}}")
+            except Exception as e:
+                print(f"[{tid}:{cid}:FAILED] {e}")
+
+    with ThreadPoolExecutor(request.num_threads) as pool:
+        futures = [pool.submit(_worker, i) for i in range(request.num_threads)]
+        for future in futures:
+            future.result()
+
+    if len(time_list) > 0:
+        return TestResponse(
+            num_success=len(time_list),
+            num_failed=request.num_threads * request.num_calls - len(time_list),
+            max_time=max(time_list),
+            min_time=min(time_list),
+            avg_time=sum(time_list) / len(time_list)
+        )
+    else:
+        return TestResponse(
+            num_success=0,
+            num_failed=request.num_threads * request.num_calls,
+        )
+
+
+def main():
+    parser = ArgumentParser()
+    parser.add_argument("url", nargs="?")
+    parser.add_argument("--method", "-m", default="GET", choices=["GET", "POST"])
+    parser.add_argument("--data", "-d")
+    parser.add_argument("--timeout", type=float, default=15)
+    parser.add_argument("--num_threads", "-t", type=int, default=1)
+    parser.add_argument("--num_calls", "-c", type=int, default=1)
+    parser.add_argument("--quiet", "-q", action="store_true")
+    args = parser.parse_args()
+
+    if args.url is None:
+        print("URL should be given.")
+        return 1
+
+    data = args.data
+    if data is not None:
+        data = json.loads(data)
+    response = test(TestRequest(
+        url=args.url,
+        method=args.method,
+        data=data,
+        timeout=args.timeout,
+        num_threads=args.num_threads,
+        num_calls=args.num_calls,
+        quiet=args.quiet
+    ))
+    print(response.model_dump_json(indent=4))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

{libentry-1.18.dist-info → libentry-1.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: libentry
-Version: 1.18
+Version: 1.19
 Summary: Entries for experimental utilities.
 Home-page: https://github.com/XoriieInpottn/libentry
 Author: xi
@@ -13,6 +13,7 @@ Requires-Dist: Flask
 Requires-Dist: PyYAML
 Requires-Dist: gunicorn
 Requires-Dist: json5
+Requires-Dist: numpy
 Requires-Dist: pydantic
 Requires-Dist: requests
 

{libentry-1.18.dist-info → libentry-1.19.dist-info}/RECORD

@@ -8,6 +8,7 @@ libentry/json.py,sha256=1-Kv5ZRb5dBrOTU84n6sZtYZV3xE-O6wEt_--ynbSaU,1209
 libentry/logging.py,sha256=IiYoCUzm8XTK1fduA-NA0FI2Qz_m81NEPV3d3tEfgdI,1349
 libentry/schema.py,sha256=BO7EE7i43Cb4xn_OLsBM-HDwWOT6V5fPs91Am3QqnNQ,8178
 libentry/server.py,sha256=gYPoZXd0umlDYZf-6ZV0_vJadg3YQvnLDc6JFDJh9jc,1503
+libentry/test_api.py,sha256=uiHo9rVli9ENkzIyyFDDK6PdteBdetFTA8ikCDNykRQ,3488
 libentry/service/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 libentry/service/common.py,sha256=OVaW2afgKA6YqstJmtnprBCqQEUZEWotZ6tHavmJJeU,42
 libentry/service/flask.py,sha256=Alpsix01ROqI28k-dabD8wJlWAWs-ObNc1nJ-LxOXn4,13349
@@ -15,9 +16,10 @@ libentry/service/list.py,sha256=ElHWhTgShGOhaxMUEwVbMXos0NQKjHsODboiQ-3AMwE,1397
 libentry/service/running.py,sha256=FrPJoJX6wYxcHIysoatAxhW3LajCCm0Gx6l7__6sULQ,5105
 libentry/service/start.py,sha256=mZT7b9rVULvzy9GTZwxWnciCHgv9dbGN2JbxM60OMn4,1270
 libentry/service/stop.py,sha256=wOpwZgrEJ7QirntfvibGq-XsTC6b3ELhzRW2zezh-0s,1187
-libentry-1.18.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-libentry-1.18.dist-info/METADATA,sha256=3LKED9qfL_-_CvMdtfNW2YnoK1LBDdnQxN7Bf2_ZsKE,791
-libentry-1.18.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-libentry-1.18.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
-libentry-1.18.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-libentry-1.18.dist-info/RECORD,,
+libentry-1.19.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+libentry-1.19.dist-info/METADATA,sha256=L3cYIpAJrvAMvlHXoqrDqlKe3HeBf2_6BI8EfvqCGuo,812
+libentry-1.19.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+libentry-1.19.dist-info/entry_points.txt,sha256=vgHmJZhM-kqM7U9S179UwDD3pM232tpzJ5NntncXi_8,62
+libentry-1.19.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
+libentry-1.19.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+libentry-1.19.dist-info/RECORD,,

libentry-1.19.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+libentry_test_api = libentry.test_api:main
+