ResCrunch/resume_parser.py at master · Chaitya62/ResCrunch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import TextConverter
from pdfminer.layout import LAParams
from pdfminer.pdfpage import PDFPage
from io import StringIO
import re

def get_links(text, url):
	match = [x[x.find(url):] for x in re.split('\n| |\)|\(|\{|\}|\[|\]|\'|\"', text) if url in x[x.find(url):]]
	print(match)
	return match

def get_github_username(github_links):
	for github_link in github_links:
		if github_link.endswith("/"):
			github_link = github_link[:-1]
		split_url = github_link.split('/')
		if len(split_url) == 2:
			return split_url[1]
	return ""

def get_linkedin_username(linkedin_links):
	for linkedin_link in linkedin_links:
		if linkedin_link.endswith("/"):
			linkedin_link = linkedin_link[:-1]
		split_url = linkedin_link.split('/')
		if len(split_url) == 3 and split_url[1] == "in":
			return split_url[2]
	return ""

def get_codechef_username(codechef_links):
	for codechef_link in codechef_links:
		if codechef_link.endswith("/"):
			codechef_link = codechef_link[:-1]
		split_url = codechef_link.split('/')
		if len(split_url) == 3 and split_url[1] == "users":
			return split_url[2]
	return ""

def convert_pdf_to_txt(path):
    rsrcmgr = PDFResourceManager()
    retstr = StringIO()
    codec = 'utf-8'
    laparams = LAParams()
    device = TextConverter(rsrcmgr, retstr, codec=codec, laparams=laparams)
    fp = open(path, 'rb')
    interpreter = PDFPageInterpreter(rsrcmgr, device)
    password = ""
    maxpages = 0
    caching = True
    pagenos=set()

    for page in PDFPage.get_pages(fp, pagenos, maxpages=maxpages, password=password,caching=caching, check_extractable=True):
        interpreter.process_page(page)

    text = retstr.getvalue()

    fp.close()
    device.close()
    retstr.close()
    return text

myContent = str(convert_pdf_to_txt('resume2.pdf').encode('ascii', 'ignore')).replace(r"\n", " ")
print(myContent)
print(get_github_username(get_links(myContent, "github.com")))
# print(get_links(myContent, "github.com"))
print(get_linkedin_username(get_links(myContent, "linkedin.com")))
print(get_codechef_username(get_links(myContent, "codechef.com")))