witine/lookingglass
1
0
Fork 0
forked from mirrors/thatmattlove-hyperglass
Code Issues Pull requests Releases Activity

Add typing to custom WSGI app and initialize logging before startup

Browse source
This commit is contained in:
thatmattlove 2021-09-17 12:08:38 -07:00
parent 7adb6ae0ec
commit 66e69db17d
1 changed files with 18 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
hyperglass/main.py
View file

@ -80,7 +80,7 @@ def unregister_all_plugins() -> None:
manager().reset() manager().reset()
def on_starting(server: "Arbiter"): def on_starting(server: "Arbiter") -> None:
"""Gunicorn pre-start tasks.""" """Gunicorn pre-start tasks."""
python_version = platform.python_version() python_version = platform.python_version()
@ -103,7 +103,7 @@ def on_starting(server: "Arbiter"):
) )
def on_exit(server: "Arbiter"): def on_exit(*_: t.Any) -> None:
"""Gunicorn shutdown tasks.""" """Gunicorn shutdown tasks."""
log.critical("Stopping hyperglass {}", __version__) log.critical("Stopping hyperglass {}", __version__)
@ -118,13 +118,13 @@ def on_exit(server: "Arbiter"):
class HyperglassWSGI(BaseApplication): class HyperglassWSGI(BaseApplication):
"""Custom gunicorn app.""" """Custom gunicorn app."""
def __init__(self, app, options): def __init__(self: "HyperglassWSGI", app: str, options: t.Dict[str, t.Any]):
"""Initialize custom WSGI.""" """Initialize custom WSGI."""
self.application = app self.application = app
self.options = options or {} self.options = options or {}
super().__init__() super().__init__()
def load_config(self): def load_config(self: "HyperglassWSGI"):
"""Load gunicorn config.""" """Load gunicorn config."""
config = { config = {
key: value key: value
@ -135,24 +135,14 @@ class HyperglassWSGI(BaseApplication):
for key, value in config.items(): for key, value in config.items():
self.cfg.set(key.lower(), value) self.cfg.set(key.lower(), value)
def load(self): def load(self: "HyperglassWSGI"):
"""Load gunicorn app.""" """Load gunicorn app."""
return self.application return self.application
def start(**kwargs): def start(*, log_level: str, workers: int, **kwargs) -> None:
"""Start hyperglass via gunicorn.""" """Start hyperglass via gunicorn."""
set_log_level(log, Settings.debug)
log.debug("System settings: {!r}", Settings)
workers, log_level = 1, "DEBUG"
if Settings.debug is False:
workers, log_level = cpu_count(2), "WARNING"
setup_lib_logging(log_level)
HyperglassWSGI( HyperglassWSGI(
app="hyperglass.api:app", app="hyperglass.api:app",
options={ options={
@ -175,7 +165,18 @@ def start(**kwargs):
if __name__ == "__main__": if __name__ == "__main__":
try: try:
start() set_log_level(log, Settings.debug)
log.debug("System settings: {!r}", Settings)
workers, log_level = 1, "DEBUG"
if Settings.debug is False:
workers, log_level = cpu_count(2), "WARNING"
setup_lib_logging(log_level)
start(log_level=log_level, workers=workers)
except Exception as error: except Exception as error:
# Handle app exceptions. # Handle app exceptions.
if not Settings.dev_mode: if not Settings.dev_mode: