Python 数据库编程

最新推荐文章于 2024-06-14 09:50:01 发布

weixin_42234837

最新推荐文章于 2024-06-14 09:50:01 发布

阅读量51

点赞数

分类专栏： python 文章标签： python 数据库开发语言

本文链接：https://blog.csdn.net/weixin_42234837/article/details/129138816

版权

python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

Python 添加与查询数据

import pymysql;

def add():
	##输入用户信息
	print("请输入用户信息：")

	name = input("姓名：")
	sex = input("性别：")
	age = input("年龄：")
	phone = input("电话号码：")

	#连接数据库
	conn = pymysql.connect(host = "localhost",port = 3306,user = "root",password = "root",database = "USER_DB")

	##获取游标
	cursor = conn.cursor()

	try:
		sql = "insert into user(name,sex,age,phone) values(%s,%s,%s,%s)"
		cursor.execute(sql,(name,sex,age,phone))
		conn.commit()
		print("添加用户信息成功！")
	finally:
		conn.close()

##显示用户信息
def display():
	print("用户信息如下：")
	print("编号\t姓名\t性别\t年龄\t电话号码\n")

	##将用户信息保存到数据库中
	conn = pymysql.connect(host = "localhost",port = 3306,user = "root",password = "root",database = "USER_DB")
	cursor = conn.cursor()

	try:
		cursor.execute("SELECT * FROM user")
		for r in cursor:
			print("%d\t%s\t%s\t%s\t%s\n" %(r[0],str(r[1]),str(r[2]),str(r[3]),str(r[4])))
	finally:
		cursor.close()
		conn.close()

def main():
	nSelection = -1 ##菜单项编号
	while(nSelection != 0):
		##输出系统菜单
		print("----------------")
		print("1.添加用户信息")
		print("2.查询用户信息")
		print("0.退出")

		##提示并接受输入的菜单
		nSelection = int(input("请输入操作编号："))

		##根据选择的菜单项编号，执行相应的功能
		if(nSelection == 1):
			add()
		elif(nSelection == 2):
			display()
		elif(nSelection == 0):
			break
			print("退出")
		else:
			print("输入的编号错误！")


if __name__ == '__main__':
	main()