import pymysql


def main():
    no = int(input('要删除的部门编号:'))
    # 1.创建连接对象
    conn = pymysql.connect(host='47.94.154.226', port=3306,
                           user='root', password='123456',
                           db='school', charset='utf8')
    try:
        # 2.获得游标对象
        with conn.cursor() as cursor:
            # 3.执行SQL得到结果
            result = cursor.execute('delete from tb_dept where dno=%s', (no,))
            if result == 1:
                print('删除成功')
            # 4.操作成功执行提交
            conn.commit()
    except pymysql.MySQLError as  error:
        print(error)
        conn.rollback()
    finally:
        conn.close()


if __name__ == '__main__':
    main()