Update main.py

main.py

@@ -1,76 +1,52 @@
-from flask import Flask, request, jsonify, Response
-from flask_executor import Executor
-import uuid
-import time
-import logging
-from collections import defaultdict, deque
-from threading import Lock
+from flask import Flask, request, jsonify
+import random
+import string
+import threading
 
 app = Flask(__name__)
-executor = Executor(app)
+connections = {}
+lock = threading.Lock()
 
-# Хранилище для передачи чанков
-transfers = defaultdict(deque)
-locks = defaultdict(Lock)
-metadata = {}
-CHUNK_TIMEOUT = 3600  # 5 минут
+def generate_id():
+    return ''.join(random.choices(string.ascii_letters + string.digits, k=8))
 
-@app.route('/create_transfer', methods=['POST'])
-def create_transfer():
-    transfer_id = str(uuid.uuid4())
-    metadata[transfer_id] = {
-        'filename': request.json.get('filename', 'file'),
-        'created_at': time.time(),
-        'completed': False
-    }
-    return jsonify({'transfer_id': transfer_id})
-
-@app.route('/upload/<transfer_id>', methods=['POST'])
-def upload_chunk(transfer_id):
-    if transfer_id not in metadata:
-        return jsonify({'error': 'Invalid transfer ID'}), 404
+@app.route('/create_offer', methods=['POST'])
+def create_offer():
+    offer = request.json
+    conn_id = generate_id()
     
-    with locks[transfer_id]:
-        chunk = request.data
-        if chunk:
-            transfers[transfer_id].append(chunk)
-        else:
-            metadata[transfer_id]['completed'] = True
+    with lock:
+        connections[conn_id] = {
+            'offer': offer,
+            'answer': None
+        }
     
-    return jsonify({'status': 'ok'})
+    return jsonify({'connection_id': conn_id})
 
-@app.route('/stream/<transfer_id>')
-def stream_chunks(transfer_id):
-    def generate():
-        last_activity = time.time()
-        
-        while True:
-            with locks[transfer_id]:
-                if transfers[transfer_id]:
-                    chunk = transfers[transfer_id].popleft()
-                    last_activity = time.time()
-                    yield chunk
-                elif metadata.get(transfer_id, {}).get('completed', False):
-                    break
-                elif time.time() - last_activity > CHUNK_TIMEOUT:
-                    break
-            
-            time.sleep(0.1)  # Задержка для уменьшения нагрузки
+@app.route('/get_offer/<conn_id>', methods=['GET'])
+def get_offer(conn_id):
+    with lock:
+        conn = connections.get(conn_id)
+        if not conn:
+            return jsonify({'error': 'Invalid ID'}), 404
+        return jsonify(conn['offer'])
+
+@app.route('/send_answer/<conn_id>', methods=['POST'])
+def send_answer(conn_id):
+    with lock:
+        if conn_id not in connections:
+            return jsonify({'error': 'Invalid ID'}), 404
         
-        # Очистка данных
-        if transfer_id in metadata:
-            del metadata[transfer_id]
-            del transfers[transfer_id]
-            del locks[transfer_id]
+        connections[conn_id]['answer'] = request.json
+        return jsonify({'status': 'success'})
 
-    return Response(
-        generate(),
-        mimetype='application/octet-stream',
-        headers={
-            'Content-Disposition': f'attachment; filename="{metadata[transfer_id]["filename"]}"',
-            'Transfer-Encoding': 'chunked'
-        }
-    )
+@app.route('/get_answer/<conn_id>', methods=['GET'])
+def get_answer(conn_id):
+    with lock:
+        conn = connections.get(conn_id)
+        if not conn or not conn['answer']:
+            return jsonify({'error': 'Not ready'}), 404
+        return jsonify(conn['answer'])
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(host='0.0.0.0', port=5000, threaded=True)