档案管理系统代码
时间: 2023-08-06 16:14:37 浏览: 144
这是一个简单的档案管理系统代码,使用Python和SQLite数据库实现。主要功能包括添加、查询、修改和删除档案信息。
```python
import sqlite3
# 连接数据库
conn = sqlite3.connect('archive.db')
# 创建表格
conn.execute('''CREATE TABLE IF NOT EXISTS ARCHIVE
(ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
GENDER CHAR(10) NOT NULL,
ADDRESS CHAR(50) NOT NULL);''')
print("档案管理系统")
while True:
print("请选择操作:")
print("1. 添加档案")
print("2. 查询档案")
print("3. 修改档案")
print("4. 删除档案")
print("0. 退出")
choice = int(input())
if choice == 1:
# 添加档案
id = int(input("请输入ID:"))
name = input("请输入姓名:")
age = int(input("请输入年龄:"))
gender = input("请输入性别:")
address = input("请输入地址:")
conn.execute(f"INSERT INTO ARCHIVE (ID, NAME, AGE, GENDER, ADDRESS) VALUES ({id}, '{name}', {age}, '{gender}', '{address}')")
conn.commit()
print("添加成功")
elif choice == 2:
# 查询档案
keyword = input("请输入查询关键字:")
cursor = conn.execute(f"SELECT * FROM ARCHIVE WHERE NAME LIKE '%{keyword}%' OR ADDRESS LIKE '%{keyword}%'")
for row in cursor:
print(f"ID: {row[0]}")
print(f"姓名: {row[1]}")
print(f"年龄: {row[2]}")
print(f"性别: {row[3]}")
print(f"地址: {row[4]}")
elif choice == 3:
# 修改档案
id = int(input("请输入要修改的档案ID:"))
name = input("请输入新的姓名:")
age = int(input("请输入新的年龄:"))
gender = input("请输入新的性别:")
address = input("请输入新的地址:")
conn.execute(f"UPDATE ARCHIVE SET NAME='{name}', AGE={age}, GENDER='{gender}', ADDRESS='{address}' WHERE ID={id}")
conn.commit()
print("修改成功")
elif choice == 4:
# 删除档案
id = int(input("请输入要删除的档案ID:"))
conn.execute(f"DELETE FROM ARCHIVE WHERE ID={id}")
conn.commit()
print("删除成功")
elif choice == 0:
# 退出
break
else:
print("无效的选择")
# 关闭数据库连接
conn.close()
```
注意:这个代码只是一个示例,实际开发中需要考虑更多的异常情况和安全问题。同时,数据库操作语句需要预防SQL注入攻击。