Starchik1 commited on
Commit
3d7b0be
·
verified ·
1 Parent(s): ee9d0ca

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +18 -5
main.py CHANGED
@@ -4,11 +4,10 @@ import string
4
 
5
  app = Flask(__name__)
6
 
7
- # В памяти храним зарегистрированные передачи файлов
 
8
  transfers = {}
9
 
10
- BASE_URL = "https://starchik1-share.hf.space/d/" # Постоянный серверный URL
11
-
12
  def generate_short_id(length=8):
13
  """
14
  Генерирует короткий уникальный идентификатор для передачи файла.
@@ -18,6 +17,18 @@ def generate_short_id(length=8):
18
  if token not in transfers:
19
  return token
20
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  @app.route('/create_transfer', methods=['POST'])
22
  def create_transfer():
23
  data = request.json
@@ -37,8 +48,9 @@ def create_transfer():
37
  'filesize': filesize,
38
  'downloaded': False
39
  }
40
-
41
- return jsonify({'token': transfer_id})
 
42
 
43
  @app.route('/d/<transfer_id>', methods=['GET'])
44
  def get_transfer(transfer_id):
@@ -48,4 +60,5 @@ def get_transfer(transfer_id):
48
  return jsonify(transfer)
49
 
50
  if __name__ == '__main__':
 
51
  app.run(host='0.0.0.0', port=7860)
 
4
 
5
  app = Flask(__name__)
6
 
7
+ # В памяти храним зарегистрированных пользователей и сессии передачи файлов
8
+ users = {}
9
  transfers = {}
10
 
 
 
11
  def generate_short_id(length=8):
12
  """
13
  Генерирует короткий уникальный идентификатор для передачи файла.
 
17
  if token not in transfers:
18
  return token
19
 
20
+ @app.route('/register', methods=['POST'])
21
+ def register():
22
+ data = request.json
23
+ username = data.get('username')
24
+ password = data.get('password')
25
+ if not username or not password:
26
+ return jsonify({'error': 'Не переданы обязательные поля'}), 400
27
+ if username in users:
28
+ return jsonify({'error': 'Пользователь уже существует'}), 400
29
+ users[username] = {'password': password}
30
+ return jsonify({'message': 'Регистрация прошла успешно'})
31
+
32
  @app.route('/create_transfer', methods=['POST'])
33
  def create_transfer():
34
  data = request.json
 
48
  'filesize': filesize,
49
  'downloaded': False
50
  }
51
+ # Формируем короткую ссылку для скачивания (маршрут изменён на /d/)
52
+ download_link = request.host_url + 'd/' + transfer_id
53
+ return jsonify({'download_link': download_link})
54
 
55
  @app.route('/d/<transfer_id>', methods=['GET'])
56
  def get_transfer(transfer_id):
 
60
  return jsonify(transfer)
61
 
62
  if __name__ == '__main__':
63
+ # Запускаем сервер на всех интерфейсах на порту 5000
64
  app.run(host='0.0.0.0', port=7860)