File size: 1,354 Bytes
a13e09c 163e27f a13e09c 163e27f a13e09c 163e27f a13e09c 163e27f a13e09c 163e27f |
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 |
import re
from datetime import datetime
import psycopg2
import pandas as pd
class SkillExtractorDetailsV1:
def GetSkillData(skill_extractor, inputData, db_params):
getdbskills = SkillExtractorDetailsV1.GetSkillDatafromDB(db_params)
print(getdbskills)
skills_list = []
annotations = skill_extractor.annotate(inputData)
matches = annotations['results']['full_matches']+annotations['results']['ngram_scored']
skills_list = []
for result in matches:
skill_id = result['skill_id']
skill_name1 = skill_extractor.skills_db[skill_id]['skill_name']
skill_name = skill_name1.split("(")[0].strip()
skill_type = skill_extractor.skills_db[skill_id]['skill_type']
skill_score = round(result['score'],2)
if(skill_type != 'Soft Skill'):
#print(skill_name)
if( skill_name in skills_list):
continue
skills_list.append(skill_name)
return skills_list
def GetSkillDatafromDB(db_params):
conn = psycopg2.connect(**db_params)
cursor_obj = conn.cursor()
query = "select skilldetails from skillmaster where weightage = 0"
cursor_obj.execute(query)
skill_data = cursor_obj.fetchall()
return skill_data
|