Update main.py

main.py

@@ -1,5 +1,6 @@
 from flask import Flask, request, jsonify
-import uuid
+import random
+import string
 
 app = Flask(__name__)
 
@@ -7,6 +8,15 @@ app = Flask(__name__)
 users = {}
 transfers = {}
 
+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
@@ -29,7 +39,8 @@ def create_transfer():
     if not filename or not ip or not port:
         return jsonify({'error': 'Отсутствуют необходимые параметры'}), 400
 
-    transfer_id = str(uuid.uuid4())
+    # Используем короткий идентификатор для передачи файла
+    transfer_id = generate_short_id(8)
     transfers[transfer_id] = {
         'filename': filename,
         'ip': ip,
@@ -37,11 +48,11 @@ def create_transfer():
         'filesize': filesize,
         'downloaded': False
     }
-    # Формируем ссылку для скачивания. request.host_url учитывает адрес сервера.
-    download_link = request.host_url + 'download/' + transfer_id
+    # Формируем короткую ссылку для скачивания (маршрут изменён на /d/)
+    download_link = request.host_url + 'd/' + transfer_id
     return jsonify({'download_link': download_link})
 
-@app.route('/download/<transfer_id>', methods=['GET'])
+@app.route('/d/<transfer_id>', methods=['GET'])
 def get_transfer(transfer_id):
     transfer = transfers.get(transfer_id)
     if not transfer:
@@ -50,4 +61,4 @@ def get_transfer(transfer_id):
 
 if __name__ == '__main__':
     # Запускаем сервер на всех интерфейсах на порту 5000
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=5000)