xiaji/temp-file-trans
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(traffic): add per-IP daily traffic tracking with 20GB limit for uploads and downloads

Browse Source
This commit is contained in:
OpenCode Bot
2026-05-01 09:02:39 +08:00
parent 3bdb417f10
commit f5727ec176
3 changed files with 90 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
app.py
View File

@@ -2,8 +2,8 @@ import os
import uuid
from datetime import datetime
from flask import Flask, request, render_template, send_file, jsonify, url_for, abort
from config import UPLOAD_FOLDER, SECRET_KEY, MAX_CONTENT_LENGTH, EXPIRY_OPTIONS
from database import init_db, add_file, get_file, delete_file, cleanup_expired
from config import UPLOAD_FOLDER, SECRET_KEY, MAX_CONTENT_LENGTH, EXPIRY_OPTIONS, DAILY_TRAFFIC_LIMIT
from database import init_db, add_file, get_file, delete_file, cleanup_expired, add_upload_traffic, add_download_traffic, get_client_ip, is_traffic_exceeded, get_daily_traffic
app = Flask(__name__)
app.config['SECRET_KEY'] = SECRET_KEY
@@ -30,6 +30,12 @@ def upload():
expiry_seconds = EXPIRY_OPTIONS.get(expiry_key, EXPIRY_OPTIONS['24h'])
expiry_hours = expiry_seconds // 3600
ip = get_client_ip(request)
content_length = request.content_length or 0
if is_traffic_exceeded(ip, content_length, 'upload'):
return jsonify({'error': 'Daily traffic limit exceeded (20GB)'}), 429
file_id = str(uuid.uuid4())
filename = file.filename
filepath = os.path.join(UPLOAD_FOLDER, file_id)
@@ -37,6 +43,7 @@ def upload():
filesize = os.path.getsize(filepath)
add_file(file_id, filename, filepath, filesize, expiry_hours)
add_upload_traffic(ip, filesize)
share_url = url_for('download_file', file_id=file_id, _external=True)
return jsonify({'id': file_id, 'filename': filename, 'share_url': share_url})
@@ -54,6 +61,12 @@ def api_upload():
expiry_seconds = EXPIRY_OPTIONS.get(expiry_key, EXPIRY_OPTIONS['24h'])
expiry_hours = expiry_seconds // 3600
ip = get_client_ip(request)
content_length = request.content_length or 0
if is_traffic_exceeded(ip, content_length, 'upload'):
return jsonify({'error': 'Daily traffic limit exceeded (20GB)'}), 429
file_id = str(uuid.uuid4())
filename = file.filename
filepath = os.path.join(UPLOAD_FOLDER, file_id)
@@ -61,6 +74,7 @@ def api_upload():
filesize = os.path.getsize(filepath)
add_file(file_id, filename, filepath, filesize, expiry_hours)
add_upload_traffic(ip, filesize)
share_url = url_for('download_file', file_id=file_id, _external=True)
return jsonify({'id': file_id, 'filename': filename, 'filesize': filesize, 'expiry_hours': expiry_hours, 'share_url': share_url})
@@ -79,12 +93,18 @@ def api_get_file(file_id):
row = get_file(file_id)
if not row:
return jsonify({'error': 'File not found or expired'}), 404
ip = get_client_ip(request)
upload, download = get_daily_traffic(ip)
return jsonify({
'id': row['id'],
'filename': row['filename'],
'filesize': row['filesize'],
'created_at': row['created_at'],
'expires_at': row['expires_at']
'expires_at': row['expires_at'],
'daily_upload': upload,
'daily_download': download,
'traffic_limit': DAILY_TRAFFIC_LIMIT
})
@app.route('/download/<file_id>')
@@ -92,6 +112,10 @@ def serve_file(file_id):
row = get_file(file_id)
if not row:
abort(404)
ip = get_client_ip(request)
add_download_traffic(ip, row['filesize'])
return send_file(row['filepath'], download_name=row['filename'], as_attachment=True)
if __name__ == '__main__':