AI智能文档扫描仪权限控制：多用户访问安全管理

1. 引言

1.1 业务场景描述

随着远程办公和数字化协作的普及，企业对文档电子化处理的需求日益增长。AI 智能文档扫描仪作为一种轻量高效的图像处理工具，广泛应用于合同归档、发票识别、教学资料整理等场景。然而，在团队共享使用环境中，如何保障敏感文件不被未授权人员访问，成为系统安全的关键挑战。

当前多数本地化部署的扫描工具仅提供基础功能，缺乏对多用户身份识别与权限隔离的支持。一旦服务暴露在局域网或公网中，任何人均可通过 WebUI 上传、查看甚至下载他人处理过的文档，存在严重的数据泄露风险。

1.2 痛点分析

原生 Smart Doc Scanner 虽具备“零依赖、高隐私”的优势，但其设计初衷是单机个人使用，存在以下安全短板：

无认证机制：Web 页面无需登录即可访问全部功能。
文件路径公开：上传与输出目录直接暴露，易被枚举访问。
无操作审计：无法追踪谁在何时处理了哪些文件。
缺乏权限分级：管理员与普通用户无区别对待。

这些问题使得该工具难以满足企业级应用中的合规性要求（如 GDPR、ISO 27001）。

1.3 方案预告

本文将介绍一种轻量级多用户权限控制系统，可在不影响原有 OpenCV 图像处理性能的前提下，为 AI 智能文档扫描仪增加三层安全防护：

用户身份认证（Authentication）
访问权限控制（Authorization）
文件隔离与操作日志记录

整个方案基于 Flask + SQLite 实现，代码简洁可嵌入，适合资源受限环境部署。

2. 技术方案选型

2.1 安全架构设计目标

目标	描述
轻量化	不引入复杂框架，避免影响原系统启动速度
零模型依赖	继承原始项目“纯算法”理念，不依赖外部 AI 模型
易集成	可作为中间件插入现有 WebUI 流程
可扩展	支持后续添加角色管理、API 密钥等功能

2.2 候选技术对比

方案	认证方式	数据库	是否需外网	部署复杂度	适用性
Flask-Login + SQLite	Session	内置	否	★★☆	✅ 推荐
JWT + Redis	Token	缓存	否	★★★	⚠️ 过重
OAuth2 (GitHub)	第三方	外部	是	★★★★	❌ 依赖网络
Basic Auth	HTTP Header	文件	否	★☆	❌ 无会话管理

综合评估后，选择Flask-Login + SQLite组合，兼顾安全性与轻量化需求。

3. 实现步骤详解

3.1 系统架构调整

在原有app.py基础上新增以下模块：

smart_doc_scanner/ ├── app.py # 主应用入口 ├── auth/ # 权限控制模块 │ ├── __init__.py │ ├── models.py # 用户模型 │ ├── routes.py # 登录/登出路由 ├── static/uploads/ # 按用户隔离存储 ├── templates/login.html # 登录页面 └── config.py # 安全配置

3.2 核心代码实现

用户模型定义（`auth/models.py`）

from flask_sqlalchemy import SQLAlchemy from flask_login import UserMixin from werkzeug.security import generate_password_hash, check_password_hash db = SQLAlchemy() class User(UserMixin, db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(80), unique=True, nullable=False) password_hash = db.Column(db.String(200), nullable=False) role = db.Column(db.String(20), default='user') # user, admin def set_password(self, password): self.password_hash = generate_password_hash(password) def check_password(self, password): return check_password_hash(self.password_hash, password)

说明：采用Werkzeug提供的哈希函数加密密码，防止明文存储；UserMixin支持 Flask-Login 快速集成。

登录路由（`auth/routes.py`）

from flask import Blueprint, request, render_template, redirect, url_for, flash from flask_login import login_user, logout_user, current_user from .models import User, db bp = Blueprint('auth', __name__) @bp.route('/login', methods=['GET', 'POST']) def login(): if current_user.is_authenticated: return redirect(url_for('main.index')) if request.method == 'POST': username = request.form['username'] password = request.form['password'] user = User.query.filter_by(username=username).first() if user and user.check_password(password): login_user(user) return redirect(url_for('main.index')) else: flash('用户名或密码错误') return render_template('login.html') @bp.route('/logout') def logout(): logout_user() return redirect(url_for('auth.login'))

文件上传路径隔离（`app.py`修改片段）

import os from flask_login import login_required, current_user @app.route('/upload', methods=['POST']) @login_required def upload_file(): file = request.files['file'] if file: # 按用户ID创建独立目录 user_dir = os.path.join('static/uploads', str(current_user.id)) os.makedirs(user_dir, exist_ok=True) filepath = os.path.join(user_dir, file.filename) file.save(filepath) # 记录操作日志 with open('logs/access.log', 'a') as f: f.write(f"{datetime.now()} - {current_user.username} uploaded {file.filename}\n") return jsonify({"status": "success", "path": filepath}) return jsonify({"status": "fail"})

关键点：
使用@login_required装饰器保护所有敏感接口
用户上传文件保存至/static/uploads/<user_id>/目录下，实现物理隔离
操作行为写入日志文件，便于审计追溯

前端登录页（`templates/login.html`）

<!DOCTYPE html> <html> <head><title>登录 - 智能文档扫描仪</title></head> <body style="text-align:center; margin-top:100px;"> <h2>🔐 登录系统</h2> <form method="POST" style="display:inline-block;padding:20px;border:1px solid #ccc;"> <p> <label>用户名：<br> <input type="text" name="username" required style="padding:5px;width:200px;"> </label> </p> <p> <label>密码：<br> <input type="password" name="password" required style="padding:5px;width:200px;"> </label> </p> <p><button type="submit" style="padding:10px 20px;">登录</button></p> {% with messages = get_flashed_messages() %} {% if messages %} <div style="color:red;">{{ messages[0] }}</div> {% endif %} {% endwith %} </form> </body> </html>

4. 实践问题与优化

4.1 遇到的问题及解决方案

问题	原因	解决方法
图像预览失败	Nginx 静态路径未授权	配置`/uploads`目录按用户权限代理
多用户并发冲突	SQLite 写锁阻塞	添加`timeout=20`参数提升容错
初始无管理员账号	需手动初始化数据库	提供`init_db.py`脚本自动创建 admin 用户

初始化脚本示例（`init_db.py`）

from auth.models import db, User db.create_all() if not User.query.filter_by(username='admin').first(): admin = User(username='admin', role='admin') admin.set_password('Admin@123') db.session.add(admin) db.session.commit() print("✅ 管理员账户已创建: admin / Admin@123") else: print("ℹ️ 管理员账户已存在")