ruff format

This commit is contained in:
hjlarry 2025-07-21 16:13:04 +08:00
parent ee86ceb906
commit 897c842637
5 changed files with 15 additions and 28 deletions

View File

@ -40,9 +40,7 @@ else:
if __name__ == "__main__":
from gevent import pywsgi
host = os.environ.get('HOST', '0.0.0.0')
port = int(os.environ.get('PORT', 5001))
print(f"Starting server on {host}:{port}")
host = os.environ.get("HOST", "0.0.0.0")
port = int(os.environ.get("PORT", 5001))
server = pywsgi.WSGIServer((host, port), socketio_app)
server.serve_forever()

View File

@ -34,6 +34,7 @@ def create_app() -> tuple[any, DifyApp]:
import socketio
from extensions.ext_socketio import sio
sio.app = app
socketio_app = socketio.WSGIApp(sio, app)

View File

@ -7,14 +7,14 @@ from libs.passport import PassportService
from services.account_service import AccountService
@sio.on('connect')
@sio.on("connect")
def socket_connect(sid, environ, auth):
"""
WebSocket connect event, do authentication here.
"""
token = None
if auth and isinstance(auth, dict):
token = auth.get('token')
token = auth.get("token")
if not token:
return False
@ -29,11 +29,7 @@ def socket_connect(sid, environ, auth):
if not user:
return False
sio.save_session(sid, {
'user_id': user.id,
'username': user.name,
'avatar': user.avatar
})
sio.save_session(sid, {"user_id": user.id, "username": user.name, "avatar": user.avatar})
return True
@ -52,7 +48,7 @@ def handle_user_connect(sid, data):
return {"msg": "workflow_id is required"}, 400
session = sio.get_session(sid)
user_id = session.get('user_id')
user_id = session.get("user_id")
if not user_id:
return {"msg": "unauthorized"}, 401
@ -66,8 +62,8 @@ def handle_user_connect(sid, data):
user_info = {
"user_id": user_id,
"username": session.get('username', 'Unknown'),
"avatar": session.get('avatar', None),
"username": session.get("username", "Unknown"),
"avatar": session.get("avatar", None),
"sid": sid,
}
@ -107,11 +103,7 @@ def broadcast_online_users(workflow_id):
users.append(json.loads(user_info_json))
except Exception:
continue
sio.emit(
"online_users",
{"workflow_id": workflow_id, "users": users},
room=workflow_id
)
sio.emit("online_users", {"workflow_id": workflow_id, "users": users}, room=workflow_id)
@sio.on("collaboration_event")
@ -140,14 +132,9 @@ def handle_collaboration_event(sid, data):
sio.emit(
"collaboration_update",
{
"type": event_type,
"userId": user_id,
"data": event_data,
"timestamp": timestamp
},
{"type": event_type, "userId": user_id, "data": event_data, "timestamp": timestamp},
room=workflow_id,
skip_sid=sid
skip_sid=sid,
)
return {"msg": "event_broadcasted"}

View File

@ -789,6 +789,7 @@ class DraftWorkflowNodeLastRunApi(Resource):
raise NotFound("last run not found")
return node_exec
class WorkflowOnlineUsersApi(Resource):
@setup_required
@login_required

View File

@ -9,9 +9,9 @@ online_user_partial_fields = {
workflow_online_users_fields = {
"workflow_id": fields.String,
"users": fields.List(fields.Nested(online_user_partial_fields))
"users": fields.List(fields.Nested(online_user_partial_fields)),
}
online_user_list_fields = {
"data": fields.List(fields.Nested(workflow_online_users_fields)),
}
}