forked from platypush/platypush
38 lines
1013 B
Python
38 lines
1013 B
Python
import os
|
|
import importlib
|
|
import inspect
|
|
from typing import List, Type
|
|
|
|
import pkgutil
|
|
|
|
from ..ws import WSRoute, logger
|
|
|
|
|
|
def get_ws_routes() -> List[Type[WSRoute]]:
|
|
"""
|
|
Scans for websocket route objects.
|
|
"""
|
|
from platypush.backend.http import HttpBackend
|
|
|
|
base_pkg = '.'.join([HttpBackend.__module__, 'app', 'ws'])
|
|
base_dir = os.path.join(os.path.dirname(inspect.getfile(HttpBackend)), 'app', 'ws')
|
|
routes = []
|
|
|
|
for _, mod_name, _ in pkgutil.walk_packages([base_dir], prefix=base_pkg + '.'):
|
|
try:
|
|
module = importlib.import_module(mod_name)
|
|
except Exception as e:
|
|
logger.warning('Could not import module %s', mod_name)
|
|
logger.exception(e)
|
|
continue
|
|
|
|
for _, obj in inspect.getmembers(module):
|
|
if (
|
|
inspect.isclass(obj)
|
|
and not inspect.isabstract(obj)
|
|
and issubclass(obj, WSRoute)
|
|
):
|
|
routes.append(obj)
|
|
|
|
return routes
|