nonebot2/packages/nonebot-plugin-docs/nonebot_plugin_docs/__init__.py

26 lines
696 B
Python
Raw Normal View History

2020-11-21 17:20:14 +00:00
import importlib
import nonebot
from nonebot.log import logger
def init():
driver = nonebot.get_driver()
try:
_module = importlib.import_module(
f"nonebot_plugin_docs.drivers.{driver.type}")
except ImportError:
logger.warning(f"Driver {driver.type} not supported")
return
register_route = getattr(_module, "register_route")
register_route(driver)
host = str(driver.config.host)
port = driver.config.port
if host in ["0.0.0.0", "127.0.0.1"]:
host = "localhost"
logger.opt(colors=True).info(f"Nonebot docs will be running at: "
f"<b><u>http://{host}:{port}/docs/</u></b>")
init()