Spaces:
Sleeping
Sleeping
| 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'): | |
| #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) | |
| query = "select upper(skilldetails) skill from skillmaster where weightage = 0" | |
| df = pd.read_sql_query(query, conn) | |
| return df | |