nonebot2/nonebot/config.py

123 lines
3.6 KiB
Python
Raw Normal View History

2020-07-04 22:51:10 +08:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from pathlib import Path
2020-08-06 17:54:55 +08:00
from datetime import timedelta
2020-07-04 22:51:10 +08:00
from ipaddress import IPv4Address
from pydantic import BaseSettings
from pydantic.env_settings import SettingsError, env_file_sentinel, read_env_file
2020-08-07 17:05:08 +08:00
from nonebot.typing import Set, Dict, Union, Mapping, Optional
class BaseConfig(BaseSettings):
def _build_environ(
self,
2020-08-13 18:16:59 +08:00
_env_file: Union[Path, str, None] = None,
_env_file_encoding: Optional[str] = None
) -> Dict[str, Optional[str]]:
"""
Build environment variables suitable for passing to the Model.
"""
d: Dict[str, Optional[str]] = {}
if self.__config__.case_sensitive:
env_vars: Mapping[str, Optional[str]] = os.environ
else:
env_vars = {k.lower(): v for k, v in os.environ.items()}
2020-08-13 18:16:59 +08:00
env_file_vars: Dict[str, Optional[str]] = {}
env_file = _env_file if _env_file != env_file_sentinel else self.__config__.env_file
2020-08-13 18:16:59 +08:00
env_file_encoding = _env_file_encoding if _env_file_encoding is not None else self.__config__.env_file_encoding
if env_file is not None:
env_path = Path(env_file)
if env_path.is_file():
env_file_vars = read_env_file(
2020-08-13 18:16:59 +08:00
env_path,
encoding=env_file_encoding,
case_sensitive=self.__config__.case_sensitive)
env_vars = {**env_file_vars, **env_vars}
for field in self.__fields__.values():
env_val: Optional[str] = None
for env_name in field.field_info.extra['env_names']:
env_val = env_vars.get(env_name)
if env_name in env_file_vars:
del env_file_vars[env_name]
if env_val is not None:
break
if env_val is None:
continue
if field.is_complex():
try:
env_val = self.__config__.json_loads(env_val)
except ValueError as e:
raise SettingsError(
f'error parsing JSON for "{env_name}"' # type: ignore
) from e
d[field.alias] = env_val
if env_file_vars:
for env_name, env_val in env_file_vars.items():
try:
env_val = self.__config__.json_loads(env_val)
except ValueError as e:
pass
d[env_name] = env_val
return d
2020-07-04 22:51:10 +08:00
class Env(BaseSettings):
environment: str = "prod"
class Config:
env_file = ".env"
class Config(BaseConfig):
"""
NoneBot Config Object
configs:
### `driver`
- 类型: `str`
- 默认值: `"nonebot.drivers.fastapi"`
- 说明:
nonebot 运行使用后端框架封装 Driver 继承自 nonebot.driver.BaseDriver
"""
2020-08-01 22:03:40 +08:00
# nonebot configs
2020-07-04 22:51:10 +08:00
driver: str = "nonebot.drivers.fastapi"
host: IPv4Address = IPv4Address("127.0.0.1")
port: int = 8080
2020-08-01 22:03:40 +08:00
secret: Optional[str] = None
2020-07-04 22:51:10 +08:00
debug: bool = False
2020-08-01 22:03:40 +08:00
# bot connection configs
2020-08-13 15:23:04 +08:00
api_root: Dict[str, str] = {}
2020-08-17 16:09:41 +08:00
api_timeout: Optional[float] = 60.
2020-08-01 22:03:40 +08:00
access_token: Optional[str] = None
# bot runtime configs
2020-07-04 22:51:10 +08:00
superusers: Set[int] = set()
nickname: Union[str, Set[str]] = ""
2020-08-17 16:09:41 +08:00
command_start: Set[str] = {"/"}
command_sep: Set[str] = {"."}
2020-08-06 17:54:55 +08:00
session_expire_timeout: timedelta = timedelta(minutes=2)
2020-07-04 22:51:10 +08:00
2020-08-01 22:03:40 +08:00
# custom configs
# custom configs can be assigned during nonebot.init
# or from env file using json loads
2020-07-04 22:51:10 +08:00
class Config:
extra = "allow"
2020-07-04 22:51:10 +08:00
env_file = ".env.prod"