resume_parse.py 64 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529
  1. #!/usr/bin/env python
  2. # coding: utf-8
  3. import os
  4. import sys
  5. import re
  6. import json
  7. import time
  8. from os import walk
  9. import subprocess
  10. import rarfile
  11. import py7zr
  12. import tarfile
  13. from zipfile import ZipFile
  14. import requests
  15. from requests.adapters import HTTPAdapter
  16. import logging
  17. from docx import Document
  18. from docx.shared import Inches
  19. from pdfminer.pdfpage import PDFPage
  20. from pdfminer.pdfparser import PDFParser
  21. from pdfminer.pdfdocument import PDFDocument
  22. from pdfminer.high_level import extract_pages
  23. from pdfminer.converter import PDFPageAggregator
  24. from pdfminer.layout import LTTextContainer, LTChar, LTLine, LAParams, LTTextBox, LTFigure, LTImage, LTText, LTAnno, LTTextLine, LTTextLineHorizontal
  25. from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
  26. import pdfplumber
  27. from paddlenlp import Taskflow
  28. class Logger:
  29. def __init__(self, name: str, console_handler_level: str = logging.INFO, fmt: str = '%(asctime)s: %(name)s: %(levelname)s: %(filename)s: %(funcName)s: %(message)s'):
  30. self.logger = logging.getLogger(name)
  31. self.logger.setLevel(logging.INFO)
  32. self.fmt = logging.Formatter(fmt)
  33. self.set_console_handler(console_handler_level)
  34. def set_console_handler(self, console_handler_level: str = logging.INFO) -> None:
  35. ch = logging.StreamHandler()
  36. ch.setLevel(console_handler_level)
  37. ch.setFormatter(self.fmt)
  38. self.logger.addHandler(ch)
  39. def set_file_handler(self, filename: str, mode: str = "a", file_handler_level: str = logging.WARNING) -> None:
  40. fh = logging.FileHandler(filename, mode=mode, encoding='utf-8')
  41. fh.setLevel(file_handler_level)
  42. fh.setFormatter(self.fmt)
  43. self.logger.addHandler(fh)
  44. def debug(self, msg):
  45. self.logger.debug(msg)
  46. def info(self, msg):
  47. self.logger.info(msg)
  48. def warning(self, msg):
  49. self.logger.warning(msg)
  50. def error(self, msg):
  51. self.logger.error(msg)
  52. def critical(self, msg):
  53. self.logger.critical(msg)
  54. logger = Logger("resume_parse")
  55. logger.set_file_handler(filename='data.log')
  56. from rich.console import Console
  57. console = Console()
  58. import uvicorn
  59. from fastapi import BackgroundTasks, FastAPI, File, UploadFile
  60. app = FastAPI()
  61. ner = Taskflow("ner", mode='fast')
  62. ner_tag = Taskflow("ner")
  63. base_info_ie = Taskflow('information_extraction', schema=["姓名","性别","婚姻状况","电子邮箱","政治面貌","手机号码","籍贯","出生日期","现任职务","参加工作时间","英语水平","计算机水平","工作年限","当前单位","所在城市","职业资格"])
  64. prize_ie = Taskflow('information_extraction', schema=["时间", "奖项"])
  65. cet_ie = Taskflow('information_extraction', schema=["时间","证书"])
  66. pro_ie = Taskflow("information_extraction", schema=["时间","项目名称","机构","职位"], task_path='./model_100')
  67. global block, block_rev
  68. with open("resources/SegmentName.json", "r", encoding="utf-8") as fp:
  69. block = json.load(fp)
  70. block_rev = {1:"基本信息", 2:"求职意向", 3:"教育经历", 4:"工作经历", 5:"项目经历", 6:"专业技能", 7:"自我评价", 8:"兴趣爱好", 9:"语言能力", 10:"证书", 11:"获奖情况", 12:"培训经历", 13:"家庭成员", "other":"其他"}
  71. # 基本信息(旧版)
  72. def get_base_info_old(lines):
  73. logger.info(lines)
  74. schema = {
  75. '姓名': None,
  76. }
  77. for line in [' '.join(' '.join(lines).split('\n'))]:
  78. line = line.replace(r'[ ]{5,}','\n')
  79. w = re.sub(r'[\W]+(\w[::])[\W]{0,}\w', r'\1', line)
  80. for i in w.split():
  81. if ':' in i:
  82. try:
  83. key, val = i.split(':')
  84. schema[key] = val
  85. except Exception as e:
  86. logger.error(e)
  87. if not schema.get('姓名'):
  88. schema['姓名'] = re.search(r'[姓名::]{3,}(\w{2,4})', w).group(1) if re.search(r'[姓名::]{3,}(\w{2,4})', w) else None
  89. if not schema.get('姓名'):
  90. for word, tag in ner_tag(w):
  91. if tag == "人物类_实体":
  92. schema['姓名'] = word
  93. if not schema.get('性别'):
  94. schema['性别'] = re.search(r'[男女]', w).group() if re.search(r'[男女]', w) else None
  95. if not schema.get('婚姻状况'):
  96. schema['婚姻状况'] = re.search(r'[已未]婚', w).group() if re.search(r'[已未]婚', w) else None
  97. if not schema.get('电子邮箱'):
  98. schema['电子邮箱'] = re.search(r'([.\w]+@[.\w]+)', w).group() if re.search(r'([.\w]+@[.\w]+)', w) else None
  99. if not schema.get('政治面貌'):
  100. schema['政治面貌'] = re.search(r'[预备中共党团员群众无派人士]{2,6}', w).group() if re.search(r'[预备中共党团员群众无派人士]{2,6}', w) else None
  101. if not schema.get('手机号码'):
  102. schema['手机号码'] = re.search(r'\W(1[\d]{10})\W', w).group(1) if re.search(r'\W(1[\d]{10})\W', w) else None
  103. # if not schema.get('籍贯'):
  104. # schema['籍贯'] = re.search(r'[籍贯::]{3,}(\w{2,5})', w).group(1) if re.search(r'[籍贯::]{3,}(\w{2,})', w) else None
  105. if not schema.get('出生年月'):
  106. schema['出生年月'] = re.search(r'\d{4}[./年\-]\d{1,2}[月]', w).group() if re.search(r'\d{4}[./年\-]\d{1,2}[月]', w) else None
  107. # if not schema.get('当前职位'):
  108. # schema['当前职位'] = re.search(r'[当前职位: ]{3,}(\w)+', w).group() if re.search(r'[当前职位: ]{3,}(\w)+', w) else None
  109. # if not schema.get('参加工作时间'):
  110. # schema['参加工作时间'] = re.search(r'[参加工作时间:]{3,}(\d{4}[./年\-]\d{1,2}[月])', w).group(1) if re.search(r'[参加工作时间:]{3,}(\d{4}[./年\-]\d{1,2}[月])', w) else None
  111. return {key:value for key, value in schema.items() if value}
  112. # 基本信息(OIE 已完成)
  113. def get_base_info(lines):
  114. if not lines:
  115. return
  116. logger.info(lines)
  117. data = " ".join(lines)
  118. rst = base_info_ie(data)[0]
  119. if rst.get("出生日期"):
  120. dates = re.findall(r'\d+' ,rst["出生日期"][0]["text"])
  121. if len(dates) == 1:
  122. rst["出生日期"][0]["text"] = "{:4d}-01-01".format(int(dates[0]))
  123. elif len(dates) == 2:
  124. rst["出生日期"][0]["text"] = "{:4d}-{:02d}-01".format(int(dates[0]), int(dates[1]))
  125. elif len(dates) == 3:
  126. rst["出生日期"][0]["text"] = "{:4d}-{:02d}-01".format(int(dates[0]), int(dates[1]))
  127. if rst.get("参加工作时间"):
  128. dates = re.findall(r'\d+' ,rst["参加工作时间"][0]["text"])
  129. if len(dates) == 1:
  130. rst["参加工作时间"][0]["text"] = "{:4d}-01-01".format(int(dates[0]))
  131. elif len(dates) == 2:
  132. rst["参加工作时间"][0]["text"] = "{:4d}-{:02d}-01".format(int(dates[0]), int(dates[1]))
  133. elif len(dates) == 3:
  134. rst["参加工作时间"][0]["text"] = "{:4d}-{:02d}-01".format(int(dates[0]), int(dates[1]))
  135. return {key:rst[key][0]["text"] for key in rst.keys()}
  136. # 求职意向(已完成)
  137. def get_job_intention(lines):
  138. logger.info(lines)
  139. schema = {}
  140. for line in lines:
  141. regex = re.compile(r'\W{0,3}[::]\s+')
  142. line = regex.sub(':', line)
  143. for i in line.split():
  144. if ":" in i:
  145. try:
  146. key, val = i.split(":")
  147. schema[key] = val
  148. except Exception as e:
  149. logger.error(e)
  150. return schema
  151. # 教育经历 (已停用)
  152. # ner + 分词 (判断学校,时间,学历) 专业需要单独处理。
  153. def get_edu_list_old(lines):
  154. logger.info(lines)
  155. job_list = []
  156. job_dict = {'edu_time_beg':'', 'edu_time_end':'', 'edu_name':'','edu_leval':'','edu_domain':'', 'edu_statue':0}
  157. re_txt = '\d{4,4}.\d{1,2}.?\040{0,2}[\-–至-\—~]\040{0,2}\d{4,4}.\d{1,2}[月]?|\d+\.\d+\-至今|\d+年\d+月\-\d+年\d+月|\d+年\d+月\-\~|\d+年\d+月[\-\~]至今|\d+-\d+\040{0,2}[\~至]\040{0,2}\d+-\d+|\d+-\d+\~|\d+-\d+\~至今|\d+-\d+\040{0,2}至今|^\d{4,4}.\d{1,2}|19\d{2,2}.|20\d{2,2}.'
  158. re_txt_1 = '\d{4,4}.\d{1,2}.?\040{0,2}[\-–至-\—~]\040{0,2}\d{4,4}.\d{1,2}[月]?|\d+\.\d+\-至今|\d+年\d+月\-\d+年\d+月|\d+年\d+月\-\~|\d+年\d+月[\-\~]至今|\d+-\d+\040{0,2}[\~至]\040{0,2}\d+-\d+|\d+-\d+\~|\d+-\d+\~至今|\d+-\d+\040{0,2}至今'
  159. nums = []
  160. for i in range(len(lines)):
  161. if re.findall(re_txt, lines[i]):
  162. nums.append(i)
  163. nums.append(len(lines))
  164. edu_level = {'本科':18, "大专":17, "博士研究生":20, "学士":18, "博士":20, "硕士":19, "研究生":19, "博后":21, '博士后':21}
  165. year_dict = {18:4, 17:3,20:3,19:3,21:2}
  166. edu_dict = {18:'本科', 17:'大专',20:'博士研究生',19:'硕士',21:'博士后'}
  167. edu_list = []
  168. for i in range(1, len(nums[:])):
  169. job_dict = {'edu_time_beg':'', 'edu_time_end':'', 'edu_name':'','edu_leval':'','edu_domain':''}
  170. data_list = lines[nums[i-1]:nums[i]]
  171. if len(data_list) > 1 and data_list[1] and data_list[1][-1] == '|' and data_list[0][-1] != '|':
  172. data_list[0] = data_list[0] + data_list[1]
  173. data_list[1] = ''
  174. if len(data_list) > 2 and data_list[2] and data_list[2][-1] == '|' and data_list[0][-1] != '|' and '|' in str(data_list[0]) and data_list[1] and data_list[1][-1] != '|':
  175. data_list[0] = data_list[0] + data_list[1] + data_list[2]
  176. data_list[1] = ''
  177. data_list[2] = ''
  178. if '' in data_list:
  179. data_list.remove('')
  180. data_line = ' '.join(data_list)
  181. data_line = re.sub('[\|]', ' ', data_line)
  182. data_line = re.sub('-{3,}', '', data_line)
  183. ner_data = ner(''.join(data_list[:2]))
  184. org = ''
  185. time_list = []
  186. for jj in range(1, len(ner_data)):
  187. if ner_data[jj][1] == ner_data[jj-1][1]:
  188. ner_data[jj] = list(ner_data[jj])
  189. ner_data[jj][0] = ner_data[jj-1][0] + ner_data[jj][0]
  190. ner_data[jj-1] = ('','')
  191. for _ in ner_data:
  192. if _[1] == 'ORG' and not org:
  193. org = _[0].strip()
  194. elif _[1] == 'TIME' and len(_[1]) >= 4:
  195. time_list.append(_[0])
  196. #TIME
  197. # print(data_line)
  198. _list_data = re.split('\040+',data_line)
  199. top_level = 18
  200. remove_list = []
  201. logger.info(_list_data)
  202. logger.info(time_list)
  203. for ii in range(len(_list_data)):
  204. for t in time_list:
  205. if t in _list_data[ii]:
  206. _list_data[ii] = ''
  207. break
  208. for i in range(len(_list_data)):
  209. #if org in _list_data[i]:
  210. # _list_data[i] = ''
  211. if re.findall('^\d{4,4}', _list_data[i]):
  212. _list_data[i] = ''
  213. _data = re.findall('本科|学士|硕士|博士研究生|博士后|博后|博士|研究生|大专', _list_data[i])
  214. if not _data:
  215. continue
  216. top_level = edu_level[_data[0]]
  217. _list_data[i] = ''
  218. break
  219. #remove_list.append(i)
  220. logger.info(_list_data)
  221. job_time = re.findall(re_txt_1, data_list[0])
  222. if job_time:
  223. job_dict['edu_time'] = job_time[0]
  224. else:
  225. job_dict['edu_time'] = ''
  226. _nums = re.findall('\d+', job_dict['edu_time'])
  227. if len(_nums) >= 4:
  228. job_dict['edu_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  229. job_dict['edu_time_end'] = '%s-%02d'%(_nums[2], int(_nums[3]))
  230. job_dict['edu_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  231. elif len(_nums) == 2:
  232. job_dict['edu_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  233. job_dict['edu_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  234. job_dict['edu_time_end'] = '%s'%('至今')
  235. elif len(time_list) == 2:
  236. nums_1 = re.findall('\d+', time_list[0])
  237. nums_2 = re.findall('\d+', time_list[1])
  238. nums_1.append('09')
  239. nums_2.append('07')
  240. job_dict['edu_time_beg'] = '%s-%02d'%(nums_1[0], int(nums_1[1]))
  241. try:
  242. job_dict['edu_time_end'] = '%s-%02d'%(nums_2[0], int(nums_2[1]))
  243. except:
  244. job_dict['edu_time_end'] = None
  245. try:
  246. job_dict['edu_time'] = '%s-%02d~%s-%02d'%(nums_1[0], int(nums_1[1]), nums_2[0], int(nums_2[1]))
  247. except:
  248. job_dict['edu_time'] = '%s-%02d~今'%(nums_1[0], int(nums_1[1]))
  249. elif len(time_list) == 1:
  250. _nums = re.findall('\d+', time_list[0])
  251. if '毕业' in data_list[0]:
  252. _nums.append('06')
  253. _nums.insert(0, '09')
  254. _nums.insert(0, str(int(_nums[1]) - year_dict[top_level]))
  255. job_dict['edu_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  256. job_dict['edu_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  257. job_dict['edu_time_end'] = '%s-%02d'%(_nums[2], int(_nums[3]))
  258. else:
  259. _nums.append('09')
  260. job_dict['edu_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  261. job_dict['edu_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  262. job_dict['edu_time_end'] = '%s'%('至今')
  263. job_dict['edu_leval'] = edu_dict[top_level]
  264. if org:
  265. job_dict['edu_name'] = org
  266. else:
  267. job_dict['edu_name'] = ''
  268. edu_domain = ''
  269. for i in range(len(_list_data)):
  270. if org in _list_data[i]:
  271. continue
  272. if not _list_data[i] and '专业' in _list_data[i]:
  273. edu_domain = _list_data[i]
  274. if not edu_domain:
  275. for i in range(len(_list_data)):
  276. if org in _list_data[i]:
  277. continue
  278. if _list_data[i] and len(_list_data[i]) >= 3:
  279. edu_domain = _list_data[i]
  280. break
  281. if not edu_domain:
  282. for i in range(len(_list_data)):
  283. if org in _list_data[i]:
  284. for j in range(i+1, len(_list_data)):
  285. if _list_data[i] and len(_list_data[j]) >= 2:
  286. edu_domain = _list_data[j]
  287. break
  288. break
  289. job_dict['edu_domain'] = edu_domain
  290. if len(job_list) ==0:
  291. job_list.append(job_dict)
  292. else:
  293. if job_dict in job_list:
  294. continue
  295. if not job_dict['edu_time']:
  296. continue
  297. if int(job_dict['edu_time'][:4]) > int(job_list[-1]['edu_time'][:4]):
  298. job_list = [job_dict] + job_list
  299. else:
  300. job_list.append(job_dict)
  301. continue
  302. data_list[0] = re.sub(job_time[0], '', data_list[0])
  303. _list = re.split('\|\040+', data_list[0])
  304. #print(_list)
  305. if len(_list) == 1:
  306. __list = re.split('\040+', data_list[0])
  307. job_dict['edu_name'] = __list[1].strip()
  308. job_dict['edu_domain'] = __list[2].strip()
  309. job_dict['edu_leval'] = __list[3].strip()
  310. else:
  311. #if job_dict['edu_leval'] not in
  312. if len(_list) > 3:
  313. job_dict['edu_name'] = _list[2].strip()
  314. job_dict['edu_domain'] = _list[3].strip()
  315. job_dict['edu_leval'] = _list[1].strip()
  316. else:
  317. job_dict['edu_leval'] = _list[0].strip()
  318. job_dict['edu_name'] = _list[1].strip()
  319. job_dict['edu_domain'] = _list[2].strip()
  320. if '硕士' in _list[0] or '研究生' in _list[0]:
  321. job_dict['edu_leval'] = '硕士'
  322. elif '博士' in _list[0]:
  323. job_dict['edu_leval'] = '博士'
  324. elif '本科' in _list[0]:
  325. job_dict['edu_leval'] = '本科'
  326. elif '学士' in _list[0]:
  327. job_dict['edu_leval'] = '本科'
  328. # print(job_dict)
  329. if len(job_list) ==0:
  330. job_list.append(job_dict)
  331. else:
  332. if job_dict in job_list:
  333. continue
  334. if int(job_dict['edu_time'][:4]) > int(job_list[-1]['edu_time'][:4]):
  335. job_list = [job_dict] + job_list
  336. else:
  337. job_list.append(job_dict)
  338. #edu_list.append(job_dict['edu_time'] + job_dict['edu_name'] + job_dict['edu_domain'] + job_dict['edu_leval'])
  339. #if job_list[0]['edu_leval'] not in ['硕士', '博士', '本科', '博后'] and len(job_list[0]['edu_leval']) > 5:
  340. # job_list[0]['edu_leval'] = '本科'
  341. return job_list
  342. # 教育经历改 (已完成)
  343. def get_edu_list(lines):
  344. logger.info(lines)
  345. edu_list = [{"Time":None, "startTime":None, "endTime":None, "edu_name":None, "edu_domain":None, "edu_level":None}]
  346. regex_time = re.compile(r'((\d{4})[年\W]{1,2}(\d{1,2})[月\W]?[\d]{0,2})[至到\W]+((\d{4})[年\W]{1,2}(\d{1,2})[月\W]?)?([今])?|(\d{4})[至\W]+([\d今]{4})')
  347. regex_end = re.compile(r'毕业时间[\w\W]{0,5}(\d{4})[\W年]?(\d{0,2})[月\W]?')
  348. regex_level = re.compile(r'[大本专科硕博士研究生后]{2,}')
  349. regex_domain = re.compile(u'[\u4E00-\u9FA5]{2,10}', re.UNICODE)
  350. count = 0
  351. for line in lines:
  352. line = line.replace("学士","本科").replace("专业","").replace("学位","")
  353. for cell in re.split(r'[·\|\t]', line):
  354. if not cell.strip():
  355. continue
  356. flags = 0
  357. edu_time = regex_time.search(cell)
  358. edu_end_time = regex_end.search(cell)
  359. edu_level = regex_level.search(cell)
  360. edu_domain = regex_domain.search(cell)
  361. # 标准时间格式
  362. if edu_time:
  363. # 提交信息
  364. if edu_list[count].get("Time") and edu_list[count].get("edu_name"):
  365. edu_list.append({"Time":None, "startTime":None, "endTime":None, "edu_name":None, "edu_domain":None, "edu_level":None})
  366. count += 1
  367. edu_list[count]["startTime"] = '{:4d}-{:02d}'.format(int(edu_time.group(2)),int(edu_time.group(3)))
  368. # 年月日
  369. if edu_time.group(5) != None:
  370. edu_list[count]["endTime"] = '{:4d}-{:02d}'.format(int(edu_time.group(5)),int(edu_time.group(6)))
  371. edu_list[count]["Time"] = '{:4d}-{:02d}~{:4d}-{:02d}'.format(int(edu_time.group(2)),int(edu_time.group(3)),int(edu_time.group(5)),int(edu_time.group(6)))
  372. # 只有年
  373. elif edu_time.group(8) != None:
  374. edu_list[count]["Time"] = '{:4d}~{:4d}'.format(int(edu_time.group(8)),int(edu_time.group(9)))
  375. edu_list[count]["startTime"] = '{:4d}'.format(int(edu_time.group(8)))
  376. edu_list[count]["endTime"] = '{:4d}'.format(int(edu_time.group(9)))
  377. # 至今类
  378. else:
  379. edu_list[count]["endTime"] = edu_time.group(7)
  380. edu_list[count]['Time'] = '{:4d}-{:02d}~{}'.format(int(edu_time.group(2)),int(edu_time.group(3)),edu_time.group(7))
  381. flags = 1
  382. # 只有毕业时间
  383. elif edu_end_time:
  384. # 提交信息
  385. if edu_list[count].get("endTime") and edu_list[count].get("edu_name"):
  386. edu_list.append({"Time":None, "startTime":None, "endTime":None, "edu_name":None, "edu_domain":None, "edu_level":None})
  387. count += 1
  388. # 年月
  389. if edu_end_time.group(2):
  390. edu_list[count]["Time"] = '{:4d}-{:02d}~{:4d}-{:02d}'.format(int(edu_end_time.group(1))-3,int(edu_end_time.group(2)),int(edu_end_time.group(1)),int(edu_end_time.group(2)))
  391. edu_list[count]["endTime"] = '{:4d}-{:02d}'.format(int(edu_end_time.group(1)),int(edu_end_time.group(2)))
  392. # 只有年
  393. elif edu_end_time.group(1):
  394. edu_list[count]["Time"] = '{:4d}~{:4d}'.format(int(edu_end_time.group(1))-3,int(edu_end_time.group(1)))
  395. edu_list[count]["endTime"] = '{:4d}'.format(int(edu_end_time.group(1)))
  396. # 学历
  397. if (not edu_list[count].get("edu_level")) and edu_level:
  398. edu_list[count]["edu_level"] = edu_level.group(0)
  399. # WordTag 识别 学校/专业
  400. for word, tag in ner_tag(cell):
  401. if (not edu_list[count].get("edu_name")) and (tag == "组织机构类_教育组织机构"):
  402. edu_list[count]["edu_name"] = word.strip()
  403. flags = 1
  404. elif (not edu_list[count].get("edu_domain")) and (tag in "_术语类型"):
  405. edu_list[count]["edu_domain"] = word.strip()
  406. elif edu_list[count].get("edu_name") and edu_list[count].get("edu_domain"):
  407. break
  408. # LAC 识别 学校
  409. else:
  410. for word, tag in ner(cell):
  411. if (tag == "ORG"):
  412. edu_list[count]["edu_name"] = word
  413. flags = 1
  414. break
  415. # 未识别成功时填充专业
  416. if (not (edu_level or flags or edu_list[count].get("edu_domain"))) and edu_domain:
  417. edu_list[count]["edu_domain"] = edu_domain.group(0)
  418. # 剔除时间不存在、学校不存在的列
  419. if (not edu_list[-1].get("Time")) or (not edu_list[-1].get("edu_name")):
  420. edu_list.pop()
  421. return edu_list
  422. # 工作经历 (已完成)
  423. # ner + 分词 机构信息,人物身份信息,时间 工作内容区分判断
  424. # 其中,时间是判断是否下一份工作情况的主要标识符之一。字符数量
  425. # 时间类 数量词
  426. def get_job_list(lines):
  427. logger.info(lines)
  428. job_list = []
  429. re_txt = '\d{4,4}\040{0,2}.\d+\040{0,2}.?\040{0,2}[\-–至-\—~]{1,2}\040{0,2}\d{4,4}\040{0,2}.\040{0,2}\d+.?|\d{4,4}.\d+.?\040{0,2}[\-–-—]{0,2}\040{0,2}至?今|\d{4,4}.\d+.?\040{0,2}[\-–-]{1,2}\040{0,2}现在|\d{4,4}年\d+月\-\d{4,4}年\d+月|\d{4,4}年\d+月\-\~|\d{4,4}年\d+月[\-\~-]至今|\d{4,4}-\d+\040{0,2}[-\~至]\040{0,2}\d{4,4}-\d+|\d{4,4}-\d+\~|\d{4,4}-\d+\[~-]至今|\d{4,4}-\d+\040{0,2}至今'
  430. nums = []
  431. for i in range(len(lines)):
  432. #print(lines[i])
  433. #print(lines[i], re.findall(re_txt, lines[i]), re.findall('\||\040{1,}', lines[i]))
  434. if re.findall(re_txt, lines[i].replace(' ', '')) and re.findall('\||\040{1,}', lines[i]):
  435. nums.append(i)
  436. continue
  437. if re.findall(re_txt, lines[i].replace(' ', '')[:20]):
  438. nums.append(i)
  439. continue
  440. if len(lines[i].strip().replace(' ', '')) > 50:
  441. continue
  442. year_list = re.findall('19\d{2,2}.\d{1,2}|20\d{2,2}.\d{1,2}', lines[i])
  443. if len(year_list) >= 2:
  444. nums.append(i)
  445. elif len(year_list) == 1 and '至今' in lines[i]:
  446. nums.append(i)
  447. nums.append(len(lines))
  448. # logger.info(nums)
  449. logger.info('get_job_list :{}'.format(nums))
  450. for i in range(1, len(nums[:])):
  451. job_dict = {'job_time':'', 'job_leval':'','job_company':'','job_content':''}
  452. data_list = lines[nums[i-1]:nums[i]]
  453. if '' in data_list:
  454. data_list.remove('')
  455. org = ''
  456. person_professor_list = []
  457. org_index = -1
  458. end_index = 3
  459. job_time = re.findall(re_txt, data_list[0])
  460. if not job_time:
  461. year_list = re.findall('19\d{2,2}.\d{1,2}|20\d{2,2}.\d{1,2}', data_list[0])
  462. if len(year_list) >= 2:
  463. job_time = ['-'.join(year_list)]
  464. elif len(year_list) == 1 and '至今' in lines[i]:
  465. job_time = [year_list[0] + '-' + '至今']
  466. if not job_time:
  467. regex = re.compile(r'((\d{4})[年\W]+(\d{1,2})[\W]?[\w]?)[至到\W]+((\d{4})[年\W]+(\d{1,2})[\W]?[\w]?)?([今])?')
  468. job_time = [re.search(regex, data_list[0]).group(0)]
  469. job_dict['job_time'] = job_time[0]
  470. _nums = re.findall('\d+', job_dict['job_time'])
  471. #print(_nums)
  472. if len(_nums) >= 4:
  473. job_dict['job_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  474. elif len(_nums) == 2:
  475. job_dict['job_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  476. if re.findall('\d', job_time[0]):
  477. job_time[0] = job_time[0][:-1]
  478. data_list[0] = re.sub(job_time[0], '', data_list[0])
  479. data_list[0] = data_list[0].strip()
  480. ner_list = []
  481. for i in range(len(data_list[:3])):
  482. if '工作' in data_list[i][:4] and (re.findall(':|\:', data_list[i])):
  483. end_index = i
  484. break
  485. if not re.findall('\040|\||/', data_list[i]) and org:
  486. end_index = i
  487. break
  488. if len(data_list[i]) > 80:
  489. end_index = i
  490. break
  491. if data_list[i]:
  492. ner_data = ner_tag(data_list[i].strip())
  493. else:
  494. continue
  495. ner_list.append(ner_data)
  496. for x in ner_data:
  497. if x[1] == '人物类_概念' and len(x[0]) > 2:
  498. person_professor_list.append(x[0].strip())
  499. elif x[1] == '组织机构类_企事业单位' or x[1] == '组织机构类_教育组织机构':
  500. if not org:
  501. org = re.split('\040|\|/', x[0].strip())[0]
  502. org_index = i
  503. if not org:
  504. for i in range(len(ner_list)):
  505. ner_data = ner_list[i]
  506. for x in ner_data:
  507. if x[1] == '组织机构类':
  508. org = re.split('\040|\|/', x[0].strip())[0]
  509. break
  510. if not person_professor_list:
  511. for i in range(len(ner_list)):
  512. ner_data = ner_list[i]
  513. for x in ner_data:
  514. if x[1] == '人物类_概念':
  515. person_professor_list = [re.split('\040|\|/', x[0].strip())[0]]
  516. break
  517. data_line = ' '.join(data_list[:end_index])
  518. data_line = re.sub('\||/', ' ', data_line)
  519. _list_data = re.split('\040+',data_line)
  520. if len(_list_data) == 1:
  521. end_index = 0
  522. if not person_professor_list:
  523. for x in range(len(_list_data)):
  524. if re.findall('经理|工程师|会计|董事长|总监|秘书|主管|处长|局长|主任|讲师|教授', _list_data[x][-4:]):
  525. person_professor_list.append(_list_data[x])
  526. if not org:
  527. for x in range(len(_list_data)):
  528. if len(_list_data[x]) < 4:
  529. _list_data[x] = ''
  530. elif person_professor_list and re.findall('|'.join(person_professor_list), _list_data[x]):
  531. _list_data[x] = ''
  532. elif '经理' == _list_data[x][-2:]:
  533. _list_data[x] = ''
  534. for x in range(len(_list_data)):
  535. if _list_data[x]:
  536. org = _list_data[x]
  537. break
  538. if not person_professor_list:
  539. for x in range(len(_list_data)):
  540. if org in _list_data[x]:
  541. for j in range(x+1, len(_list_data)):
  542. if _list_data[j]:
  543. person_professor_list = [_list_data[j]]
  544. break
  545. break
  546. #print(org, person_professor_list, job_time)
  547. job_dict['job_company'] = org
  548. job_dict['job_leval'] = ' '.join(person_professor_list)
  549. job_dict['job_content'] = re.sub('工工作作内内容容::|工工作作内内容容::|工工作作内内容容', '工作内容:', ''.join(data_list[end_index:]))
  550. job_dict['job_content'] = re.sub('/', '-', job_dict['job_content'])
  551. job_list.append(job_dict)
  552. continue
  553. if len(data_list) > 1 and data_list[1] and data_list[1][-1] == '|':# and data_list[0] and data_list[0][-1] != '|':
  554. data_list[0] = data_list[0] + data_list[1]
  555. data_list[1] = ''
  556. elif len(data_list) > 2 and data_list[2] and data_list[2][-1] == '|' and data_list[0][-1] != '|' and '|' in str(data_list[0]) and data_list[1] and data_list[1][-1] != '|':
  557. data_list[0] = data_list[0] + data_list[1] + data_list[2]
  558. data_list[1] = ''
  559. data_list[2] = ''
  560. elif len(data_list) > 1 and data_list[1] and '工作职责:' in data_list[2]:
  561. data_list[0] = data_list[0] + data_list[1]
  562. data_list[1] = ''
  563. elif len(data_list) > 1 and '工作职责:' in data_list[3]:
  564. data_list[0] = data_list[0] + data_list[1] + data_list[2]
  565. data_list[1] = ''
  566. data_list[2] = ''
  567. job_time = re.findall(re_txt, data_list[0])
  568. job_dict['job_time'] = job_time[0]
  569. _nums = re.findall('\d+', job_dict['job_time'])
  570. #print(_nums)
  571. if len(_nums) >= 4:
  572. job_dict['job_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  573. elif len(_nums) == 2:
  574. job_dict['job_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  575. data_list[0] = re.sub(job_time[0], '', data_list[0])
  576. data_list[0] = data_list[0].strip()
  577. data_list[0] = re.sub('历任:', ' ', data_list[0])
  578. _list = data_list[0].split('|')
  579. if len(_list) == 1:
  580. __list = re.split('\040{2,}', data_list[0])
  581. #print(__list)
  582. job_dict['job_leval'] = __list[1].strip()
  583. job_dict['job_company'] = __list[0].strip()
  584. else:
  585. job_dict['job_leval'] = _list[0].strip()
  586. job_dict['job_company'] = _list[1].strip()
  587. if '职级:' in data_list[1:]:
  588. data_list.remove('职级:')
  589. job_dict['job_content'] = re.sub('工工作作内内容容::|工工作作内内容容::|工工作作内内容容', '工作内容:', ''.join(data_list[1:]))
  590. job_dict['job_content'] = re.sub('/', '-', job_dict['job_content'])
  591. #print(job_dict)
  592. job_list.append(job_dict)
  593. return job_list
  594. # 项目经历 (已弃用)
  595. # 项目名称未知
  596. def get_pro_list_old(lines):
  597. logger.info(lines)
  598. pro_list = [{"Time":None,"startTime":None,"endTime":None,"pro_name":None,"job_leval":None,"job_company":None,"content":None,},]
  599. regex = re.compile(r'((\d{4})[年\W]+(\d{1,2})[\W]?[\w]?)[至到\W]+((\d{4})[年\W]+(\d{1,2})[\W]?[\w]?)?([今])?')
  600. re_con = re.compile(r'负责内容(.*?)')
  601. re_na = re.compile(r'\W(.*?项目)\W')
  602. count = 0
  603. for line in lines:
  604. regex_time = regex.search(line)
  605. regex_content = re_con.search(line)
  606. regex_name = re_na.search(line)
  607. if regex_time:
  608. if pro_list[count].get("Time"):
  609. pro_list.append({"Time":None,"startTime":None,"endTime":None,"pro_name":None,"job_leval":None,"job_company":None,"content":None,})
  610. count += 1
  611. pro_list[count]["startTime"] = '{:4d}-{:02d}'.format(int(regex_time.group(2)),int(regex_time.group(3)))
  612. if regex_time.group(5) != None:
  613. pro_list[count]["endTime"] = '{:4d}-{:02d}'.format(int(regex_time.group(5)),int(regex_time.group(6)))
  614. pro_list[count]["Time"] = '{:4d}-{:02d}~{:4d}-{:02d}'.format(int(regex_time.group(2)),int(regex_time.group(3)),int(regex_time.group(5)),int(regex_time.group(6)))
  615. else:
  616. pro_list[count]["endTime"] = regex_time.group(7)
  617. pro_list[count]['Time'] = '{:4d}-{:02d}~{}'.format(int(regex_time.group(2)),int(regex_time.group(3)),regex_time.group(7))
  618. elif regex_name and (not pro_list[count].get("job_name")):
  619. pro_list[count]["pro_name"] = regex_name.group()
  620. elif pro_list[count].get("content"):
  621. pro_list[count]["content"] += line
  622. else:
  623. try:
  624. for word, tag in ner_tag(line):
  625. if (not pro_list[count].get("job_leval")) and (tag == "人物类_概念"):
  626. pro_list[count]["job_leval"] = word
  627. if (not pro_list[count].get("job_company")) and (tag in "组织机构类_企事业单位"):
  628. pro_list[count]["job_company"] = word
  629. except Exception as e:
  630. logger.error(e)
  631. pro_list[count]["content"] = line
  632. return pro_list
  633. # 项目经历 (UIE 已完成)
  634. def get_pro_list(lines):
  635. logger.info(lines)
  636. starts = []
  637. # 时间查找
  638. for index, line in enumerate(lines):
  639. if re.search(r'\d{4}', line):
  640. starts.append(index)
  641. # 简单筛选
  642. count = len(starts)
  643. c = (starts[-1] - starts[0])/count
  644. for i in range(count-1):
  645. if (starts[i+1]-starts[i] < c/2):
  646. starts[i+1] = starts[i]
  647. # 合并
  648. pro_list = []
  649. pros = {}
  650. index = 0
  651. for i in range(len(lines)):
  652. if i in starts:
  653. index = i
  654. pros[index] = [lines[i], []]
  655. pros[index][1].append(lines[i])
  656. elif not pros:
  657. continue
  658. else:
  659. pros[index][0] += lines[i]
  660. pros[index][1].append(lines[i])
  661. # 提取
  662. for key in pros.keys():
  663. info = pro_ie(pros[key][0])
  664. src = pros[key][1]
  665. for rst in info:
  666. if not rst.get("时间") or not rst.get("项目名称"):
  667. continue
  668. rst["工作内容"] = [{"text":""}]
  669. logger.info(rst)
  670. for l in src:
  671. if rst["时间"][0]["text"] in l:
  672. continue
  673. else:
  674. rst["工作内容"][0]["text"] += l
  675. for key in rst.keys():
  676. if key == "时间":
  677. time_list = [None, None, None, None, None, None]
  678. tim_list = re.findall(r'\d+', rst["时间"][0]["text"])
  679. i = 0
  680. for t in tim_list:
  681. if (len(t) == 4) and (i != 0):
  682. i = 3
  683. time_list[i] = t
  684. else:
  685. time_list[i] = t
  686. i += 1
  687. else:
  688. continue
  689. if time_list[3] is not None:
  690. if time_list[4] is not None:
  691. rst["时间"][0]["text"] = "{:4d}-{:02d}~{:4d}-{:02d}".format(int(time_list[0]),int(time_list[1]),int(time_list[3]),int(time_list[4]))
  692. else:
  693. rst["时间"][0]["text"] = "{:4d}~{:4d}".format(int(time_list[0]),int(time_list[3]))
  694. else:
  695. if time_list[1] is not None:
  696. rst["时间"][0]["text"] = "{:4d}-{:02d}~至今".format(int(time_list[0]),int(time_list[1]))
  697. else:
  698. rst["时间"][0]["text"] = "{:4d}~至今".format(int(time_list[0]))
  699. pro_list.extend([{key:rst[key][0]["text"] for key in rst.keys()} for rst in info])
  700. return pro_list
  701. # 培训经历 (已完成)
  702. # ner + 分词 (机构名) 培训项目 时间
  703. def get_cultivate_list(lines):
  704. logger.info(lines)
  705. job_list = []
  706. re_txt = '\d{4,4}.\d{1,2}.?\040{0,2}[\-–至-\—~]\040{0,2}\d{4,4}.\d{1,2}[月]?|\d+\.\d+\-至今|\d+年\d+月\-\d+年\d+月|\d+年\d+月\-\~|\d+年\d+月[\-\~]至今|\d+-\d+\040{0,2}[\~至]\040{0,2}\d+-\d+|\d+-\d+\~|\d+-\d+\~至今|\d+-\d+\040{0,2}至今|^\d{4,4}.\d{1,2}|\d{4,4}.'
  707. re_txt_1 = '\d{4,4}.\d{1,2}.?\040{0,2}[\-–至-\—~]\040{0,2}\d{4,4}.\d{1,2}[月]?|\d+\.\d+\-至今|\d+年\d+月\-\d+年\d+月|\d+年\d+月\-\~|\d+年\d+月[\-\~]至今|\d+-\d+\040{0,2}[\~至]\040{0,2}\d+-\d+|\d+-\d+\~|\d+-\d+\~至今|\d+-\d+\040{0,2}至今'
  708. nums = []
  709. for i in range(len(lines)):
  710. if re.findall(re_txt, lines[i].replace(' ', '')) and re.findall('\||\040{1,}', lines[i]):
  711. nums.append(i)
  712. continue
  713. if re.findall(re_txt, lines[i].replace(' ', '')[:20]):
  714. nums.append(i)
  715. if len(lines[i].strip().replace(' ', '')) > 50:
  716. continue
  717. nums.append(len(lines))
  718. year_dict = {18:4, 17:3,20:3,19:3,21:2,22:1}
  719. for i in range(1, len(nums[:])):
  720. job_dict = {'cultivate_time':'', 'cultivate_time_beg':'', 'cultivate_time_end':'', 'cultivate_name':'','cultivate_leval':'','cultivate_content':''}
  721. data_list = lines[nums[i-1]:nums[i]]
  722. data_line = ' '.join(data_list)
  723. data_line = re.sub('[\|\t]', ' ', data_line)
  724. data_line = re.sub('-{3,}', '', data_line)
  725. ner_data = ner(''.join(data_list[:2]))
  726. org = ''
  727. time_list = []
  728. for _ in ner_data:
  729. if _[1] == 'ORG' and not org:
  730. org = _[0].strip()
  731. elif _[1] == 'TIME' and len(_[1]) >= 4:
  732. time_list.append(_[0])
  733. #TIME
  734. logger.info(data_line)
  735. _list_data = re.split('\040+', data_line)
  736. top_level = 22
  737. end_index = 0
  738. remove_list = []
  739. if len(_list_data) <= 2:
  740. end_index = 0
  741. #continue
  742. job_time = re.findall(re_txt_1, data_list[0])
  743. if job_time:
  744. job_dict['cultivate_time'] = job_time[0]
  745. data_list[0] = re.sub(job_time[0], '', data_list[0])
  746. else:
  747. job_dict['cultivate_time'] = ''
  748. for t in time_list:
  749. data_list[0] = re.sub(t, '', data_list[0])
  750. _list = data_list[0].split('|')
  751. if len(_list) >= 2:
  752. job_dict['cultivate_name'] = _list[0].strip()
  753. job_dict['cultivate_leval'] = _list[1].strip()
  754. end_index = 1
  755. _nums = re.findall('\d+', job_dict['cultivate_time'])
  756. if len(_nums) >= 4:
  757. job_dict['cultivate_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  758. job_dict['cultivate_time_end'] = '%s-%02d'%(_nums[2], int(_nums[3]))
  759. job_dict['cultivate_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  760. elif len(_nums) == 2:
  761. job_dict['cultivate_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  762. job_dict['cultivate_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  763. job_dict['cultivate_time_end'] = '%s'%('至今')
  764. elif len(time_list) == 2:
  765. nums_1 = re.findall('\d+', time_list[0])
  766. nums_2 = re.findall('\d+', time_list[1])
  767. nums_1.append('09')
  768. nums_2.append('07')
  769. job_dict['cultivate_time_beg'] = '%s-%02d'%(nums_1[0], int(nums_1[1]))
  770. job_dict['cultivate_time_end'] = '%s-%02d'%(nums_2[0], int(nums_2[1]))
  771. job_dict['cultivate_time'] = '%s-%02d~%s-%02d'%(nums_1[0], int(nums_1[1]), nums_2[0], int(nums_2[1]))
  772. elif len(time_list) == 1:
  773. _nums = re.findall('\d+', time_list[0])
  774. if '获得' in data_list[0]:
  775. _nums.append('01')
  776. _nums.insert(0, '01')
  777. _nums.insert(0, str(int(_nums[1]) - year_dict[top_level]))
  778. job_dict['cultivate_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  779. job_dict['cultivate_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  780. job_dict['cultivate_time_end'] = '%s-%02d'%(_nums[2], int(_nums[3]))
  781. else:
  782. _nums.append('01')
  783. job_dict['cultivate_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  784. job_dict['cultivate_time_beg'] = '%s-%02d'%(_nums[0], int(_nums[1]))
  785. job_dict['cultivate_time_end'] = '%s'%('至今')
  786. job_dict['cultivate_content'] = re.sub('培培训训内内容容::|培培训训内内容容::|培培训训内内容容', '培训内容:', ''.join(data_list[end_index:]))
  787. if not job_dict['cultivate_name']:
  788. job_dict['cultivate_name'] = org
  789. logger.info(job_dict)
  790. job_list.append(job_dict)
  791. continue
  792. '''
  793. #print(nums)
  794. for i in range(1, len(nums[:])):
  795. job_dict = {'cultivate_time':'', 'cultivate_name':'','cultivate_leval':'','cultivate_content':''}
  796. data_list = lines[nums[i-1]:nums[i]]
  797. if '' in data_list:
  798. data_list.remove('')
  799. if len(data_list) > 1 and data_list[1] and data_list[1][-1] == '|' and data_list[0][-1] != '|':
  800. data_list[0] = data_list[0] + data_list[1]
  801. data_list[1] = ''
  802. job_time = re.findall(re_txt_1, data_list[0])
  803. job_dict['cultivate_time'] = job_time[0]
  804. _nums = re.findall('\d+', job_dict['cultivate_time'])
  805. if len(_nums) >= 4:
  806. job_dict['cultivate_time'] = '%s-%02d~%s-%02d'%(_nums[0], int(_nums[1]), _nums[2], int(_nums[3]))
  807. elif len(_nums) == 2:
  808. job_dict['cultivate_time'] = '%s-%02d~%s'%(_nums[0], int(_nums[1]), '至今')
  809. data_list[0] = re.sub(job_time[0], '', data_list[0])
  810. _list = data_list[0].split('|')
  811. if len(_list) >= 2:
  812. job_dict['cultivate_name'] = _list[0].strip()
  813. job_dict['cultivate_leval'] = _list[1].strip()
  814. job_dict['cultivate_content'] = re.sub('培培训训内内容容|培培训训内内容容::|培培训训内内容容::', '培训内容:', ''.join(data_list[1:]))
  815. else:
  816. job_dict['cultivate_content'] = re.sub('培培训训内内容容|培培训训内内容容::|培培训训内内容容::', '培训内容:', ''.join(data_list[0:]))
  817. #print(job_dict)
  818. '''
  819. return job_list
  820. # 语言能力
  821. def get_lag_list(lines):
  822. logger.info(lines)
  823. job_list = []
  824. re_lan = re.compile(r'(\w+[语话])')
  825. lag_dict = {'lag_name':'', 'lag_leval':""}
  826. for l in lines:
  827. if not l.strip():
  828. continue
  829. lag_name = re.search(re_lan, l)
  830. if lag_name and lag_name.group(1):
  831. if lag_dict['lag_name']:
  832. job_list.append(lag_dict)
  833. lag_dict['lag_name'] = lag_name.group(1)
  834. return job_list
  835. # 家庭情况(已弃用)
  836. def get_fam_list(lines):
  837. job_list = []
  838. fam_dict = {}
  839. for l in lines:
  840. if not l.strip():
  841. continue
  842. ls = l.split('|')
  843. if len(ls) == 1:
  844. continue
  845. fam_dict = {'fam_name':"",'fam_company':"",'fam_lable':"","fam_status":"", 'fam_job':""}
  846. fam_dict["fam_lable"] = ls[0].strip()
  847. fam_dict["fam_name"] = ls[1].strip()
  848. flag = 0
  849. if re.findall('\d岁|\d{4,5}', ls[2]):
  850. flag = 1
  851. fam_dict["fam_company"] = ls[flag+2].strip()
  852. fam_dict["fam_job"] = ls[flag+3].strip()
  853. fam_dict["fam_status"] = ls[flag+4].strip()
  854. #print(fam_dict)
  855. job_list.append(fam_dict)
  856. return job_list
  857. # 证书情况 时间+证书名称 (旧版)
  858. def get_cet_list_old(lines):
  859. logger.info(lines)
  860. job_list = []
  861. re_txt = '\d+年\d+月|\d+-\d+|\d+\.\d+'
  862. lines_word = ' '.join(lines)
  863. lines = re.findall('\d+年\d+月|\d+-\d+|\d+\.\d+', lines_word)
  864. nums = []
  865. for x in range(len(lines) - 1):
  866. _index = lines_word.index(lines[x])
  867. _end_index = lines_word.index(lines[x+1])
  868. l = lines_word[_index : _end_index]
  869. if not l.strip():
  870. continue
  871. lines_word = lines_word[_end_index:]
  872. job_time = re.findall(re_txt, l)
  873. cet_dict = {'cet_name':'','cet_time':""}
  874. if job_time:
  875. cet_dict['prize_time'] = job_time[0]
  876. l = re.sub(job_time[0], '', l)
  877. else:
  878. continue
  879. ls = re.split('\||\040+|\t+', l)
  880. logger.info(ls)
  881. for l in ls:
  882. if len(l) <= 3:
  883. continue
  884. cet_dict['prize_name'] = l.strip()
  885. break
  886. job_list.append(cet_dict)
  887. return job_list
  888. # 证书情况 时间+证书名称 (UIE已完成)
  889. def get_cet_list(lines):
  890. logger.info(lines)
  891. cet_list = []
  892. for line in lines:
  893. info = cet_ie(line)
  894. cet_list.extend([{key:rst[key][0]["text"] for key in rst.keys()} for rst in info if rst.get("证书")])
  895. return cet_list
  896. # 获奖情况 时间+获奖名称 (旧版)
  897. def get_prize_list_old(lines):
  898. logger.info(lines)
  899. job_list = []
  900. re_txt = '\d+年\d+月|\d+-\d+|\d{4,4}.\d{1,2}'
  901. lines_word = ' '.join(lines)
  902. lines = re.findall('\d+年\d+月|\d{4,4}-\d+|\d{4,4}.\d{1,2}', lines_word)
  903. nums = []
  904. for x in range(len(lines) - 1):
  905. _index = lines_word.index(lines[x])
  906. _end_index = lines_word.index(lines[x+1])
  907. l = lines_word[_index : _end_index]
  908. if not l.strip():
  909. continue
  910. lines_word = lines_word[_end_index:]
  911. job_time = re.findall(re_txt, l)
  912. cet_dict = {'prize_name':'','prize_time':""}
  913. if job_time:
  914. cet_dict['prize_time'] = job_time[0]
  915. l = re.sub(job_time[0], '', l)
  916. else:
  917. continue
  918. ls = re.split('\||\040+|\t+', l)
  919. logger.info(ls)
  920. for l in ls:
  921. if len(l) <= 3:
  922. continue
  923. cet_dict['prize_name'] = l.strip()
  924. break
  925. logger.info(cet_dict)
  926. job_list.append(cet_dict)
  927. return job_list
  928. # 获奖情况 时间+获奖名称 (UIE已完成)
  929. def get_prize_list(lines):
  930. logger.info(lines)
  931. prize_list = []
  932. for line in lines:
  933. info = prize_ie(line)
  934. prize_list.extend([{key:rst[key][0]["text"] for key in rst.keys()} for rst in info if rst.get("奖项")])
  935. return prize_list
  936. # 返回其他信息
  937. def get_other_list(lines):
  938. other_list = []
  939. other_list.append("\n".join(lines))
  940. return other_list
  941. # Linux doc 文件处理
  942. def doc2pdf_linux(docPath, pdfPath):
  943. """
  944. 允许的文档格式:doc,docx
  945. 仅在linux平台下可以
  946. 需要在linux中下载好libreoffice
  947. """
  948. # 注意cmd中的libreoffice要和linux中安装的一致
  949. cmd = 'libreoffice6.3 --headless --convert-to pdf'.split() + [docPath] + ['--outdir'] + [pdfPath]
  950. # cmd = 'libreoffice6.2 --headless --convert-to pdf'.split() + [docPath]
  951. p = subprocess.Popen(cmd, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
  952. p.wait(timeout=30) # 停顿30秒等待转化
  953. stdout, stderr = p.communicate()
  954. if stderr:
  955. raise subprocess.SubprocessError(stderr)
  956. # Windows doc 文件处理
  957. def doc2pdf_win(docPath, pdfPath):
  958. console.print(pdfPath+'/'+os.path.splitext(os.path.split(docPath)[-1])[0] + '.pdf')
  959. import win32com
  960. from win32com.client import DispatchEx, constants
  961. word = DispatchEx("Word.Application") #内部方法
  962. word.Visible = 1 # 后台运行,不显示
  963. word.DisplayAlerts = 0 # 不警告
  964. doc = word.Documents.Open(docPath) #转换源文件
  965. doc.SaveAs(pdfPath+'/'+os.path.splitext(os.path.split(docPath)[-1])[0] + '.pdf', FileFormat=17) #txt=4,html=10,docx=16,pdf=17 #新文件
  966. doc.Close() #关闭
  967. word.Quit() #退出
  968. # doc 文件处理
  969. def doc2pdf(docPath, pdfPath, system):
  970. """
  971. 注意使用绝对路径
  972. pdf的生成只写路径,不写名字
  973. """
  974. docPathTrue = os.path.abspath(docPath) # bugfix - searching files in windows/system32
  975. if system == "Linux":
  976. return doc2pdf_linux(docPathTrue, pdfPath)
  977. if system == "Windows":
  978. return doc2pdf_win(docPathTrue, pdfPath)
  979. # txt 纯文本解析(已完成)
  980. def parse_txt(path, save_dir):
  981. with open(path, 'r', encoding='utf-8') as fp:
  982. data = fp.read()
  983. global block, block_rev
  984. chun = 1
  985. page = {1: []}
  986. if len(data.split("\n")) <= 2:
  987. for line in data.split("\n"):
  988. line = line.replace("\xa0", "").replace("【","").replace("】","").replace("教育/培训","教育经历").strip()
  989. for word in line.split():
  990. if word in block.keys():
  991. chun = block[word]
  992. page[chun] = []
  993. elif word:
  994. page[chun].append(word)
  995. else:
  996. for line in data.split("\n"):
  997. line = line.replace("\xa0", "").replace("【","").replace("】","").replace("教育/培训","教育经历")
  998. regex = re.compile(u'[\u3000]+',re.UNICODE)
  999. line = regex.sub('', line.strip())
  1000. if line in block.keys():
  1001. chun = block[line]
  1002. page[chun] = []
  1003. elif line:
  1004. page[chun].append(line)
  1005. result_data = []
  1006. for key in page.keys():
  1007. for index, func in zip([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [get_base_info, get_job_intention, get_edu_list, get_job_list, get_pro_list, get_other_list, get_other_list, get_other_list, get_lag_list, get_cet_list, get_prize_list, get_cultivate_list]):
  1008. if key == index:
  1009. result_data.append({block_rev[index]:func(page[index])})
  1010. filename = os.path.splitext(os.path.split(path)[-1])[0]+'.json'
  1011. with open(os.path.join(save_dir, filename), 'w', encoding="utf-8") as fp:
  1012. json.dump({"result":result_data}, fp, indent=4, ensure_ascii=False)
  1013. # 纯文本 word 解析
  1014. def read_from_word(doc, path, save_dir):
  1015. para_text = []
  1016. for para in doc.paragraphs:
  1017. para_text.append(para.text)
  1018. global block, block_rev
  1019. chun = 1
  1020. page = {1: []}
  1021. for line in para_text:
  1022. regex = re.compile(u'[\uF000-\uF0FF]+',re.UNICODE)
  1023. line = regex.sub('', line)
  1024. if line in block.keys():
  1025. chun = block[line]
  1026. page[chun] = []
  1027. elif line:
  1028. page[chun].append(line)
  1029. result_data = []
  1030. for key in page.keys():
  1031. for index, func in zip([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [get_base_info, get_job_intention, get_edu_list, get_job_list, get_pro_list, get_other_list, get_other_list, get_other_list, get_lag_list, get_cet_list, get_prize_list, get_cultivate_list]):
  1032. if key == index:
  1033. result_data.append({block_rev[index]:func(page[index])})
  1034. filename = os.path.splitext(os.path.split(path)[-1])[0]+'.json'
  1035. with open(os.path.join(save_dir, filename), 'w', encoding="utf-8") as fp:
  1036. json.dump({"result":result_data}, fp, indent=4, ensure_ascii=False)
  1037. # 提取 word 表格(已完成)
  1038. def check_word(path, save_dir):
  1039. doc = Document(path)
  1040. tables = doc.tables
  1041. if not tables:
  1042. logger.info("this is raw text")
  1043. read_from_word(doc, path, save_dir=save_dir)
  1044. logger.info("this is a Table")
  1045. global block
  1046. with open("resources/keys.json", "r", encoding="utf-8") as fp:
  1047. prk = json.load(fp)
  1048. chun = 1
  1049. page = {1: []}
  1050. regex = re.compile(r'(\(\w{2,8}\))?((\w{2,8}))?')
  1051. for table in tables:
  1052. lo = {} # 存储每一行去重后的数据
  1053. for row in range(0, len(table.rows)):
  1054. row_list = []
  1055. for col in range(0, len(table.row_cells(row))): # 提取row行的全部列数据
  1056. if len(''.join(table.cell(row, col).text)) <= 20:
  1057. row_list.append(re.sub(r'(\w)\n', r'\1', table.cell(row, col).text))
  1058. else:
  1059. row_list.append(regex.sub("", table.cell(row, col).text.replace(" ","").replace(":", ":").replace("学历\n学位","学历学位"))) # 去除字符串中的特殊字符,并添加到临时列表中
  1060. lo[row] = (sorted(set(row_list), key=row_list.index)) # 在不变顺序的前提下,去除List中的重复项
  1061. # 去除空项
  1062. for key in list(lo.keys()):
  1063. if "" in lo[key]:
  1064. lo[key].remove("")
  1065. if not lo[key]:
  1066. lo.pop(key)
  1067. for _, line in lo.items():
  1068. if (line[0] in block.keys()) or (line[0] in prk.keys()):
  1069. # 包含大类目名
  1070. if line[0] in block.keys():
  1071. # 指向当前类目
  1072. chun = block[line[0]]
  1073. if not page.get(chun):
  1074. page[chun] = []
  1075. # 去除类目名
  1076. line = '\n'.join(line[1:])
  1077. # 包含小类目
  1078. elif line[0] in prk.keys():
  1079. # 指向当前类目
  1080. chun = prk[line[0]]
  1081. if not page.get(chun):
  1082. page[chun] = []
  1083. # 不去除
  1084. line = '\n'.join(line)
  1085. else:
  1086. line = '\n'.join(line)
  1087. # 标准化小类目
  1088. for k in prk.keys():
  1089. line = line.replace(k+"\n", k+":")
  1090. page[chun].extend(line.split())
  1091. result_data = []
  1092. for key in page.keys():
  1093. for index, func in zip([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [get_base_info, get_job_intention, get_edu_list, get_job_list, get_pro_list, get_other_list, get_other_list, get_other_list, get_lag_list, get_cet_list, get_prize_list, get_cultivate_list]):
  1094. if key == index:
  1095. result_data.append({block_rev[index]:func(page[index])})
  1096. filename = os.path.splitext(os.path.split(path)[-1])[0]+'.json'
  1097. with open(os.path.join(save_dir, filename), 'w', encoding="utf-8") as fp:
  1098. json.dump({"result":result_data}, fp, indent=4, ensure_ascii=False)
  1099. # pdf 解析句子(已完成)
  1100. def parse_line_layout(layout, b):
  1101. texts = []
  1102. """解析页面内容,一行一行的解析"""
  1103. # bbox:
  1104. # x0:从页面左侧到框左边缘的距离。
  1105. # y0:从页面底部到框的下边缘的距离。
  1106. # x1:从页面左侧到方框右边缘的距离。
  1107. # y1:从页面底部到框的上边缘的距离
  1108. for textbox in layout:
  1109. if isinstance(textbox, LTTextBox) or isinstance(textbox, LTTextLine):
  1110. for char in textbox:
  1111. if isinstance(char, LTTextLineHorizontal):
  1112. texts.append([char.bbox[0], char.bbox[3], char.get_text().strip()])
  1113. # 按行排序
  1114. texts.sort(key=lambda x:-x[1])
  1115. global block, block_rev
  1116. chun = b
  1117. page = {chun: []}
  1118. for _, _, line in texts:
  1119. regex = re.compile(u'[\u007F|\u25A0|\u00B7|\uF000-\uF0FF]+',re.UNICODE)
  1120. line = regex.sub('', line)
  1121. regex_tips = re.compile(r'(\(.*?\))?((.*?))?')
  1122. # line = regex_tips.sub('', line)
  1123. line = line.strip()
  1124. if regex_tips.sub('', line).strip() in block.keys():
  1125. chun = block[regex_tips.sub('', line).strip()]
  1126. page[chun] = []
  1127. elif line:
  1128. page[chun].append(line)
  1129. return page, chun
  1130. # pdf 样式解析(已完成)
  1131. def read_from_pdf(path, save_dir):
  1132. result = {}
  1133. global block_rev
  1134. with open(path, 'rb') as in_file:
  1135. parser = PDFParser(in_file) # 用文件对象来创建一个pdf文档分析器
  1136. doc: PDFDocument = PDFDocument(parser) # 创建pdf文档
  1137. rsrcmgr = PDFResourceManager() # 创建PDF,资源管理器,来共享资源
  1138. # 创建一个PDF设备对象
  1139. laparams = LAParams()
  1140. device = PDFPageAggregator(rsrcmgr, laparams=laparams)
  1141. # 创建一个PDF解释其对象
  1142. interpreter = PDFPageInterpreter(rsrcmgr, device)
  1143. # 循环遍历列表,每次处理一个page内容
  1144. # doc.get_pages() 获取page列表
  1145. interpreter = PDFPageInterpreter(rsrcmgr, device)
  1146. # 处理文档对象中每一页的内容
  1147. # 循环遍历列表,每次处理一个page的内容
  1148. b = 1
  1149. for page in PDFPage.create_pages(doc):
  1150. logger.debug('================ 新页面 ================')
  1151. interpreter.process_page(page)
  1152. layout = device.get_result()
  1153. r, b = parse_line_layout(layout, b)
  1154. for key in r.keys():
  1155. if result.get(key):
  1156. result[key].extend(r[key])
  1157. else:
  1158. result[key] = r[key]
  1159. result_data = []
  1160. for key in result.keys():
  1161. for index, func in zip([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [get_base_info, get_job_intention, get_edu_list, get_job_list, get_pro_list, get_other_list, get_other_list, get_other_list, get_lag_list, get_cet_list, get_prize_list, get_cultivate_list]):
  1162. if key == index:
  1163. result_data.append({block_rev[index]:func(result[index])})
  1164. filename = os.path.splitext(os.path.split(path)[-1])[0]+'.json'
  1165. with open(os.path.join(save_dir, filename), 'w', encoding="utf-8") as fp:
  1166. json.dump({"result":result_data}, fp, indent=4, ensure_ascii=False)
  1167. # pdf 表格解析 (已完成)
  1168. def parse_table_from_pdf(path, save_dir):
  1169. global block, block_rev
  1170. lo = {}
  1171. with pdfplumber.open(path) as pdf:
  1172. for page in pdf.pages:
  1173. for table in page.extract_tables():
  1174. for line in table:
  1175. row_list = []
  1176. for word in line:
  1177. row_list.append(word)
  1178. lo[len(lo.keys())] = row_list
  1179. # 去除空项
  1180. for key in list(lo.keys()):
  1181. if "" in lo[key]:
  1182. lo[key].remove("")
  1183. if not lo[key]:
  1184. lo.pop(key)
  1185. for _, line in lo.items():
  1186. if (line[0] in block.keys()) or (line[0] in prk.keys()):
  1187. # 包含大类目名
  1188. if line[0] in block.keys():
  1189. # 指向当前类目
  1190. chun = block[line[0]]
  1191. if not page.get(chun):
  1192. page[chun] = []
  1193. # 去除类目名
  1194. line = '\n'.join(line[1:])
  1195. # 包含小类目
  1196. elif line[0] in prk.keys():
  1197. # 指向当前类目
  1198. chun = prk[line[0]]
  1199. if not page.get(chun):
  1200. page[chun] = []
  1201. # 不去除
  1202. line = '\n'.join(line)
  1203. else:
  1204. line = '\n'.join(line)
  1205. # 标准化小类目
  1206. for k in prk.keys():
  1207. line = line.replace(k+"\n", k+":")
  1208. page[chun].extend(line.split())
  1209. result_data = []
  1210. for key in page.keys():
  1211. for index, func in zip([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [get_base_info, get_job_intention, get_edu_list, get_job_list, get_pro_list, get_other_list, get_other_list, get_other_list, get_lag_list, get_cet_list, get_prize_list, get_cultivate_list]):
  1212. if key == index:
  1213. result_data.append({block_rev[index]:func(page[index])})
  1214. filename = os.path.splitext(os.path.split(path)[-1])[0]+'.json'
  1215. with open(os.path.join(save_dir, filename), 'w', encoding="utf-8") as fp:
  1216. json.dump({"result":result_data}, fp, indent=4, ensure_ascii=False)
  1217. # 检测 pdf 格式 (已完成)
  1218. def check_pdf(path):
  1219. """
  1220. # 输入:
  1221. # pdf 文件路径
  1222. # 输出:
  1223. # 文件包含元素 [Word, Table]
  1224. """
  1225. rst = []
  1226. for page_layout in extract_pages(path):
  1227. for element in page_layout:
  1228. if isinstance(element, LTFigure):
  1229. for cell in element:
  1230. if isinstance(cell, LTChar):
  1231. rst.append("Table")
  1232. break
  1233. elif isinstance(element, LTTextContainer):
  1234. rst.append("Word")
  1235. return set(rst)
  1236. # 检测编码(已完成)
  1237. def decode_path(path):
  1238. '''zipfile解压出现乱码,将乱码的路径编码为UTF8'''
  1239. try:
  1240. path_name = path.decode('utf-8')
  1241. except:
  1242. path_name = path.encode('437').decode('gbk')
  1243. path_name = path_name.encode('utf-8').decode('utf-8')
  1244. return path_name
  1245. # 检测传入格式(已完成)
  1246. def detection_type(path, system):
  1247. tempdir = time.strftime("%Y_%m_%dT%H_%M_%S")
  1248. os.mkdir(tempdir)
  1249. # 传入 rar 压缩文件
  1250. if os.path.isfile(path) and path.endswith('.rar'):
  1251. rar = rarfile.RarFile(path)
  1252. rar.extractall('./cache/' + tempdir)
  1253. path = "./cache/" + tempdir
  1254. # 传入 tar.gz 压缩文件
  1255. if os.path.isfile(path) and path.endswith('.tar.gz'):
  1256. tf = tarfile.open(path)
  1257. tf.extractall('./cache/' + tempdir)
  1258. tf.close()
  1259. path = "./cache/" + tempdir
  1260. # 传入 .zip .7z 压缩文件
  1261. try:
  1262. if os.path.isfile(path) and path.endswith('.zip'):
  1263. ## 解压方式1:存在乱码
  1264. # f = zipfile.ZipFile(file, mode='r')
  1265. # f.extractall(target_dir)
  1266. ## 解压方式2:防止乱码
  1267. with ZipFile(path, allowZip64=True) as zf:
  1268. # 排除目录文件
  1269. print("zf.filelist", zf.filelist)
  1270. file_iter = (filename for filename in zf.filelist if os.path.isfile(path))
  1271. for filename in file_iter:
  1272. # 编码文件名称为 utf 格式
  1273. filename.filename = decode_path(filename.filename) # 防止乱码的操作
  1274. zf.extract(filename, "./cache/" + tempdir)
  1275. path = "./cache/" + tempdir
  1276. elif os.path.isfile(path) and path.endswith('.7z'): # .7z格式文件解压
  1277. zf = py7zr.SevenZipFile(path, mode='r')
  1278. zf.extractall("./cache/" + tempdir)
  1279. path = "./cache/" + tempdir
  1280. except Exception as e:
  1281. logger.error(e)
  1282. # 传入为 doc
  1283. if os.path.isfile(path) and path.endswith('.doc'):
  1284. doc2pdf(docPath = path, pdfPath = './pdf', system=system)
  1285. newfile = './pdf/' + os.path.splitext(os.path.split(path)[-1])[0] + '.pdf'
  1286. if os.path.exists(newfile):
  1287. rst = check_pdf(newfile)
  1288. if "Table" in rst:
  1289. parse_table_from_pdf(newfile, save_dir=tempdir)
  1290. pass
  1291. if "Word" in rst:
  1292. read_from_pdf(newfile, save_dir=tempdir)
  1293. # 传入为 docx
  1294. elif os.path.isfile(path) and path.endswith('.docx'):
  1295. check_word(path, save_dir=tempdir)
  1296. # 传入为 pdf
  1297. elif os.path.isfile(path) and path.endswith('.pdf'):
  1298. rst = check_pdf(path)
  1299. if "Table" in rst:
  1300. parse_table_from_pdf(path, save_dir=tempdir)
  1301. if "Word" in rst:
  1302. read_from_pdf(path, save_dir=tempdir)
  1303. # 传入为 txt
  1304. elif os.path.isfile(path) and path.endswith('.txt'):
  1305. parse_txt(path, save_dir=tempdir)
  1306. # 传入目录
  1307. elif os.path.isdir(path):
  1308. for filename in os.listdir(path):
  1309. filename = os.path.join(path, filename)
  1310. # 传入为 doc
  1311. logger.info(filename)
  1312. if filename.endswith('.doc') and not filename.startswith('.~'):
  1313. doc2pdf(docPath = filename, pdfPath = './pdf', system=system)
  1314. newfile = './pdf/' + os.path.splitext(os.path.split(filename)[-1])[0] + '.pdf'
  1315. if os.path.exists(newfile):
  1316. rst = check_pdf(newfile)
  1317. if "Table" in rst:
  1318. parse_table_from_pdf(newfile, save_dir=tempdir)
  1319. pass
  1320. if "Word" in rst:
  1321. read_from_pdf(newfile, save_dir=tempdir)
  1322. # 传入为 docx
  1323. elif os.path.isfile(filename) and filename.endswith('.docx'):
  1324. check_word(filename, save_dir=tempdir)
  1325. # 传入为 pdf
  1326. if os.path.isfile(filename) and filename.endswith('.pdf'):
  1327. rst = check_pdf(filename)
  1328. if "Table" in rst:
  1329. parse_table_from_pdf(filename, save_dir=tempdir)
  1330. pass
  1331. if "Word" in rst:
  1332. read_from_pdf(filename, save_dir=tempdir)
  1333. # 传入为 txt
  1334. elif os.path.isfile(filename) and filename.endswith('.txt'):
  1335. parse_txt(filename, save_dir=tempdir)
  1336. # 结果返回
  1337. for file in os.listdir(tempdir):
  1338. filename = os.path.join(tempdir, file)
  1339. with open(filename, "r", encoding="utf-8") as ff:
  1340. rst = json.load(ff)
  1341. url = "http://192.168.1.110:9999/talent/getResumeData"
  1342. session = requests.Session()
  1343. session.mount('http://', HTTPAdapter(max_retries = 3))
  1344. try:
  1345. headers = {
  1346. 'contentType':'Application/json'
  1347. }
  1348. response = session.post(url=url, headers=headers, json={"ResumeData":rst}, timeout=10)
  1349. print(response.text)
  1350. except Exception as e:
  1351. print(e)
  1352. console.print(rst, style="red", justify="left")
  1353. @app.post("/resume_parse")
  1354. async def file_upload(background_tasks: BackgroundTasks, file: UploadFile = File(...)):
  1355. """
  1356. 简历上传
  1357. 格式:pdf,docx,doc,txt,tar,zip,7z
  1358. """
  1359. res = await file.read()
  1360. with open('./uploads/' + file.filename, "wb") as f:
  1361. f.write(res)
  1362. background_tasks.add_task(detection_type, './uploads/' + file.filename, system)
  1363. return {"errno": 0, "msg": "Upload Success"}
  1364. if __name__ == '__main__':
  1365. import platform
  1366. system = platform.system()
  1367. if (system == "Windows"):
  1368. logger.info("Windows")
  1369. elif (system == "Linux"):
  1370. logger.info("Linux")
  1371. else:
  1372. logger.error("Unnot support this system")
  1373. if not os.path.exists("./uploads"):
  1374. os.mkdir("./uploads")
  1375. if not os.path.exists("./pdf"):
  1376. os.mkdir("./pdf")
  1377. if not os.path.exists("./cache"):
  1378. os.mkdir("./cache")
  1379. uvicorn.run(app=app, host="0.0.0.0", port=8320)