Smart-Talent-Resume-Ranker/database_functionalities.py at kapil-development · Code-with-ease/Smart-Talent-Resume-Ranker · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
from pymongo import MongoClient
import dns
from bson.objectid import ObjectId


def connect_db():
    try:
        client = MongoClient()
        client = MongoClient(
            'mongodb+srv://admin:admin@cluster0.ekv0t.mongodb.net/Resume-parser?retryWrites=true&w=majority')
        return client
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def insert_resume(client, details):
    try:
        db = client["Resume-parser"]
        resume_collection = db.resumes
        # ObjectId id = new ObjectId();
        details['_id'] = ObjectId()
        resume_id = resume_collection.insert_one(details).inserted_id
        return resume_id
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def insertIntoCategoryCollection(dict):
    try:
        client = connect_db()
        data_arr = []
        for key, value in dict.items():
            temp = {}
            temp["category"] = key
            temp["encoded_value"] = value
            data_arr.append(temp)
        db = client["Resume-parser"]
        category_collection = db.categories
        category_collection.insert_many(data_arr)
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def insert_jd(client, details):
    try:
        db = client["Resume-parser"]
        jd_collection = db.jobs
        # ObjectId id = new ObjectId();
        details['_id'] = ObjectId()
        jd_id = jd_collection.insert_one(details).inserted_id
        return jd_id
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def find_jd_by_details(client, details):
    try:
        db = client["Resume-parser"]
        jd_collection = db.jobs
        # ObjectId id = new ObjectId();
        # details['_id'] = ObjectId()
        jd = jd_collection.find_one(details)
        return jd
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def find_jds_by_details(client, details):
    try:
        db = client["Resume-parser"]
        jd_collection = db.jobs
        # ObjectId id = new ObjectId();
        # details['_id'] = ObjectId()
        jds = jd_collection.find(details)
        return jds
    except ValueError:
        print("Value Error :", sys.exc_info()[0])
    except:
        print("Unexpected error:", sys.exc_info()[0])
        raise


def find_resume_by_details(client, details):
    db = client["Resume-parser"]
    resume_collection = db.resumes
    details = resume_collection.find_one(details)
    return details


def find_resumes_by_skill(client, required_skill=[], one_of_skill=[]):
    db = client["Resume-parser"]
    resume_collection = db.resumes
    if (len(required_skill) != 0 and len(one_of_skill) != 0):
        details = resume_collection.aggregate(
            [{"$match": {"skills": {"$all": required_skill}}}, {"$match": {"skills": {"$in": one_of_skill}}}])
    elif (len(required_skill) != 0):
        details = resume_collection.aggregate([{"$match": {"skills": {"$all": required_skill}}}])
    elif (len(one_of_skill) != 0):
        details = resume_collection.aggregate([{"$match": {"skills": {"$in": one_of_skill}}}])
    else:
        details = resume_collection.find({})
    return details


# sample details structure to be passed
# details = {
#  'email': 'an431999@gmail.com',
#  'location': 'Nagar',
#  'name': 'Ayush Nagar',
#  'phone': '+919717504706',
#  'skills': ['github',
#   'iit',
#   'bootstrap',
#   'mongodb',
#   'git',
#   'twitter',
#   'ml',
#   'python',
#   'mysql',
#   'twitter api',
#   'data science',
#   'api',
#   'pcm',
#   'reactjs',
#   'foundation',
#   'rest',
#   'nlp',
#   'nodejs',
#   'pvr',
#   'google',
#   'c++',
#   'flask',
#   'it',
#   'angular',
#   'dps']}

# insert new resume that is parsed
# resume_id = insert_resume(details)
client = connect_db()
# find resume by email id
# email_searched_resumes = find_resume_by_details(client,{"email":"an431999@gmail.com"})
# print(email_searched_resumes)

# required + one of the following skills case
# y = find_resume_by_skill(required_skill = ['nodejs'],one_of_skill=['reactjs','angular','vuejs'])

# required skills case
# y = find_resume_by_skill(required_skill = ['nodejs'])

# one of the following skills case
# y = find_resume_by_skill(one_of_skill=['reactjs','angular','vuejs'])