Merge pull request jhao104#256 from nnnewb/feature/docker-enhancement

[refine] 允许 docker-compose up 直接运行服务而无需修改配置

Config/setting.py

@@ -12,12 +12,29 @@
 """
 
 # database config
+from os import getenv
+
+
+class ConfigError(BaseException):
+    pass
+
+
+DB_TYPE = getenv('db_type', 'SSDB')
+
+if DB_TYPE == 'SSDB':
+    DB_HOST = getenv('ssdb_host', '127.0.0.1')
+    DB_PORT = getenv('ssdb_port', '6379')
+elif DB_TYPE == 'MONGODB':
+    DB_HOST = getenv('mongodb_host', '127.0.0.1')
+    DB_PORT = getenv('mongodb_host', '27017')
+else:
+    raise ConfigError('Unknown database type, your environment variable `db_type` should be one of SSDB/MONGODB.')
 
 DATABASES = {
     "default": {
-        "TYPE": "SSDB",  # TYPE SSDB/MONGODB if use redis, only modify the host port, the type should be SSDB
-        "HOST": "127.0.0.1",
-        "PORT": 8888,
+        "TYPE": DB_TYPE,  # TYPE SSDB/MONGODB if use redis, only modify the host port, the type should be SSDB
+        "HOST": DB_HOST,
+        "PORT": DB_PORT,
         "NAME": "proxy",
         "PASSWORD": ""
 
@@ -45,7 +62,6 @@
     # "freeProxyWallThird"
 ]
 
-
 # # API config http://127.0.0.1:5010
 
 SERVER_API = {

Docker/Dockerfile → Dockerfile

@@ -1,13 +1,10 @@
 FROM python:3.6
-WORKDIR /usr/src/app
-COPY . .
-
 ENV DEBIAN_FRONTEND noninteractive
 ENV TZ Asia/Shanghai
-
+WORKDIR /usr/src/app
+COPY ./requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
-
+COPY . .
 EXPOSE 5010
-
 WORKDIR /usr/src/app/
 CMD [ "python", "Run/main.py" ]

docker-compose.yml

@@ -0,0 +1,14 @@
+version: '2'
+services:
+  proxy_pool:
+    build: .
+    ports:
+      - "5010:5010"
+    links:
+      - proxy_redis
+    environment:
+      db_type: SSDB
+      ssdb_host: proxy_redis
+      ssdb_port: 6379
+  proxy_redis:
+    image: "redis"

requirements.txt

@@ -3,8 +3,5 @@ werkzeug==0.11.15
 Flask==0.12
 requests==2.20.0
 lxml==3.7.2
-
 pymongo
 redis
-
-