Germey 3 tahun lalu
induk
melakukan
20d16f97b4
2 mengubah file dengan 8 tambahan dan 5 penghapusan
  1. 4 4
      proxypool/scheduler.py
  2. 4 1
      proxypool/setting.py

+ 4 - 4
proxypool/scheduler.py

@@ -3,7 +3,7 @@ import multiprocessing
 from proxypool.processors.server import app
 from proxypool.processors.getter import Getter
 from proxypool.processors.tester import Tester
-from proxypool.setting import CYCLE_GETTER, CYCLE_TESTER, API_HOST, \
+from proxypool.setting import APP_PROD_METHOD_GEVENT, APP_PROD_METHOD_MEINHELD, APP_PROD_METHOD_TORNADO, CYCLE_GETTER, CYCLE_TESTER, API_HOST, \
     API_THREADED, API_PORT, ENABLE_SERVER, IS_PROD, APP_PROD_METHOD, \
     ENABLE_GETTER, ENABLE_TESTER, IS_WINDOWS
 from loguru import logger
@@ -58,7 +58,7 @@ class Scheduler():
             logger.info('server not enabled, exit')
             return
         if IS_PROD:
-            if APP_PROD_METHOD == 'gevent':
+            if APP_PROD_METHOD == APP_PROD_METHOD_GEVENT:
                 try:
                     from gevent.pywsgi import WSGIServer
                 except ImportError as e:
@@ -67,7 +67,7 @@ class Scheduler():
                     http_server = WSGIServer((API_HOST, API_PORT), app)
                     http_server.serve_forever()
 
-            elif APP_PROD_METHOD == 'tornado':
+            elif APP_PROD_METHOD == APP_PROD_METHOD_TORNADO:
                 try:
                     from tornado.wsgi import WSGIContainer
                     from tornado.httpserver import HTTPServer
@@ -79,7 +79,7 @@ class Scheduler():
                     http_server.listen(API_PORT)
                     IOLoop.instance().start()
 
-            elif APP_PROD_METHOD == "meinheld":
+            elif APP_PROD_METHOD == APP_PROD_METHOD_MEINHELD:
                 try:
                     import meinheld
                 except ImportError as e:

+ 4 - 1
proxypool/setting.py

@@ -26,7 +26,10 @@ APP_TEST = IS_TEST = APP_ENV == TEST_MODE
 # - gevent: pip install gevent
 # - tornado: pip install tornado
 # - meinheld: pip install meinheld
-APP_PROD_METHOD = env.str('APP_PROD_METHOD', "gevent").lower()
+APP_PROD_METHOD_GEVENT = 'gevent'
+APP_PROD_METHOD_TORNADO = 'tornado'
+APP_PROD_METHOD_MEINHELD = 'meinheld'
+APP_PROD_METHOD = env.str('APP_PROD_METHOD', APP_PROD_METHOD_GEVENT).lower()
 
 # redis host
 REDIS_HOST = env.str('PROXYPOOL_REDIS_HOST',