封装一个方法实现教师数据的添加功能,接收从控制台输入的信息写入数据库,验证工号重复不允许添加并提示。
时间: 2023-12-03 18:46:07 浏览: 70
假设使用MySQL数据库,首先需要安装MySQL驱动程序。可以使用以下命令进行安装:
```
pip install mysql-connector-python
```
然后,可以使用以下代码实现教师数据的添加功能:
```python
import mysql.connector
def add_teacher():
# 连接数据库
conn = mysql.connector.connect(
host='localhost',
user='root',
password='password',
database='test'
)
# 获取数据库游标
cursor = conn.cursor()
# 从控制台输入教师信息
tid = input("请输入工号:")
name = input("请输入姓名:")
age = input("请输入年龄:")
gender = input("请输入性别:")
subject = input("请输入所教学科:")
# 查询工号是否已存在
cursor.execute("SELECT * FROM teacher WHERE tid=%s", (tid,))
result = cursor.fetchone()
if result:
print("工号已存在,无法添加!")
else:
# 将教师信息插入数据库
cursor.execute("INSERT INTO teacher (tid, name, age, gender, subject) VALUES (%s, %s, %s, %s, %s)", (tid, name, age, gender, subject))
conn.commit()
print("添加成功!")
# 关闭数据库连接
cursor.close()
conn.close()
```
上述代码首先从控制台输入教师信息,然后查询工号是否已存在,如果不存在,则将教师信息插入数据库中。如果工号已经存在,则无法添加,并提示用户。最后关闭数据库连接。
阅读全文