majiahui@haimaqingfan.com
3 days ago
3 changed files with 509 additions and 60 deletions
@ -0,0 +1,410 @@ |
|||||
|
import os |
||||
|
from flask import Flask, request, render_template_string, redirect, url_for, jsonify |
||||
|
from werkzeug.utils import secure_filename |
||||
|
from flask_cors import CORS |
||||
|
from clickhouse_client import clickhouse_client |
||||
|
from config import Config |
||||
|
from datetime import datetime |
||||
|
|
||||
|
|
||||
|
app = Flask(__name__) |
||||
|
CORS(app) # 允许跨域请求 |
||||
|
app.config.from_object(Config) |
||||
|
|
||||
|
# 配置 |
||||
|
UPLOAD_FOLDER = 'uploads' |
||||
|
ALLOWED_EXTENSIONS = {'txt'} |
||||
|
|
||||
|
# 确保上传目录存在 |
||||
|
if not os.path.exists(UPLOAD_FOLDER): |
||||
|
os.makedirs(UPLOAD_FOLDER) |
||||
|
|
||||
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER |
||||
|
|
||||
|
def upload_file(): |
||||
|
message = '' |
||||
|
filename = '' |
||||
|
|
||||
|
# 检查是否有文件部分 |
||||
|
if 'file' not in request.files: |
||||
|
message = '没有文件部分' |
||||
|
return_info = { |
||||
|
"code": 203, |
||||
|
"message": message |
||||
|
} |
||||
|
return jsonify(return_info) |
||||
|
|
||||
|
file = request.files['file'] |
||||
|
|
||||
|
# 如果用户没有选择文件 |
||||
|
if file.filename == '': |
||||
|
message = '没有选择文件' |
||||
|
return_info = { |
||||
|
"code": 203, |
||||
|
"message": message |
||||
|
} |
||||
|
|
||||
|
return jsonify(return_info) |
||||
|
|
||||
|
if file and allowed_file(file.filename): |
||||
|
# 安全处理文件名 |
||||
|
filename = secure_filename(file.filename) |
||||
|
|
||||
|
# 处理同名文件：添加时间戳 |
||||
|
name, ext = os.path.splitext(filename) |
||||
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") |
||||
|
filename = f"{name}_{timestamp}{ext}" |
||||
|
|
||||
|
# 保存文件 |
||||
|
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename) |
||||
|
file.save(filepath) |
||||
|
|
||||
|
message = '文件上传成功！' |
||||
|
return_info = { |
||||
|
"code": 203, |
||||
|
"message": message |
||||
|
} |
||||
|
|
||||
|
return jsonify(return_info) |
||||
|
else: |
||||
|
message = '文件类型不允许' |
||||
|
|
||||
|
def allowed_file(filename): |
||||
|
"""检查文件扩展名是否允许""" |
||||
|
return '.' in filename and \ |
||||
|
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS |
||||
|
|
||||
|
# ========== API路由 ========== |
||||
|
|
||||
|
@app.route('/') |
||||
|
def index(): |
||||
|
"""首页""" |
||||
|
return jsonify({ |
||||
|
'message': 'ClickHouse Flask API', |
||||
|
'endpoints': { |
||||
|
'GET /health': '检查服务健康状态', |
||||
|
'POST /users/<user_uuid>/records': '创建用户记录', |
||||
|
'POST /users/<user_uuid>/records/batch': '批量创建记录', |
||||
|
'GET /users/<user_uuid>/records': '获取用户所有记录', |
||||
|
'GET /users/<user_uuid>/records/<record_id>': '获取单条记录', |
||||
|
'PUT /users/<user_uuid>/records/<record_id>': '更新记录', |
||||
|
'DELETE /users/<user_uuid>/records/<record_id>': '删除单条记录', |
||||
|
'DELETE /users/<user_uuid>/records': '删除用户所有记录', |
||||
|
'PUT /users/<user_uuid>/records/<record_id>/restore': '恢复已删除记录', |
||||
|
'GET /users/<user_uuid>/search': '搜索用户记录', |
||||
|
'GET /users/<user_uuid>/stats': '获取用户统计信息' |
||||
|
} |
||||
|
}) |
||||
|
|
||||
|
|
||||
|
@app.route('/health', methods=['GET']) |
||||
|
def health_check(): |
||||
|
"""健康检查""" |
||||
|
try: |
||||
|
# 测试ClickHouse连接 |
||||
|
clickhouse_client.client.execute('SELECT 1') |
||||
|
return jsonify({'status': 'healthy', 'database': 'connected'}) |
||||
|
except Exception as e: |
||||
|
return jsonify({'status': 'unhealthy', 'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
# ========== 用户记录操作 ========== |
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records', methods=['POST']) |
||||
|
def create_record(user_uuid): |
||||
|
"""创建用户记录""" |
||||
|
try: |
||||
|
data = request.get_json() |
||||
|
if not data: |
||||
|
return jsonify({'error': 'No data provided'}), 400 |
||||
|
|
||||
|
file_path = data.get('file_path') |
||||
|
is_delete = data.get('is_delete', 0) |
||||
|
|
||||
|
if not file_path: |
||||
|
return jsonify({'error': 'file_path is required'}), 400 |
||||
|
|
||||
|
# 调用更新后的方法，返回record_id |
||||
|
record_id = clickhouse_client.create_user_record(user_uuid, file_path, is_delete) |
||||
|
|
||||
|
if record_id: |
||||
|
return jsonify({ |
||||
|
'message': 'Record created successfully', |
||||
|
'user_uuid': user_uuid, |
||||
|
'record_id': record_id, |
||||
|
'file_path': file_path |
||||
|
}), 201 |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to create record'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records/batch', methods=['POST']) |
||||
|
def batch_create_records(user_uuid): |
||||
|
"""批量创建用户记录""" |
||||
|
try: |
||||
|
data = request.get_json() |
||||
|
if not data or not isinstance(data, list): |
||||
|
return jsonify({'error': 'Data must be a list of records'}), 400 |
||||
|
|
||||
|
# 为每条记录添加user_uuid |
||||
|
records = [] |
||||
|
for item in data: |
||||
|
if not isinstance(item, dict): |
||||
|
continue |
||||
|
|
||||
|
record = { |
||||
|
'user_uuid': user_uuid, |
||||
|
'file_path': item.get('file_path', ''), |
||||
|
'is_delete': item.get('is_delete', 0) |
||||
|
} |
||||
|
|
||||
|
if record['file_path']: # 只添加有文件路径的记录 |
||||
|
records.append(record) |
||||
|
|
||||
|
if not records: |
||||
|
return jsonify({'error': 'No valid records provided'}), 400 |
||||
|
|
||||
|
# 调用更新后的方法，返回record_ids |
||||
|
record_ids = clickhouse_client.batch_create_records(records) |
||||
|
|
||||
|
if record_ids: |
||||
|
return jsonify({ |
||||
|
'message': f'{len(record_ids)} records created successfully', |
||||
|
'user_uuid': user_uuid, |
||||
|
'record_ids': record_ids, |
||||
|
'count': len(record_ids) |
||||
|
}), 201 |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to create records'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records', methods=['GET']) |
||||
|
def get_records(user_uuid): |
||||
|
"""获取用户的所有记录""" |
||||
|
try: |
||||
|
include_deleted = request.args.get('include_deleted', 'false').lower() == 'true' |
||||
|
|
||||
|
records = clickhouse_client.get_user_records(user_uuid, include_deleted) |
||||
|
|
||||
|
return jsonify({ |
||||
|
'user_uuid': user_uuid, |
||||
|
'count': len(records), |
||||
|
'records': records |
||||
|
}) |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records/<record_id>', methods=['GET']) |
||||
|
def get_single_record(user_uuid, record_id): |
||||
|
"""获取单条记录""" |
||||
|
try: |
||||
|
record = clickhouse_client.get_user_record_by_id(record_id) |
||||
|
|
||||
|
if not record: |
||||
|
return jsonify({'error': 'Record not found'}), 404 |
||||
|
|
||||
|
if record['user_uuid'] != user_uuid: |
||||
|
return jsonify({'error': 'Record does not belong to this user'}), 403 |
||||
|
|
||||
|
return jsonify(record) |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records/<record_id>', methods=['PUT']) |
||||
|
def update_record(user_uuid, record_id): |
||||
|
"""更新记录""" |
||||
|
try: |
||||
|
# 先检查记录是否存在且属于该用户 |
||||
|
record = clickhouse_client.get_user_record_by_id(record_id) |
||||
|
if not record: |
||||
|
return jsonify({'error': 'Record not found'}), 404 |
||||
|
|
||||
|
if record['user_uuid'] != user_uuid: |
||||
|
return jsonify({'error': 'Record does not belong to this user'}), 403 |
||||
|
|
||||
|
data = request.get_json() |
||||
|
if not data: |
||||
|
return jsonify({'error': 'No data provided'}), 400 |
||||
|
|
||||
|
file_path = data.get('file_path') |
||||
|
is_delete = data.get('is_delete') |
||||
|
|
||||
|
success = clickhouse_client.update_user_record(record_id, file_path, is_delete) |
||||
|
|
||||
|
if success: |
||||
|
return jsonify({ |
||||
|
'message': 'Record updated successfully', |
||||
|
'record_id': record_id, |
||||
|
'user_uuid': user_uuid |
||||
|
}) |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to update record'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records/<record_id>', methods=['DELETE']) |
||||
|
def delete_record(user_uuid, record_id): |
||||
|
"""删除单条记录""" |
||||
|
try: |
||||
|
# 先检查记录是否存在且属于该用户 |
||||
|
record = clickhouse_client.get_user_record_by_id(record_id) |
||||
|
if not record: |
||||
|
return jsonify({'error': 'Record not found'}), 404 |
||||
|
|
||||
|
if record['user_uuid'] != user_uuid: |
||||
|
return jsonify({'error': 'Record does not belong to this user'}), 403 |
||||
|
|
||||
|
# 获取删除类型参数 |
||||
|
soft_delete = request.args.get('soft_delete', 'true').lower() == 'true' |
||||
|
|
||||
|
success = clickhouse_client.delete_user_record(record_id, soft_delete) |
||||
|
|
||||
|
if success: |
||||
|
action = 'soft deleted' if soft_delete else 'permanently deleted' |
||||
|
return jsonify({ |
||||
|
'message': f'Record {action} successfully', |
||||
|
'record_id': record_id, |
||||
|
'user_uuid': user_uuid |
||||
|
}) |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to delete record'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records', methods=['DELETE']) |
||||
|
def delete_all_records(user_uuid): |
||||
|
"""删除用户的所有记录""" |
||||
|
try: |
||||
|
# 获取删除类型参数 |
||||
|
soft_delete = request.args.get('soft_delete', 'true').lower() == 'true' |
||||
|
|
||||
|
success = clickhouse_client.delete_all_user_records(user_uuid, soft_delete) |
||||
|
|
||||
|
if success: |
||||
|
action = 'soft deleted' if soft_delete else 'permanently deleted' |
||||
|
return jsonify({ |
||||
|
'message': f'All records for user {user_uuid} have been {action}', |
||||
|
'user_uuid': user_uuid |
||||
|
}) |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to delete records'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/records/<record_id>/restore', methods=['PUT']) |
||||
|
def restore_record(user_uuid, record_id): |
||||
|
"""恢复已删除的记录""" |
||||
|
try: |
||||
|
# 先检查记录是否存在 |
||||
|
record = clickhouse_client.get_user_record_by_id(record_id) |
||||
|
if not record: |
||||
|
return jsonify({'error': 'Record not found'}), 404 |
||||
|
|
||||
|
if record['user_uuid'] != user_uuid: |
||||
|
return jsonify({'error': 'Record does not belong to this user'}), 403 |
||||
|
|
||||
|
success = clickhouse_client.restore_user_record(record_id) |
||||
|
|
||||
|
if success: |
||||
|
return jsonify({ |
||||
|
'message': 'Record restored successfully', |
||||
|
'record_id': record_id, |
||||
|
'user_uuid': user_uuid |
||||
|
}) |
||||
|
else: |
||||
|
return jsonify({'error': 'Failed to restore record'}), 500 |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/search', methods=['GET']) |
||||
|
def search_records(user_uuid): |
||||
|
"""搜索用户记录""" |
||||
|
try: |
||||
|
keyword = request.args.get('keyword') |
||||
|
is_delete_param = request.args.get('is_delete') |
||||
|
|
||||
|
is_delete = None |
||||
|
if is_delete_param is not None: |
||||
|
is_delete = int(is_delete_param) |
||||
|
|
||||
|
records = clickhouse_client.search_user_records(user_uuid, keyword, is_delete) |
||||
|
|
||||
|
return jsonify({ |
||||
|
'user_uuid': user_uuid, |
||||
|
'count': len(records), |
||||
|
'records': records |
||||
|
}) |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
@app.route('/users/<user_uuid>/stats', methods=['GET']) |
||||
|
def get_user_stats(user_uuid): |
||||
|
"""获取用户统计信息""" |
||||
|
try: |
||||
|
stats = clickhouse_client.get_user_stats(user_uuid) |
||||
|
|
||||
|
if stats: |
||||
|
return jsonify({ |
||||
|
'user_uuid': user_uuid, |
||||
|
'stats': stats |
||||
|
}) |
||||
|
else: |
||||
|
return jsonify({ |
||||
|
'user_uuid': user_uuid, |
||||
|
'stats': { |
||||
|
'total_count': 0, |
||||
|
'active_count': 0, |
||||
|
'deleted_count': 0, |
||||
|
'first_created': None, |
||||
|
'last_created': None |
||||
|
} |
||||
|
}) |
||||
|
|
||||
|
except Exception as e: |
||||
|
return jsonify({'error': str(e)}), 500 |
||||
|
|
||||
|
|
||||
|
# ========== 错误处理 ========== |
||||
|
|
||||
|
@app.errorhandler(404) |
||||
|
def not_found(error): |
||||
|
return jsonify({'error': 'Not found'}), 404 |
||||
|
|
||||
|
|
||||
|
@app.errorhandler(500) |
||||
|
def internal_error(error): |
||||
|
return jsonify({'error': 'Internal server error'}), 500 |
||||
|
|
||||
|
|
||||
|
# ========== 主程序 ========== |
||||
|
|
||||
|
if __name__ == '__main__': |
||||
|
# 创建上传目录（如果需要） |
||||
|
os.makedirs('uploads', exist_ok=True) |
||||
|
|
||||
|
# 启动Flask应用 |
||||
|
app.run( |
||||
|
host='0.0.0.0', |
||||
|
port=28002, |
||||
|
debug=Config.DEBUG |
||||
|
) |
||||
Loading…
Reference in new issue