RecommendationAPI / AddSkillDetails.py
Vaibhav84's picture
Change
b6ee5b6
raw
history blame
1.34 kB
from psycopg2 import sql
from DbConnection import DbConnection
class AddSkill:
def AddSkillDetails(skills):
returnMsg=''
details = skills.split(',')
skill_details = details[0]
skill_type = details [1]
skill_score1 = details[2]
weightage = -2
is_active = True
conn = DbConnection.GetMySQLDbConnection()
cursor = conn.cursor()
print("Adding Skill " + skill_details)
query = "SELECT skillid FROM skillmaster WHERE upper(skillDetails) IN (%s)"
params = (skill_details.upper(),) # Replace 'Test' with your actual variable or user input
cursor.execute(query, params)
if cursor.rowcount == 0:
insert_query = sql.SQL("""INSERT INTO SkillMaster (SkillDetails, SkillType, Weightage, IsActive, skill_score)
VALUES (%s, %s, %s, %s, %s) RETURNING SkillID""")
cursor.execute(insert_query, (skill_details, skill_type, weightage, is_active, skill_score1))
conn.commit()
returnMsg = 'Skill Added successfully'
else:
returnMsg = 'Skill Already in DB'
# Close the cursor and connection
cursor.close()
# Close the connection
conn.close()
print(returnMsg)
return returnMsg