|
|
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'): |
|
|
|
|
|
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 |
|
|
|
|
|
|
