yeq6x commited on
Commit
dd8db9d
·
1 Parent(s): 6a66010
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from flask import Flask, request, render_template, send_file, jsonify, send_from_directory
2
  from flask_socketio import SocketIO, join_room, leave_room, close_room, rooms, disconnect
3
  from flask_cors import CORS
4
  from flask_limiter import Limiter
@@ -59,7 +59,7 @@ def update_queue_status(message):
59
 
60
  def process_task(task):
61
  try:
62
- client_id = request.sid
63
  task.is_processing = True
64
  # ファイルデータをPIL Imageに変換
65
  image = Image.open(io.BytesIO(task.file_data))
@@ -118,6 +118,7 @@ connected_clients = 0
118
  tasks_per_client = {}
119
  @socketio.on('connect', namespace='/')
120
  def handle_connect(auth):
 
121
  client_id = request.sid # クライアントIDを取得
122
  join_room(client_id) # クライアントを自身のルームに入れる
123
  global connected_clients
@@ -148,7 +149,7 @@ def submit_task():
148
 
149
  # クライアントIPアドレスを取得
150
  client_ip = get_remote_address()
151
- client_id = request.sid # クライアントIDを取得
152
 
153
  # 同一IPからの同時タスク数を制限
154
  if tasks_per_client.get(client_ip, 0) >= 2:
 
1
+ from flask import Flask, request, render_template, send_file, jsonify, send_from_directory, session
2
  from flask_socketio import SocketIO, join_room, leave_room, close_room, rooms, disconnect
3
  from flask_cors import CORS
4
  from flask_limiter import Limiter
 
59
 
60
  def process_task(task):
61
  try:
62
+ client_id = session.get("sid")
63
  task.is_processing = True
64
  # ファイルデータをPIL Imageに変換
65
  image = Image.open(io.BytesIO(task.file_data))
 
118
  tasks_per_client = {}
119
  @socketio.on('connect', namespace='/')
120
  def handle_connect(auth):
121
+ session["sid"] = request.sid
122
  client_id = request.sid # クライアントIDを取得
123
  join_room(client_id) # クライアントを自身のルームに入れる
124
  global connected_clients
 
149
 
150
  # クライアントIPアドレスを取得
151
  client_ip = get_remote_address()
152
+ client_id = session.get("sid")
153
 
154
  # 同一IPからの同時タスク数を制限
155
  if tasks_per_client.get(client_ip, 0) >= 2: