cleanup
This commit is contained in:
1 parent
1e53904cc6
commit
c29032a765
8 files changed
-1450
No files matched your search
@@ -1,63 +0,0 @@
|
||||
"""Config loading and mode helpers shared across monitors and the bot."""
|
||||
|
||||
import logging
|
||||
from enum import Enum
|
||||
|
||||
from monitor_utils.constants import LOGGERS
|
||||
from monitor_utils.database import get_collection
|
||||
|
||||
CONFIGS_COLLECTION_NAME = "configs"
|
||||
|
||||
|
||||
class ConfigNotFoundError(Exception):
|
||||
"""Raised when a monitor has no config document in Mongo."""
|
||||
|
||||
|
||||
class DevelopmentMode(str, Enum):
|
||||
"""Runtime mode selecting production or development behavior."""
|
||||
|
||||
PRODUCTION = "PRODUCTION"
|
||||
DEVELOPMENT = "DEVELOPMENT"
|
||||
|
||||
|
||||
def is_production(development_mode: DevelopmentMode) -> bool:
|
||||
"""Return True when the mode is PRODUCTION."""
|
||||
|
||||
return development_mode == DevelopmentMode.PRODUCTION
|
||||
|
||||
|
||||
async def get_config(monitor_name: str, use_production: bool = False) -> dict:
|
||||
"""Load a monitor's config document from Mongo.
|
||||
|
||||
Reads the configs collection in production and configs_testing in
|
||||
development, so a dev run never reads production routing.
|
||||
|
||||
Args:
|
||||
monitor_name: Monitor identifier (the config document _id).
|
||||
use_production: Read the production configs collection instead of
|
||||
configs_testing.
|
||||
|
||||
Returns:
|
||||
The config document, including its categories and channels.
|
||||
|
||||
Raises:
|
||||
ConfigNotFoundError: When no document exists for the monitor.
|
||||
"""
|
||||
|
||||
collection_name = (
|
||||
CONFIGS_COLLECTION_NAME
|
||||
if use_production
|
||||
else f"{CONFIGS_COLLECTION_NAME}_testing"
|
||||
)
|
||||
config = await get_collection(collection_name).find_one({"_id": monitor_name})
|
||||
if config is None:
|
||||
raise ConfigNotFoundError(monitor_name)
|
||||
|
||||
return config
|
||||
|
||||
|
||||
def set_logger_levels(loggers: list[str] = LOGGERS, level: int = logging.WARNING):
|
||||
"""Set the given loggers to the given level to quiet noisy libraries."""
|
||||
|
||||
for name in loggers:
|
||||
logging.getLogger(name).setLevel(level)
|
||||
Reference in new issue
Block a user