fire-rpc 0.0.1.dev0__tar.gz → 0.1.0__tar.gz

{fire_rpc-0.0.1.dev0 → fire_rpc-0.1.0}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.1
 Name: fire-rpc
-Version: 0.0.1.dev0
-Summary: 
+Version: 0.1.0
+Summary: turn fire command line to a rpc server
 License: MIT
 Author: weihong.xu
 Author-email: xuweihong.cn@gmail.com
-Requires-Python: >=3.8,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: aiohttp (>=3.11.12,<4.0.0)
 Requires-Dist: fire (>=0.5.0,<0.6.0)
 Description-Content-Type: text/markdown
 

fire_rpc-0.1.0/fire_rpc/__init__.py

@@ -0,0 +1,35 @@
+from aiohttp import web
+import asyncio
+import fire
+import json
+
+def make_rpc_server(base_url: str, cmd_entry, secret = None,
+                    auth_header='X-Auth-Token', json_dumps=json.dumps):
+    async def handler(request: web.Request) -> web.Response:
+        try:
+            if secret is not None:
+                auth_token = request.headers.get(auth_header)
+                if auth_token != secret:
+                    return web.json_response({'error': 'unauthorized'}, status=401)
+            data = await request.json()
+            args = data.get('args')
+            if args is None:
+                return web.json_response({'error': 'args not found'}, status=400)
+            if not isinstance(args, list):
+                return web.json_response({'error': 'args must be a list'}, status=400)
+            ret = fire.Fire(cmd_entry, args)
+            res = {'result': ret}
+            return web.json_response(res, dumps=json_dumps)
+
+        except (Exception, SystemExit) as e:
+            return web.json_response({'error': str(e)}, status=400)
+
+    app = web.Application()
+    app.add_routes([web.post(base_url, handler)])
+    return app
+
+def start_rpc_server(base_url: str, cmd_entry, secret=None, json_dumps=json.dumps,
+                     host='localhost', port=8000, auth_header='X-Auth-Token'):
+    loop = asyncio.new_event_loop()
+    app = make_rpc_server(base_url, cmd_entry, secret=secret, auth_header=auth_header, json_dumps=json_dumps)
+    web.run_app(app, loop=loop, host=host, port=port)

fire_rpc-0.1.0/fire_rpc/__main__.py

@@ -0,0 +1,6 @@
+from . import start_rpc_server
+
+def echo(*args, **kwargs):
+    return {'args': args, 'kwargs': kwargs}
+
+start_rpc_server('/echo', echo, secret='changeme')

{fire_rpc-0.0.1.dev0 → fire_rpc-0.1.0}/pyproject.toml

@@ -1,14 +1,15 @@
 [tool.poetry]
 name = "fire-rpc"
-version = "0.0.1.dev0"
-description = ""
+version = "0.1.0"
+description = "turn fire command line to a rpc server"
 authors = ["weihong.xu <xuweihong.cn@gmail.com>"]
 license = "MIT"
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = "^3.8"
+python = "^3.9"
 fire = "^0.5.0"
+aiohttp = "^3.11.12"
 
 
 [build-system]