add httpout.exceptions

examples/exc.py

@@ -1,12 +1,13 @@
 
 from httpout import run
+from httpout.exceptions import WebSocketException
 
 
 print('Hi')
 
 
 async def main():
-    raise ValueError
+    raise WebSocketException('')
 
 
 run(main())

httpout/__init__.py

@@ -1,6 +1,6 @@
 # Copyright (c) 2024 nggit
 
-__version__ = '0.0.65'
+__version__ = '0.0.66'
 __all__ = ('HTTPOut',)
 
 from .httpout import HTTPOut  # noqa: E402

httpout/exceptions.py

@@ -0,0 +1,7 @@
+# Copyright (c) 2024 nggit
+
+from tremolo.exceptions import (  # noqa: F401
+    WebSocketException,
+    WebSocketClientClosed,
+    WebSocketServerClosed
+)

httpout/httpout.py

@@ -112,7 +112,9 @@ def ho_import(name, globals=None, locals=None, fromlist=(), level=0):
 
                     return module
 
-                if name == 'httpout' or name.startswith('httpout.'):
+                parent, _, child = name.partition('.')
+
+                if parent == 'httpout' and (child == '' or child in globals):
                     if '__server__' in globals:
                         module = globals['__main__'].__server__['modules'][
                             globals['__name__']

tests/test_http.py

@@ -139,7 +139,8 @@ def test_exc_after_print(self):
         self.assertEqual(header[:header.find(b'\r\n')], b'HTTP/1.1 200 OK')
         self.assertEqual(
             body,
-            b'3\r\nHi\n\r\n1F\r\n<ul><li>ValueError: </li></ul>\n\r\n0\r\n\r\n'
+            b'3\r\nHi\n\r\n27\r\n<ul><li>WebSocketException: </li></ul>\n\r\n'
+            b'0\r\n\r\n'
         )
 
     def test_exit_after_print(self):