uasyncio: Add StreamWriter.get_extra_info() to get peername.

dpgeorge · dpgeorge · commit 72d1f5ecfa18 · 2015-06-03T13:25:27.000Z
This is the standard way of getting the name of the client that is
connected.
diff --git a/uasyncio/uasyncio/__init__.py b/uasyncio/uasyncio/__init__.py
@@ -95,8 +95,9 @@ def __repr__(self):
 
 class StreamWriter:
 
-    def __init__(self, s):
+    def __init__(self, s, extra):
         self.s = s
+        self.extra = extra
 
     def awrite(self, buf):
         # This method is called awrite (async write) to not proliferate
@@ -130,6 +131,9 @@ def aclose(self):
         yield IOWriteDone(self.s)
         self.s.close()
 
+    def get_extra_info(self, name, default=None):
+        return self.extra.get(name, default)
+
     def __repr__(self):
         return "<StreamWriter %r>" % self.s
 
@@ -153,7 +157,7 @@ def open_connection(host, port):
         assert s2.fileno() == s.fileno()
     if __debug__:
         log.debug("open_connection: After iowait: %s", s)
-    return StreamReader(s), StreamWriter(s)
+    return StreamReader(s), StreamWriter(s, {})
 
 
 def start_server(client_coro, host, port, backlog=10):
@@ -176,7 +180,8 @@ def start_server(client_coro, host, port, backlog=10):
         s2.setblocking(False)
         if __debug__:
             log.debug("start_server: After accept: %s", s2)
-        yield client_coro(StreamReader(s2), StreamWriter(s2))
+        extra = {"peername": client_addr}
+        yield client_coro(StreamReader(s2), StreamWriter(s2, extra))
 
 
 import uasyncio.core
