change

SkillExtractV1.py

@@ -7,7 +7,7 @@ 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']
@@ -18,6 +18,8 @@ class SkillExtractorDetailsV1:
             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)
             if(skill_type != 'Soft Skill'):
                 #print(skill_name)
                 if( skill_name in skills_list):
@@ -28,11 +30,9 @@ class SkillExtractorDetailsV1:
 
     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
+        
+        query = "select upper(skilldetails) skill from skillmaster where weightage = 0"
+        df = pd.read_sql_query(query, conn)
+        return df