Gateway: Add 'are we running this correctly?' check

Also some misc import order fixups
This commit is contained in:
Athanasius 2022-03-18 14:21:57 +00:00
parent fb53653430
commit 6a4bad49c8
No known key found for this signature in database
GPG Key ID: 8C392035DD80FD62

View File

@ -6,12 +6,32 @@ EDDN Gateway, which receives message from uploaders.
Contains the necessary ZeroMQ socket and a helper function to publish Contains the necessary ZeroMQ socket and a helper function to publish
market data to the Announcer daemons. market data to the Announcer daemons.
""" """
import argparse import argparse
import logging import sys
import zlib import zlib
from datetime import datetime from datetime import datetime
from typing import Dict from typing import Dict
if sys.path[0].endswith('/eddn'):
print(sys.path)
print(
'''
You're not running this script correctly.
Do not do:
python <path to>/Gateway.py <other arguments>
instead do:
cd <src directory>
python -m eddn.Gateway <other arguments>
'''
)
sys.exit(-1)
import gevent import gevent
import simplejson import simplejson
import urlparse import urlparse
@ -21,13 +41,17 @@ from pkg_resources import resource_string
from zmq import PUB as ZMQ_PUB from zmq import PUB as ZMQ_PUB
from eddn.conf.Settings import Settings, load_config from eddn.conf.Settings import Settings, load_config
from eddn.core.Validator import ValidationSeverity, Validator
from eddn.core.logger import logger from eddn.core.logger import logger
from eddn.core.Validator import ValidationSeverity, Validator
monkey.patch_all() monkey.patch_all()
import bottle # noqa: E402 import bottle # noqa: E402
from bottle import Bottle, request, response # noqa: E402 from bottle import Bottle, request, response # noqa: E402
from eddn.core.EDDNWSGIHandler import EDDNWSGIHandler # noqa: E402
# This import must be done post-monkey-patching!
from eddn.core.StatsCollector import StatsCollector # noqa: E402
bottle.BaseRequest.MEMFILE_MAX = 1024 * 1024 # 1MiB, default is/was 100KiB bottle.BaseRequest.MEMFILE_MAX = 1024 * 1024 # 1MiB, default is/was 100KiB
app = Bottle() app = Bottle()
@ -39,10 +63,6 @@ sender = zmq_context.socket(ZMQ_PUB)
validator = Validator() validator = Validator()
# This import must be done post-monkey-patching!
from eddn.core.StatsCollector import StatsCollector # noqa: E402
from eddn.core.EDDNWSGIHandler import EDDNWSGIHandler
stats_collector = StatsCollector() stats_collector = StatsCollector()
stats_collector.start() stats_collector.start()