소스 검색

合同时间提取规则优化

znj 10 달 전
부모
커밋
eb9594c219
2개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      BiddingKG/dl/interface/extract.py
  2. 2 2
      BiddingKG/dl/interface/getAttributes.py

+ 1 - 1
BiddingKG/dl/interface/extract.py

@@ -386,7 +386,7 @@ def predict(doc_id,text,title="",page_time="",web_source_no='',web_source_name="
     # predictor.getPredictor("product").predict(list_sentences, list_entitys)
     log("get product done of doc_id%s"%(doc_id))
     cost_time["product"] = round(time.time()-start_time,2)
-    prem[0].update(getAttributes.getOtherAttributes(list_entitys[0],page_time,prem))
+    prem[0].update(getAttributes.getOtherAttributes(list_entitys[0],page_time,prem,channel_dic))
 
     '''更新单一来源招标公告中标角色为预中标'''
     getAttributes.fix_single_source(prem[0], channel_dic, original_docchannel)

+ 2 - 2
BiddingKG/dl/interface/getAttributes.py

@@ -3939,7 +3939,7 @@ def extract_serviceTime(service_time,page_time):
 
     return serviceTime_dict
 
-def getOtherAttributes(list_entity,page_time,prem):
+def getOtherAttributes(list_entity,page_time,prem,channel_dic):
     dict_other = {"moneysource":"",
                   "person_review":[],
                   "serviceTime":"",
@@ -4042,7 +4042,7 @@ def getOtherAttributes(list_entity,page_time,prem):
         service_days = get_days_between(serviceTime_dict['service_start'],serviceTime_dict['service_end'])
         serviceTime_dict['service_days'] = str(service_days) + "天"
     dict_other["serviceTime"] = serviceTime_dict
-    if not time_contractEnd and prem[0]['docchannel']['docchannel']=='合同公告': # 用serviceTime补充合同开始结束时间,公告类型为合同公告
+    if not time_contractEnd and channel_dic['docchannel']['docchannel']=='合同公告': # 用serviceTime补充合同开始结束时间,公告类型为合同公告
         if serviceTime_dict['service_start'] and serviceTime_dict['service_end']:
             prem[0]["time_contractStart"] = serviceTime_dict['service_start']
             prem[0]["time_contractEnd"] = serviceTime_dict['service_end']