forked from platypush/platypush
36 lines
855 B
Python
36 lines
855 B
Python
import os
|
|
import importlib
|
|
import inspect
|
|
from typing import List, Type
|
|
|
|
import pkgutil
|
|
|
|
from ._base import WSRoute, logger
|
|
|
|
|
|
def scan_routes() -> List[Type[WSRoute]]:
|
|
"""
|
|
Scans for websocket route objects.
|
|
"""
|
|
|
|
base_dir = os.path.dirname(__file__)
|
|
routes = []
|
|
|
|
for _, mod_name, _ in pkgutil.walk_packages([base_dir], prefix=__package__ + '.'):
|
|
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
|