Fix #513 pkasyncio protect getpeername better

pykern/http.py

@@ -148,10 +148,10 @@ def __init__(self, http_config):
         )
         self._tornado = tornado.httpclient.AsyncHTTPClient(force_instance=True)
 
-    async def post(self, api_name, api_arg):
+    async def post(self, api_name, api_args):
         r = await self._tornado.fetch(
             self._uri,
-            body=_pack_msg(PKDict(api_name=api_name, api_arg=api_arg)),
+            body=_pack_msg(PKDict(api_name=api_name, api_args=api_args)),
             headers=self._headers,
             method="POST",
         )
@@ -160,7 +160,7 @@ async def post(self, api_name, api_arg):
             raise InvalidResponse(*e)
         if rv.api_error:
             raise APIError(
-                "api_error={} api_name={} api_arg={}", rv.api_error, api_name, api_arg
+                "api_error={} api_name={} api_args={}", rv.api_error, api_name, api_args
             )
         return rv.api_result
 
@@ -212,9 +212,9 @@ def _api_map():
         loop.http_server(h)
 
     async def dispatch(self, handler):
-        async def _call(api, api_arg):
+        async def _call(api, api_args):
             with pykern.quest.start(api.api_class, self.attr_classes) as qcall:
-                return await getattr(qcall, api.api_func_name)(api_arg)
+                return await getattr(qcall, api.api_func_name)(api_args)
 
         m = None
         try:
@@ -226,7 +226,7 @@ async def _call(api, api_arg):
                     raise Forbidden(*e)
                 if not (a := self.api_map.get(m.api_name)):
                     raise NotFound("unknown api={}", m.api_name)
-                r = Reply(result=await _call(a, m.api_arg))
+                r = Reply(result=await _call(a, m.api_args))
             except APIError as e:
                 r = Reply(api_error=e.pk_args.api_error)
             except Exception as e:

pykern/pkasyncio.py

@@ -70,8 +70,10 @@ def _remote_peer(request):
             # implementation may change; Code in tornado.httputil check connection.
             if c := request.connection:
                 # socket is not set on stream for websockets.
-                if hasattr(c, "stream") and hasattr(c.stream, "socket"):
-                    return "{}:{}".format(*c.stream.socket.getpeername())
+                if getattr(c, "stream", None) and (
+                    s := getattr(c.stream, "socket", None)
+                ):
+                    return "{}:{}".format(*s.getpeername())
             i = request.headers.get("proxy-for", request.remote_ip)
             return f"{i}:0"
 

pykern/pkdebug.py

@@ -280,6 +280,19 @@ def pkdp(fmt_or_arg, *args, **kwargs):
 pkdlog = pkdp
 
 
+def pkdppretty(obj):
+    """Print `obj` using `pkdpretty` and return `obj`
+
+    Args:
+        obj (object): JSON string or python object
+
+    Returns:
+        object: `obj`
+    """
+    pkdp("{}", pkdpretty(obj))
+    return obj
+
+
 def pkdpretty(obj):
     """Return pretty print the object.
 

pyproject.toml

@@ -57,5 +57,8 @@ pykern = "pykern.pykern_console:main"
 [project.urls]
 Homepage = "http://pykern.org"
 
+[tool.setuptools.packages.find]
+include = ["pykern*"]
+
 [tool.setuptools.package-data]
 pykern = ["package_data/**"]