This commit is contained in:
Marvin Zhang
2019-06-29 11:52:41 +08:00
parent 9f333f025c
commit d66f4d4b39
2 changed files with 7 additions and 3 deletions

View File

@@ -2,6 +2,7 @@ import os
import sys
from datetime import datetime
from time import sleep
import traceback
from bson import ObjectId
from pymongo import ASCENDING, DESCENDING
@@ -213,8 +214,10 @@ def execute_config_spider(self, id: str, params: str = None):
env['MONGO_HOST'] = MONGO_HOST
env['MONGO_PORT'] = str(MONGO_PORT)
env['MONGO_DB'] = MONGO_DB
env['MONGO_USERNAME'] = MONGO_USERNAME
env['MONGO_PASSWORD'] = MONGO_PASSWORD
if MONGO_USERNAME is not None:
env['MONGO_USERNAME'] = MONGO_USERNAME
if MONGO_PASSWORD:
env['MONGO_PASSWORD'] = MONGO_PASSWORD
cmd_arr = [
sys.executable,
@@ -246,6 +249,7 @@ def execute_config_spider(self, id: str, params: str = None):
else:
status = TaskStatus.FAILURE
except Exception as err:
traceback.print_exc()
logger.error(err)
stderr.write(str(err))
status = TaskStatus.FAILURE

View File

@@ -12,7 +12,7 @@ from pymongo import MongoClient
class XueqiuPipeline(object):
mongo = MongoClient(
host=os.environ.get('MONGO_HOST') or 'localhost',
port=int(os.environ.get('MONGO_PORT')) or 27017
port=int(os.environ.get('MONGO_PORT') or 27017)
)
db = mongo[os.environ.get('MONGO_DB') or 'crawlab_test']
col = db.get_collection(os.environ.get('CRAWLAB_COLLECTION') or 'results_xueqiu')