python如何做简单登录数据库,具有查找,保存TXT文件,查询、添加、修改等功能
时间: 2024-02-16 13:59:47 浏览: 22
从TXT文件中,找自己要的数据并且保存在数据库中
可以使用Python内置的sqlite3模块来实现简单的数据库操作,同时也可以使用Python的文件读写操作来实现数据的保存。以下是一个示例代码,实现了查找、保存、查询、添加和修改等功能:
```python
import sqlite3
# 数据库连接
conn = sqlite3.connect('user.db')
c = conn.cursor()
# 创建表格
c.execute('''CREATE TABLE IF NOT EXISTS users
(id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT NOT NULL,
password TEXT NOT NULL)''')
conn.commit()
# 添加一条数据
def add_user(username, password):
c.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, password))
conn.commit()
# 查找一条数据
def find_user(username):
c.execute("SELECT * FROM users WHERE username=?", (username,))
return c.fetchone()
# 修改一条数据
def modify_user(username, password):
c.execute("UPDATE users SET password=? WHERE username=?", (password, username))
conn.commit()
# 保存数据到TXT文件
def save_to_file(filename):
c.execute("SELECT * FROM users")
with open(filename, 'w') as f:
for row in c.fetchall():
f.write(str(row[0]) + ',' + row[1] + ',' + row[2] + '\n')
# 从TXT文件加载数据
def load_from_file(filename):
with open(filename, 'r') as f:
for line in f:
parts = line.strip().split(',')
add_user(parts[1], parts[2])
# 测试代码
add_user('alice', '123456')
add_user('bob', '654321')
add_user('charlie', 'abcdef')
save_to_file('users.txt')
alice = find_user('alice')
print(alice)
modify_user('alice', 'new_password')
alice = find_user('alice')
print(alice)
load_from_file('users.txt')
```
以上代码演示了如何通过SQLite3模块实现简单的数据库操作,同时也演示了如何通过文件读写来保存和加载数据。通过这些代码可以实现基本的查询、添加、修改和保存等功能。当然,实际的应用场景可能需要更完善的安全性和可靠性措施。
阅读全文