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