|
|
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) |
|
|
|
|
|
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) |
|
|
result = getdbskills[getdbskills['skill'] == skill_name.upper()] |
|
|
print(result.empty) |
|
|
if(result.empty): |
|
|
print('----**') |
|
|
else: |
|
|
print('----'+ result) |
|
|
if(skill_type != 'Soft Skill'): |
|
|
|
|
|
if( skill_name in skills_list): |
|
|
continue |
|
|
skills_list.append(skill_name) |
|
|
|
|
|
return skills_list |
|
|
|
|
|
def GetSkillDatafromDB(db_params): |
|
|
conn = psycopg2.connect(**db_params) |
|
|
|
|
|
query = "select upper(skilldetails) skill from skillmaster where weightage = 0" |
|
|
df = pd.read_sql_query(query, conn) |
|
|
return df |
|
|
|
|
|
|
