说明
针对数据库的操作,一共分三步:
1.建立数据库连接
2.执行SQL语句
3.关闭数据库连接
Python操作数据库(如MySQL),Python发送能够与数据库直接通信的数据包,并获取数据库服务器的相应结果。是一种典型的基于TCP/IP的通信过程,要求必须要满足数据库服务器的数据包规则。在Python中,要操作MySQL,需要依赖于第三方库:pymysql,先安装:pip install pymysql
代码实现
common.py(通用函数模块)
import pymysql
from pymysql.cursors import DictCursor
#针对数据库连接进行封装操作
def query_mysql(sql):conn = pymysql.connect(host='localhost', user='root', password='root', database='security', charset='utf8')cursor = conn.cursor()cursor.execute(sql)result=cursor.fetchall()conn.close()return result
app.py(主程序)
from exercise.common import *
def do_reg():username=input("请输入用户名:")password=input("请输入密码:")result=query_mysql(f"select username from users where username='{username}'")if len(result)==0:query_mysql(f"insert into users (username,password) values ('{username}','{password}')")else:print("用户名已经存在,不允许注册.")if __name__ == '__main__':do_reg()