mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
import importlib
|
|
from ipaddress import IPv4Address
|
|
|
|
from nonebot.log import logger
|
|
from nonebot.typing import Union, Optional, NoReturn
|
|
from nonebot.config import Env, Config
|
|
from nonebot.drivers import BaseDriver
|
|
|
|
_driver: Optional[BaseDriver] = None
|
|
|
|
|
|
def get_driver() -> Union[NoReturn, BaseDriver]:
|
|
if _driver is None:
|
|
raise ValueError("NoneBot has not been initialized.")
|
|
return _driver
|
|
|
|
|
|
def get_app():
|
|
driver = get_driver()
|
|
return driver.server_app
|
|
|
|
|
|
def get_asgi():
|
|
driver = get_driver()
|
|
return driver.asgi
|
|
|
|
|
|
def init(*, _env_file: Optional[str] = None, **kwargs):
|
|
global _driver
|
|
env = Env()
|
|
config = Config(**kwargs, _env_file=_env_file or f".env.{env.environment}")
|
|
|
|
logger.setLevel(logging.DEBUG if config.debug else logging.INFO)
|
|
logger.debug(f"Loaded config: {config.dict()}")
|
|
|
|
Driver = getattr(importlib.import_module(config.driver), "Driver")
|
|
_driver = Driver(env, config)
|
|
|
|
|
|
def run(host: Optional[IPv4Address] = None,
|
|
port: Optional[int] = None,
|
|
*args,
|
|
**kwargs):
|
|
get_driver().run(host, port, *args, **kwargs)
|
|
|
|
|
|
from nonebot.plugin import load_plugins, get_loaded_plugins
|