用python写一个能连接数据库的登陆注册系统
时间: 2023-06-19 16:05:25 浏览: 163
好的,我可以提供一个基于Python和MySQL实现的简单登陆注册系统的代码。首先,需要安装MySQL Connector库,可以使用下面的命令进行安装:
```
pip install mysql-connector-python
```
然后创建一个MySQL数据库,可以使用下面的命令创建一个名为“users”的表:
```sql
CREATE TABLE users (
id INT(11) NOT NULL AUTO_INCREMENT,
username VARCHAR(50) NOT NULL,
password VARCHAR(50) NOT NULL,
PRIMARY KEY (id)
);
```
接下来是Python代码:
```python
import mysql.connector
# 连接数据库
db = mysql.connector.connect(
host="localhost",
user="root",
password="password",
database="test"
)
# 创建游标
cursor = db.cursor()
# 登录函数
def login():
print("请输入用户名和密码:")
username = input("用户名:")
password = input("密码:")
sql = "SELECT * FROM users WHERE username=%s AND password=%s"
cursor.execute(sql, (username, password))
result = cursor.fetchone()
if result:
print("登录成功")
else:
print("用户名或密码错误")
# 注册函数
def register():
print("请输入用户名和密码:")
username = input("用户名:")
password = input("密码:")
sql = "INSERT INTO users (username, password) VALUES (%s, %s)"
cursor.execute(sql, (username, password))
db.commit()
print("注册成功")
# 主函数
def main():
while True:
print("请选择操作:")
print("1. 登录")
print("2. 注册")
print("3. 退出")
choice = input("请选择:")
if choice == "1":
login()
elif choice == "2":
register()
elif choice == "3":
break
else:
print("无效选择")
# 执行主函数
if __name__ == "__main__":
main()
# 关闭数据库连接
db.close()
```
这段代码实现了一个简单的登陆注册系统,用户可以选择登录或注册,用户名和密码将存储在MySQL数据库中。当用户登录时,程序将从数据库中检索用户名和密码,如果匹配成功,则登录成功;当用户注册时,程序将向数据库中插入新的用户名和密码。
阅读全文