"""
修改数据
"""
import pymysql

no = int(input('部门编号: '))
name = input('部门名称: ')
location = input('部门所在地: ')

# 第一步：创建连接
conn = pymysql.connect(host='10.7.174.103', port=3306,
                       user='guest', password='Guest.618',
                       database='hrs', charset='utf8mb4')
try:
    # 第二步：获得游标对象
    with conn.cursor() as cursor:
        # 第三步：通过游标向数据库服务器发出SQL语句，获取执行结果
        affected_rows = cursor.execute(
            'update tb_dept set dname=%s, dloc=%s where dno=%s',
            (name, location, no)
        )
        if affected_rows == 1:
            print('更新部门信息成功！')
    # 第四步：提交（操作成功）
    conn.commit()
except pymysql.MySQLError as err:
    print(err)
    # 第四步：回滚（操作失败）
    conn.rollback()
finally:
    # 第五步：关闭连接（释放资源）
    conn.close()