编写一段登陆注册代码,引入数据库
时间: 2023-04-07 16:01:30 浏览: 45
好的,以下是一个简单的登陆注册代码,使用了MySQL数据库:
```
import mysql.connector
# 连接数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="mydatabase"
)
# 创建用户表
mycursor = mydb.cursor()
mycursor.execute("CREATE TABLE users (id INT AUTO_INCREMENT PRIMARY KEY, username VARCHAR(255), password VARCHAR(255))")
# 注册函数
def register(username, password):
# 检查用户名是否已存在
mycursor.execute("SELECT * FROM users WHERE username = %s", (username,))
result = mycursor.fetchone()
if result:
print("该用户名已被注册")
return False
# 插入新用户
sql = "INSERT INTO users (username, password) VALUES (%s, %s)"
val = (username, password)
mycursor.execute(sql, val)
mydb.commit()
print("注册成功")
return True
# 登陆函数
def login(username, password):
# 检查用户名和密码是否匹配
mycursor.execute("SELECT * FROM users WHERE username = %s AND password = %s", (username, password))
result = mycursor.fetchone()
if result:
print("登陆成功")
return True
else:
print("用户名或密码错误")
return False
```
请注意,这只是一个简单的示例代码,实际应用中需要更多的安全措施,例如加密密码、防止 SQL 注入等。