Update main.py

main.py

@@ -1,64 +1,67 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, Response
 import random
 import string
+from collections import deque
+import threading
 
 app = Flask(__name__)
 
-# В памяти храним зарегистрированных пользователей и сессии передачи файлов
-users = {}
 transfers = {}
+transfer_data = {}
+transfer_lock = threading.Lock()
 
 def generate_short_id(length=8):
-    """
-    Генерирует короткий уникальный идентификатор для передачи файла.
-    """
     while True:
         token = ''.join(random.choices(string.ascii_letters + string.digits, k=length))
         if token not in transfers:
             return token
 
-@app.route('/register', methods=['POST'])
-def register():
-    data = request.json
-    username = data.get('username')
-    password = data.get('password')
-    if not username or not password:
-        return jsonify({'error': 'Не переданы обязательные поля'}), 400
-    if username in users:
-        return jsonify({'error': 'Пользователь уже существует'}), 400
-    users[username] = {'password': password}
-    return jsonify({'message': 'Регистрация прошла успешно'})
-
 @app.route('/create_transfer', methods=['POST'])
 def create_transfer():
-    data = request.json
-    filename = data.get('filename')
-    ip = data.get('ip')
-    port = data.get('port')
-    filesize = data.get('filesize', 0)
-    if not filename or not ip or not port:
-        return jsonify({'error': 'Отсутствуют необходимые параметры'}), 400
-
-    # Используем короткий идентификатор для передачи файла
+    filename = request.json.get('filename')
+    filesize = request.json.get('filesize', 0)
+    
     transfer_id = generate_short_id(8)
-    transfers[transfer_id] = {
-        'filename': filename,
-        'ip': ip,
-        'port': port,
-        'filesize': filesize,
-        'downloaded': False
-    }
-    # Формируем короткую ссылку для скачивания (маршрут изменён на /d/)
-    download_link = request.host_url + 'd/' + transfer_id
-    return jsonify({'download_link': download_link})
+    with transfer_lock:
+        transfers[transfer_id] = {
+            'filename': filename,
+            'filesize': filesize,
+            'completed': False
+        }
+        transfer_data[transfer_id] = deque(maxlen=100)  # Ограничение буфера
+    return jsonify({'transfer_id': transfer_id})
+
+@app.route('/upload/<transfer_id>', methods=['POST'])
+def upload_file(transfer_id):
+    if transfer_id not in transfers:
+        return jsonify({'error': 'Invalid transfer ID'}), 404
+    
+    data = request.data
+    with transfer_lock:
+        transfer_data[transfer_id].extend(data)
+        transfers[transfer_id]['completed'] = request.headers.get('X-Transfer-Complete') == 'true'
+    
+    return jsonify({'status': 'chunk uploaded'})
+
+@app.route('/download/<transfer_id>', methods=['GET'])
+def download_file(transfer_id):
+    if transfer_id not in transfers:
+        return jsonify({'error': 'Invalid transfer ID'}), 404
+
+    def generate():
+        while True:
+            with transfer_lock:
+                if transfer_data[transfer_id]:
+                    chunk = bytes(transfer_data[transfer_id].popleft())
+                    yield chunk
+                elif transfers[transfer_id]['completed']:
+                    break
 
-@app.route('/d/<transfer_id>', methods=['GET'])
-def get_transfer(transfer_id):
-    transfer = transfers.get(transfer_id)
-    if not transfer:
-        return jsonify({'error': 'Неверный идентификатор передачи'}), 404
-    return jsonify(transfer)
+    return Response(
+        generate(),
+        mimetype='application/octet-stream',
+        headers={'Content-Disposition': f'attachment; filename="{transfers[transfer_id]["filename"]}"'}
+    )
 
 if __name__ == '__main__':
-    # Запускаем сервер на всех интерфейсах на порту 5000
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)