# HG changeset patch
# User Thierry Florac <tflorac@ulthar.net>
# Date 1561568299 -7200
# Node ID 4f59f94ad901e718732b942dcb5ccbf6bdec102e
# Parent  5d4441d834c717ba46fc5d264d7a62b45ee61cdb
Store original host in websocket subscription

diff -r 5d4441d834c7 -r 4f59f94ad901 src/pyams_notify_ws/subscribe.py
--- a/src/pyams_notify_ws/subscribe.py	Wed Jun 26 11:13:45 2019 +0200
+++ b/src/pyams_notify_ws/subscribe.py	Wed Jun 26 18:58:19 2019 +0200
@@ -35,8 +35,9 @@
     subscription) for which we subscribe for a list of events types
     """
 
-    def __init__(self, principal):
+    def __init__(self, principal, host):
         self.principal = principal
+        self.host = host
         self.contexts = {}
 
     def update_contexts(self, new_contexts):
@@ -81,7 +82,7 @@
                 async with users_lock:
                     subscription = users.get(self.ws)
                     if subscription is None:
-                        subscription = WebsocketSubscription(principal)
+                        subscription = WebsocketSubscription(principal, message.get('host'))
                     subscription.update_contexts(message.get('context'))
                     users[self.ws] = subscription