2021-12-01 14:09:33 +08:00
|
|
|
|
import time
|
|
|
|
|
from urllib import parse
|
|
|
|
|
|
|
|
|
|
import pymongo
|
|
|
|
|
|
|
|
|
|
from setting import DB_USER, DB_PASS, DB_HOST, DB_PORT
|
|
|
|
|
|
|
|
|
|
USER = parse.quote_plus(DB_USER)
|
|
|
|
|
PASSWORD = parse.quote_plus(DB_PASS)
|
|
|
|
|
|
|
|
|
|
client = pymongo.MongoClient('mongodb://%s:%s@%s:%s/' % (USER, PASSWORD, DB_HOST, DB_PORT))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def conserve_data(db, name, key, value):
|
|
|
|
|
"""
|
|
|
|
|
保存天眼查api接口数据
|
|
|
|
|
传入数据库名称,企业名称,集合名称,需要保存的数据
|
|
|
|
|
"""
|
|
|
|
|
insert_data = dict()
|
|
|
|
|
insert_data['企业名称'] = name
|
|
|
|
|
insert_data['录入时间'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
|
|
|
insert_data[key] = value
|
|
|
|
|
# 数据库操作
|
|
|
|
|
my_col = client[db][key]
|
|
|
|
|
my_col.update_one({"企业名称": name}, {"$set": insert_data}, upsert=True)
|
2021-12-01 17:05:47 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def judege_api_code(code):
|
|
|
|
|
if code == 300001:
|
|
|
|
|
return '请求失败'
|
|
|
|
|
elif code == 300002:
|
|
|
|
|
return '账号失效'
|
|
|
|
|
elif code == 300003:
|
|
|
|
|
return '账号过期'
|
|
|
|
|
elif code == 300006:
|
|
|
|
|
return '余额不足'
|
|
|
|
|
elif code == 300007:
|
|
|
|
|
return '剩余次数不足'
|
|
|
|
|
elif code == 300009:
|
|
|
|
|
return '账号信息有误'
|
|
|
|
|
elif code == 300000:
|
|
|
|
|
return '请求成功'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
res = judege_api_code(300000)
|
|
|
|
|
print(res)
|