141 lines
5.0 KiB
Markdown
141 lines
5.0 KiB
Markdown
# 照片管理系统项目文档
|
||
|
||
## 项目概述
|
||
本项目是一个基于 Flask 框架的照片管理系统，主要功能包括：
|
||
1. **人脸管理**：增删改查用户的人脸数据。
|
||
2. **照片上传与分类**：用户可以上传照片，系统会根据 EXIF 时间信息和数据库中的人脸数据进行分类存储。
|
||
3. **照片查询**：用户可以根据姓名和日期查找相应的照片。
|
||
|
||
## 技术栈
|
||
- **后端**：Flask（Python）
|
||
- **数据库**：SQLAlchemy（SQLite / MySQL）
|
||
- **前端**：HTML + CSS + Jinja2
|
||
- **存储**：文件系统存储上传的照片
|
||
|
||
## 数据库设计
|
||
### 1. `users` 表（用户信息表）
|
||
| 字段名 | 类型 | 说明 |
|
||
|--------|------|------|
|
||
| id | 主键 | 自动生成 |
|
||
| username | 字符串 | 用户名，唯一且非空 |
|
||
| encoding | 二进制数据 | 人脸识别编码，用于匹配 |
|
||
| photo | 字符串 | 用户头像照片路径 |
|
||
|
||
### 2. `photos` 表（照片信息表）
|
||
| 字段名 | 类型 | 说明 |
|
||
|--------|------|------|
|
||
| id | 主键 | 自动生成 |
|
||
| user_id | 外键 | 关联到 `users` 表的 `id` |
|
||
| created_at | 时间戳 | 照片拍摄时间（来自 EXIF 信息） |
|
||
| classification | 布尔值 | 照片是否分类（默认 `False`，分类成功后 `True`） |
|
||
| classification_confidence | 浮点数 | 照片分类置信度（默认 `0`） |
|
||
| classification_image_path | 字符串 | 分类后的照片存储路径 |
|
||
|
||
## 主要功能实现
|
||
### 1. 用户管理
|
||
- **添加用户**：用户可上传照片，系统提取人脸编码并存入数据库。
|
||
- **查看用户列表**：显示所有用户及其头像。
|
||
- **删除用户**：从数据库中删除用户信息及相关照片。
|
||
|
||
### 2. 照片上传与分类
|
||
- **上传照片**：用户上传照片，系统解析 EXIF 数据确定拍摄时间。
|
||
- **人脸匹配分类**：系统根据数据库中的人脸编码进行匹配，分类并存储到相应目录。
|
||
- **存储路径规则**：
|
||
```
|
||
static/uploads/users/{用户名}/{日期（年-月-日）}/照片文件名
|
||
```
|
||
|
||
### 3. 照片查询
|
||
- **基于姓名和日期查询**：用户输入姓名和日期，系统返回符合条件的照片。
|
||
- **展示分类结果**：前端页面显示照片缩略图，并提供原始照片下载链接。
|
||
|
||
## 关键代码
|
||
### 1. 照片上传（`upload` 端点）
|
||
```python
|
||
@app.route('/upload', methods=['GET', 'POST'])
|
||
def upload():
|
||
if request.method == 'POST':
|
||
photos = request.files.getlist('photos')
|
||
classify_photos(photos) # 进行分类处理
|
||
|
||
# 获取最新照片并关联用户信息
|
||
recent_photos = Photo.query.order_by(Photo.created_at.desc()).limit(len(photos)).all()
|
||
photos_with_users = []
|
||
for photo in recent_photos:
|
||
user = User.query.get(photo.user_id)
|
||
photos_with_users.append({
|
||
"filename": photo.classification_image_path,
|
||
"username": user.username if user else "未知",
|
||
"created_at": photo.created_at
|
||
})
|
||
|
||
return render_template('upload.html', photos=photos_with_users)
|
||
|
||
return render_template('upload.html', photos=[])
|
||
```
|
||
|
||
### 2. 照片查询（`search` 端点）
|
||
```python
|
||
@app.route('/search', methods=['POST'])
|
||
def search():
|
||
name = request.form['name']
|
||
date = request.form['date']
|
||
|
||
user = User.query.filter_by(username=name).first()
|
||
if not user:
|
||
return render_template('search.html', photos=[], message="用户不存在")
|
||
|
||
photos = Photo.query.filter_by(user_id=user.id).filter(Photo.created_at.like(f"{date}%")).all()
|
||
return render_template('search.html', photos=photos)
|
||
```
|
||
|
||
## 前端页面示例
|
||
### 1. 照片查询页面（`search.html`）
|
||
```html
|
||
<form action="{{ url_for('search') }}" method="POST">
|
||
<label for="name">姓名：</label>
|
||
<input type="text" id="name" name="name" placeholder="请输入用户名" required>
|
||
<label for="date">日期：</label>
|
||
<input type="date" id="date" name="date" required>
|
||
<button type="submit">查询照片</button>
|
||
</form>
|
||
```
|
||
|
||
## 目录结构
|
||
```
|
||
project_root/
|
||
│── app.py # Flask 入口文件
|
||
│── functions.py # 处理照片分类等功能
|
||
│── models.py # 数据库模型
|
||
│── templates/
|
||
│ ├── base.html # 公共模板
|
||
│ ├── upload.html # 上传页面
|
||
│ ├── search.html # 查询页面
|
||
│── static/
|
||
│ ├── uploads/ # 存储照片
|
||
│ ├── styles.css # 样式表
|
||
│── .venv/ # 虚拟环境
|
||
│── requirements.txt # 依赖文件
|
||
```
|
||
|
||
## 部署方式
|
||
1. **安装依赖**
|
||
```sh
|
||
pip install -r requirements.txt
|
||
```
|
||
2. **运行 Flask 服务器**
|
||
```sh
|
||
flask run
|
||
```
|
||
3. **访问应用**
|
||
在浏览器中打开 `http://127.0.0.1:5000`
|
||
|
||
## 未来优化方向
|
||
- **添加身份验证**：使用 Flask-Login 实现用户登录功能。
|
||
- **优化分类算法**：提高人脸识别和分类的准确率。
|
||
- **支持云存储**：将照片存储到云端，如 AWS S3 或阿里云 OSS。
|
||
|
||
---
|
||
本项目旨在提供一个高效、易用的照片管理系统，欢迎优化和扩展！
|
||
|