|
|
|
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(),) |
|
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' |
|
|
|
|
|
cursor.close() |
|
|
|
conn.close() |
|
print(returnMsg) |
|
return returnMsg |