289 lines
9.8 KiB
Python
289 lines
9.8 KiB
Python
from datetime import datetime
|
|
|
|
from bson import ObjectId
|
|
|
|
from Company.Basic.Business.BusinessImpl import BusinessImpl
|
|
from Company.Basic.Change.ChangehObj import BusinessChangeObj
|
|
from Company.CompanyUtils import CompanyUtils
|
|
from DBHelper.MongoHelperInstance import DB_TYC, DB_TEST
|
|
from Utils.CommonUtil import CommonUtils
|
|
|
|
|
|
class ChangeImpl(object):
|
|
"""工商变更"""
|
|
|
|
@staticmethod
|
|
def verify_time(**kwargs):
|
|
"""校验更新时间"""
|
|
|
|
def calculate_jet_lag(begin_time):
|
|
"""计算时差"""
|
|
end_time = CommonUtils.get_current_time()
|
|
time_1_struct = datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S")
|
|
time_2_struct = datetime.strptime(begin_time, "%Y-%m-%d %H:%M:%S")
|
|
seconds = (time_1_struct - time_2_struct).seconds
|
|
if seconds > 60:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
condition = kwargs['condition']
|
|
start_time = DB_TEST.find_all_data(
|
|
'WR_DataBase_v2_test',
|
|
'C3.6_工商变更',
|
|
condition,
|
|
['更新时间']
|
|
)
|
|
if start_time:
|
|
res = calculate_jet_lag(start_time[-1]['更新时间'])
|
|
else:
|
|
res = True
|
|
return res
|
|
|
|
def auto_update(self, **kwargs):
|
|
"""新增/更新（自动）"""
|
|
condition = kwargs['condition']
|
|
|
|
def save_data(**args):
|
|
"""查询天眼查数据"""
|
|
company_name = args['cname']
|
|
company_id = args['cid']
|
|
records = DB_TYC.find_single_column(
|
|
'公司背景',
|
|
'变更记录',
|
|
{'企业名称': company_name},
|
|
'变更记录'
|
|
)
|
|
if records:
|
|
records = records['result']
|
|
list_data = list()
|
|
for record in records:
|
|
dict_data = dict()
|
|
dict_data['企业ID'] = company_id
|
|
dict_data['变更时间'] = record['changeTime']
|
|
dict_data['变更项目'] = record['changeItem']
|
|
dict_data['变更前'] = record['contentBefore']
|
|
dict_data['变更后'] = record['contentAfter']
|
|
dict_data['更新时间'] = CommonUtils.get_current_time()
|
|
list_data.append(dict_data)
|
|
return list_data
|
|
else:
|
|
return False
|
|
|
|
if len(condition['cid']):
|
|
cid = condition['cid']
|
|
cname = DB_TEST.find_single_column(
|
|
'WR_DataBase_v2_test',
|
|
'C3.1_工商信息',
|
|
{'企业ID': cid},
|
|
'企业名称'
|
|
)
|
|
# 判断更新时间
|
|
verify_res = self.verify_time(condition={"企业ID": cid})
|
|
if not verify_res:
|
|
return 'Please try again in a minute'
|
|
update = save_data(cname=cname, cid=cid)
|
|
if update:
|
|
DB_TEST.insert_many_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
update
|
|
)
|
|
return 'success'
|
|
else:
|
|
return 'update failed'
|
|
elif len(condition['cname']):
|
|
cname = condition['cname']
|
|
cid = DB_TEST.find_single_column(
|
|
'WR_DataBase_v2_test',
|
|
'C3.1_工商信息',
|
|
{'企业名称': cname},
|
|
'企业ID'
|
|
)
|
|
if not cid:
|
|
res = BusinessImpl.auto_update(condition={"cname": cname, "cid": ""})
|
|
if res == 'success':
|
|
cid = DB_TEST.find_single_column(
|
|
'WR_DataBase_v2_test',
|
|
'C3.1_工商信息',
|
|
{'企业名称': cname},
|
|
'企业ID'
|
|
)
|
|
else:
|
|
return 'new failed'
|
|
# 判断更新时间
|
|
verify_res = self.verify_time(condition={"企业ID": cid})
|
|
if not verify_res:
|
|
return 'Please try again in a minute'
|
|
insert = save_data(cname=cname, cid=cid)
|
|
if insert:
|
|
DB_TEST.insert_many_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
insert
|
|
)
|
|
return 'success'
|
|
else:
|
|
return 'new failed'
|
|
else:
|
|
return 'parameter exception'
|
|
|
|
def manu_update(self, **kwargs):
|
|
"""新增/更新（手动）"""
|
|
cid = kwargs['cid']
|
|
cname = kwargs['cname']
|
|
template = kwargs['template']
|
|
data = CompanyUtils.excel_sheet_parser(file=template)
|
|
if len(cid):
|
|
# 判断更新时间
|
|
verify_res = self.verify_time(condition={"企业ID": cid})
|
|
if not verify_res:
|
|
return 'Please try again in a minute'
|
|
if data:
|
|
data_list = list()
|
|
for item in data:
|
|
change = BusinessChangeObj()
|
|
change.cid = cid
|
|
change.change_time = CommonUtils.excel_time_stamp(item['变更时间'])
|
|
change.change_project = item['变更项目']
|
|
change.before = item['变更前']
|
|
change.after = item['变更后']
|
|
change.update_time = CommonUtils.get_current_time()
|
|
data_list.append(change.fields_toggle())
|
|
DB_TEST.insert_many_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
data_list
|
|
)
|
|
return 'success'
|
|
else:
|
|
return 'update failed'
|
|
elif len(cname):
|
|
cid = DB_TEST.find_single_column(
|
|
'WR_DataBase_v2_test',
|
|
'C3.1_工商信息',
|
|
{'企业名称': cname},
|
|
'企业ID'
|
|
)
|
|
if not cid:
|
|
res = BusinessImpl.auto_update(condition={"cname": cname, "cid": ""})
|
|
if res == 'success':
|
|
cid = DB_TEST.find_single_column(
|
|
'WR_DataBase_v2_test',
|
|
'C3.1_工商信息',
|
|
{'企业名称': cname},
|
|
'企业ID'
|
|
)
|
|
else:
|
|
return 'new failed'
|
|
verify_res = self.verify_time(condition={"企业ID": cid})
|
|
if not verify_res:
|
|
return 'Please try again in a minute'
|
|
if data:
|
|
data_list = list()
|
|
for item in data:
|
|
change = BusinessChangeObj()
|
|
change.cid = cid
|
|
change.change_time = CommonUtils.excel_time_stamp(item['变更时间'])
|
|
change.change_project = item['变更项目']
|
|
change.before = item['变更前']
|
|
change.after = item['变更后']
|
|
change.update_time = CommonUtils.get_current_time()
|
|
data_list.append(change.fields_toggle())
|
|
DB_TEST.insert_many_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
data_list
|
|
)
|
|
return 'success'
|
|
else:
|
|
return 'parameter exception'
|
|
|
|
@staticmethod
|
|
def delete(**kwargs):
|
|
"""删除"""
|
|
param = kwargs['param']
|
|
_id = param['_id']
|
|
|
|
res = DB_TEST.delete_single_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
{"_id": ObjectId(_id)}
|
|
)
|
|
|
|
if res:
|
|
result = {
|
|
"info": "success",
|
|
}
|
|
else:
|
|
result = {
|
|
"info": "no result"
|
|
}
|
|
return result
|
|
|
|
@staticmethod
|
|
def edit(**kwargs):
|
|
"""编辑"""
|
|
param = kwargs['param']
|
|
_id = param['_id']
|
|
body = param['body']
|
|
try:
|
|
change = BusinessChangeObj()
|
|
change.change_time = body['变更时间']
|
|
change.change_project = body['变更项目']
|
|
change.before = body['变更前']
|
|
change.after = body['变更后']
|
|
change.update_time = CommonUtils.get_current_time()
|
|
update_data = change.fields_toggle()
|
|
DB_TEST.update_single_data(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
{"_id": ObjectId(_id)},
|
|
update_data
|
|
)
|
|
return 'success'
|
|
except Exception:
|
|
return 'edit failed'
|
|
|
|
@staticmethod
|
|
def view(**kwargs):
|
|
"""查看"""
|
|
query = kwargs['query']
|
|
cid = query['cid']
|
|
page_num = int(query['page_num'])
|
|
page_size = int(query['page_size'])
|
|
page_size = 10 if page_size > 10 else page_size
|
|
|
|
total = DB_TEST.find_all_data_with_count(
|
|
'WR_DataBase_v2_test',
|
|
'C3.6_工商变更',
|
|
{"企业ID": cid}
|
|
)
|
|
|
|
record = DB_TEST.find_data_by_page_with_sort(
|
|
"WR_DataBase_v2_test",
|
|
"C3.6_工商变更",
|
|
{"企业ID": cid},
|
|
[],
|
|
{'_id': 1},
|
|
page_size,
|
|
page_num
|
|
)
|
|
if record:
|
|
for rec in record:
|
|
rec['_id'] = str(rec['_id'])
|
|
update_time = record[-1]['更新时间']
|
|
result = {
|
|
"info": "success",
|
|
"result": {
|
|
"total": total,
|
|
"update_time": update_time,
|
|
"records": record
|
|
}
|
|
}
|
|
else:
|
|
result = {
|
|
"info": "no result"
|
|
}
|
|
return result
|