diff options
| -rw-r--r-- | config/gunicorn.conf.py | 2 | ||||
| -rw-r--r-- | snekbox/__init__.py | 8 | ||||
| -rw-r--r-- | snekbox/__main__.py | 2 | ||||
| -rw-r--r-- | snekbox/api/resources/eval.py | 2 | ||||
| -rw-r--r-- | snekbox/nsjail.py | 2 | ||||
| -rw-r--r-- | snekbox/utils/cgroup.py | 2 | ||||
| -rw-r--r-- | snekbox/utils/swap.py | 2 |
7 files changed, 16 insertions, 4 deletions
diff --git a/config/gunicorn.conf.py b/config/gunicorn.conf.py index 8b40d35..563f8ea 100644 --- a/config/gunicorn.conf.py +++ b/config/gunicorn.conf.py @@ -3,4 +3,4 @@ bind = "0.0.0.0:8060" logger_class = "snekbox.utils.gunicorn.GunicornLogger" access_logformat = "%(m)s %(U)s%(q)s %(s)s %(b)s %(L)ss" access_logfile = "-" -wsgi_app = "snekbox.api:SnekAPI" +wsgi_app = "snekbox:SnekAPI" diff --git a/snekbox/__init__.py b/snekbox/__init__.py index 1525ebd..ccb8b11 100644 --- a/snekbox/__init__.py +++ b/snekbox/__init__.py @@ -1,8 +1,12 @@ import os -from snekbox.utils.logging import init_logger, init_sentry - DEBUG = os.environ.get("DEBUG", False) +from snekbox.api import SnekAPI # noqa: E402 +from snekbox.nsjail import NsJail # noqa: E402 +from snekbox.utils.logging import init_logger, init_sentry # noqa: E402 + +__all__ = ("NsJail", "SnekAPI") + init_sentry() init_logger(DEBUG) diff --git a/snekbox/__main__.py b/snekbox/__main__.py index 704ec9d..7ac10e9 100644 --- a/snekbox/__main__.py +++ b/snekbox/__main__.py @@ -1,7 +1,7 @@ import argparse import sys -from snekbox.nsjail import NsJail +from snekbox import NsJail def parse_args() -> argparse.Namespace: diff --git a/snekbox/api/resources/eval.py b/snekbox/api/resources/eval.py index 9560d0b..0a59f2e 100644 --- a/snekbox/api/resources/eval.py +++ b/snekbox/api/resources/eval.py @@ -5,6 +5,8 @@ from falcon.media.validators.jsonschema import validate from snekbox.nsjail import NsJail +__all__ = ("EvalResource",) + log = logging.getLogger(__name__) diff --git a/snekbox/nsjail.py b/snekbox/nsjail.py index ac36551..1aca637 100644 --- a/snekbox/nsjail.py +++ b/snekbox/nsjail.py @@ -13,6 +13,8 @@ from google.protobuf import text_format from snekbox import DEBUG, utils from snekbox.config_pb2 import NsJailConfig +__all__ = ("NsJail",) + log = logging.getLogger(__name__) # [level][timestamp][PID]? function_signature:line_no? message diff --git a/snekbox/utils/cgroup.py b/snekbox/utils/cgroup.py index 3e12406..cd515ab 100644 --- a/snekbox/utils/cgroup.py +++ b/snekbox/utils/cgroup.py @@ -5,6 +5,8 @@ from snekbox.config_pb2 import NsJailConfig log = logging.getLogger(__name__) +__all__ = ("get_version", "init", "init_v1", "init_v2") + def get_version(config: NsJailConfig) -> int: """ diff --git a/snekbox/utils/swap.py b/snekbox/utils/swap.py index 3e0d0aa..6a919cb 100644 --- a/snekbox/utils/swap.py +++ b/snekbox/utils/swap.py @@ -4,6 +4,8 @@ from pathlib import Path from snekbox.config_pb2 import NsJailConfig +__all__ = ("controller_exists", "is_enabled", "should_ignore_limit") + log = logging.getLogger(__name__) |