mirror of
				https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
				synced 2025-11-04 04:12:45 +00:00 
			
		
		
		
	Reorganize Socket.IO code
This commit is contained in:
		@@ -132,9 +132,6 @@ def create_app(config: Config = Config) -> Flask:
 | 
				
			|||||||
    # region SocketIO Namespaces
 | 
					    # region SocketIO Namespaces
 | 
				
			||||||
    from .namespaces.cqi_over_sio import CQiOverSocketIONamespace
 | 
					    from .namespaces.cqi_over_sio import CQiOverSocketIONamespace
 | 
				
			||||||
    socketio.on_namespace(CQiOverSocketIONamespace('/cqi_over_sio'))
 | 
					    socketio.on_namespace(CQiOverSocketIONamespace('/cqi_over_sio'))
 | 
				
			||||||
 | 
					 | 
				
			||||||
    from .namespaces.users import UsersNamespace
 | 
					 | 
				
			||||||
    socketio.on_namespace(UsersNamespace('/users'))
 | 
					 | 
				
			||||||
    # endregion SocketIO Namespaces
 | 
					    # endregion SocketIO Namespaces
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    # region Database event Listeners
 | 
					    # region Database event Listeners
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -15,4 +15,4 @@ def before_request():
 | 
				
			|||||||
    pass
 | 
					    pass
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from . import cli, json_routes, routes, settings
 | 
					from . import cli, events, json_routes, routes, settings
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										82
									
								
								app/blueprints/users/events.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								app/blueprints/users/events.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,82 @@
 | 
				
			|||||||
 | 
					from flask_login import current_user
 | 
				
			||||||
 | 
					from flask_socketio import join_room, leave_room
 | 
				
			||||||
 | 
					from app import hashids, socketio
 | 
				
			||||||
 | 
					from app.decorators import socketio_login_required
 | 
				
			||||||
 | 
					from app.models import User
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@socketio.on('users.get_user')
 | 
				
			||||||
 | 
					@socketio_login_required
 | 
				
			||||||
 | 
					def get_user(user_hashid: str) -> dict:
 | 
				
			||||||
 | 
					    user_id = hashids.decode(user_hashid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not isinstance(user_id, int):
 | 
				
			||||||
 | 
					        return {'status': 400, 'statusText': 'Bad Request'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    user = User.query.get(user_id)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if user is None:
 | 
				
			||||||
 | 
					        return {'status': 404, 'statusText': 'Not found'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not (
 | 
				
			||||||
 | 
					        user == current_user
 | 
				
			||||||
 | 
					        or current_user.is_administrator
 | 
				
			||||||
 | 
					    ):
 | 
				
			||||||
 | 
					        return {'status': 403, 'statusText': 'Forbidden'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return {
 | 
				
			||||||
 | 
					        'body': user.to_json_serializeable(
 | 
				
			||||||
 | 
					            backrefs=True,
 | 
				
			||||||
 | 
					            relationships=True
 | 
				
			||||||
 | 
					        ),
 | 
				
			||||||
 | 
					        'status': 200,
 | 
				
			||||||
 | 
					        'statusText': 'OK'
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@socketio.on('users.subscribe_user')
 | 
				
			||||||
 | 
					@socketio_login_required
 | 
				
			||||||
 | 
					def subscribe_user(user_hashid: str) -> dict:
 | 
				
			||||||
 | 
					    user_id = hashids.decode(user_hashid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not isinstance(user_id, int):
 | 
				
			||||||
 | 
					        return {'status': 400, 'statusText': 'Bad Request'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    user = User.query.get(user_id)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if user is None:
 | 
				
			||||||
 | 
					        return {'status': 404, 'statusText': 'Not found'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not (
 | 
				
			||||||
 | 
					        user == current_user
 | 
				
			||||||
 | 
					        or current_user.is_administrator
 | 
				
			||||||
 | 
					    ):
 | 
				
			||||||
 | 
					        return {'status': 403, 'statusText': 'Forbidden'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    join_room(f'/users/{user.hashid}')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return {'status': 200, 'statusText': 'OK'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@socketio.on('users.unsubscribe_user')
 | 
				
			||||||
 | 
					@socketio_login_required
 | 
				
			||||||
 | 
					def on_unsubscribe_user(user_hashid: str) -> dict:
 | 
				
			||||||
 | 
					    user_id = hashids.decode(user_hashid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not isinstance(user_id, int):
 | 
				
			||||||
 | 
					        return {'status': 400, 'statusText': 'Bad Request'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    user = User.query.get(user_id)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if user is None:
 | 
				
			||||||
 | 
					        return {'status': 404, 'statusText': 'Not found'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if not (
 | 
				
			||||||
 | 
					        user == current_user
 | 
				
			||||||
 | 
					        or current_user.is_administrator
 | 
				
			||||||
 | 
					    ):
 | 
				
			||||||
 | 
					        return {'status': 403, 'statusText': 'Forbidden'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    leave_room(f'/users/{user.hashid}')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return {'status': 200, 'statusText': 'OK'}
 | 
				
			||||||
@@ -1,78 +0,0 @@
 | 
				
			|||||||
from flask_login import current_user
 | 
					 | 
				
			||||||
from flask_socketio import join_room, leave_room, Namespace
 | 
					 | 
				
			||||||
from app import hashids
 | 
					 | 
				
			||||||
from app.decorators import socketio_login_required
 | 
					 | 
				
			||||||
from app.models import User
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class UsersNamespace(Namespace):
 | 
					 | 
				
			||||||
    @socketio_login_required
 | 
					 | 
				
			||||||
    def on_get_user(self, user_hashid: str) -> dict:
 | 
					 | 
				
			||||||
        user_id = hashids.decode(user_hashid)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not isinstance(user_id, int):
 | 
					 | 
				
			||||||
            return {'status': 400, 'statusText': 'Bad Request'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        user = User.query.get(user_id)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if user is None:
 | 
					 | 
				
			||||||
            return {'status': 404, 'statusText': 'Not found'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not (
 | 
					 | 
				
			||||||
            user == current_user
 | 
					 | 
				
			||||||
            or current_user.is_administrator
 | 
					 | 
				
			||||||
        ):
 | 
					 | 
				
			||||||
            return {'status': 403, 'statusText': 'Forbidden'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {
 | 
					 | 
				
			||||||
            'body': user.to_json_serializeable(
 | 
					 | 
				
			||||||
                backrefs=True,
 | 
					 | 
				
			||||||
                relationships=True
 | 
					 | 
				
			||||||
            ),
 | 
					 | 
				
			||||||
            'status': 200,
 | 
					 | 
				
			||||||
            'statusText': 'OK'
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    @socketio_login_required
 | 
					 | 
				
			||||||
    def on_subscribe_user(self, user_hashid: str) -> dict:
 | 
					 | 
				
			||||||
        user_id = hashids.decode(user_hashid)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not isinstance(user_id, int):
 | 
					 | 
				
			||||||
            return {'status': 400, 'statusText': 'Bad Request'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        user = User.query.get(user_id)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if user is None:
 | 
					 | 
				
			||||||
            return {'status': 404, 'statusText': 'Not found'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not (
 | 
					 | 
				
			||||||
            user == current_user
 | 
					 | 
				
			||||||
            or current_user.is_administrator
 | 
					 | 
				
			||||||
        ):
 | 
					 | 
				
			||||||
            return {'status': 403, 'statusText': 'Forbidden'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        join_room(f'/users/{user.hashid}')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {'status': 200, 'statusText': 'OK'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    @socketio_login_required
 | 
					 | 
				
			||||||
    def on_unsubscribe_user(self, user_hashid: str) -> dict:
 | 
					 | 
				
			||||||
        user_id = hashids.decode(user_hashid)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not isinstance(user_id, int):
 | 
					 | 
				
			||||||
            return {'status': 400, 'statusText': 'Bad Request'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        user = User.query.get(user_id)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if user is None:
 | 
					 | 
				
			||||||
            return {'status': 404, 'statusText': 'Not found'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not (
 | 
					 | 
				
			||||||
            user == current_user
 | 
					 | 
				
			||||||
            or current_user.is_administrator
 | 
					 | 
				
			||||||
        ):
 | 
					 | 
				
			||||||
            return {'status': 403, 'statusText': 'Forbidden'}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        leave_room(f'/users/{user.hashid}')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {'status': 200, 'statusText': 'OK'}
 | 
					 | 
				
			||||||
@@ -11,11 +11,9 @@ nopaque.App = class App {
 | 
				
			|||||||
      subscribeUser: {}
 | 
					      subscribeUser: {}
 | 
				
			||||||
    };
 | 
					    };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    this.sockets = {
 | 
					    this.socket = io({transports: ['websocket'], upgrade: false});
 | 
				
			||||||
      users: io('/users', {transports: ['websocket'], upgrade: false})
 | 
					 | 
				
			||||||
    };
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    this.sockets.users.on('patch_user', (patch) => {this.onPatch(patch);});
 | 
					    this.socket.on('patch_user', (patch) => {this.onPatch(patch);});
 | 
				
			||||||
  }
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
  getUser(userId) {
 | 
					  getUser(userId) {
 | 
				
			||||||
@@ -23,10 +21,8 @@ nopaque.App = class App {
 | 
				
			|||||||
      return this.#promises.getUser[userId];
 | 
					      return this.#promises.getUser[userId];
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let socket = this.sockets.users;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    this.#promises.getUser[userId] = new Promise((resolve, reject) => {
 | 
					    this.#promises.getUser[userId] = new Promise((resolve, reject) => {
 | 
				
			||||||
      socket.emit('get_user', userId, (response) => {
 | 
					      this.socket.emit('users.get_user', userId, (response) => {
 | 
				
			||||||
        if (response.status === 200) {
 | 
					        if (response.status === 200) {
 | 
				
			||||||
          this.data.users[userId] = response.body;
 | 
					          this.data.users[userId] = response.body;
 | 
				
			||||||
          resolve(this.data.users[userId]);
 | 
					          resolve(this.data.users[userId]);
 | 
				
			||||||
@@ -44,10 +40,8 @@ nopaque.App = class App {
 | 
				
			|||||||
      return this.#promises.subscribeUser[userId];
 | 
					      return this.#promises.subscribeUser[userId];
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let socket = this.sockets.users;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    this.#promises.subscribeUser[userId] = new Promise((resolve, reject) => {
 | 
					    this.#promises.subscribeUser[userId] = new Promise((resolve, reject) => {
 | 
				
			||||||
      socket.emit('subscribe_user', userId, (response) => {
 | 
					      this.socket.emit('users.subscribe_user', userId, (response) => {
 | 
				
			||||||
        if (response.status === 200) {
 | 
					        if (response.status === 200) {
 | 
				
			||||||
          resolve(response);
 | 
					          resolve(response);
 | 
				
			||||||
        } else {
 | 
					        } else {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -8,7 +8,7 @@ nopaque.resource_displays.ResourceDisplay = class ResourceDisplay {
 | 
				
			|||||||
    if (this.userId) {
 | 
					    if (this.userId) {
 | 
				
			||||||
      app.subscribeUser(this.userId)
 | 
					      app.subscribeUser(this.userId)
 | 
				
			||||||
        .then((response) => {
 | 
					        .then((response) => {
 | 
				
			||||||
          app.sockets.users.on('patch_user', (patch) => {
 | 
					          app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
            if (this.isInitialized) {this.onPatch(patch);}
 | 
					            if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
          });
 | 
					          });
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -15,7 +15,7 @@ nopaque.resource_lists.CorpusFileList = class CorpusFileList extends nopaque.res
 | 
				
			|||||||
    this.hasPermissionManageFiles =  listContainerElement.dataset?.hasPermissionManageFiles == 'true' || false;
 | 
					    this.hasPermissionManageFiles =  listContainerElement.dataset?.hasPermissionManageFiles == 'true' || false;
 | 
				
			||||||
    if (this.userId === undefined || this.corpusId === undefined) {return;}
 | 
					    if (this.userId === undefined || this.corpusId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -13,7 +13,7 @@ nopaque.resource_lists.CorpusFollowerList = class CorpusFollowerList extends nop
 | 
				
			|||||||
    this.corpusId = listContainerElement.dataset.corpusId;
 | 
					    this.corpusId = listContainerElement.dataset.corpusId;
 | 
				
			||||||
    if (this.userId === undefined || this.corpusId === undefined) {return;}
 | 
					    if (this.userId === undefined || this.corpusId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -12,7 +12,7 @@ nopaque.resource_lists.CorpusList = class CorpusList extends nopaque.resource_li
 | 
				
			|||||||
    this.userId = listContainerElement.dataset.userId;
 | 
					    this.userId = listContainerElement.dataset.userId;
 | 
				
			||||||
    if (this.userId === undefined) {return;}
 | 
					    if (this.userId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -13,7 +13,7 @@ nopaque.resource_lists.JobList = class JobList extends nopaque.resource_lists.Re
 | 
				
			|||||||
    this.userId = listContainerElement.dataset.userId;
 | 
					    this.userId = listContainerElement.dataset.userId;
 | 
				
			||||||
    if (this.userId === undefined) {return;}
 | 
					    if (this.userId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@ nopaque.resource_lists.JobResultList = class JobResultList extends nopaque.resou
 | 
				
			|||||||
    this.jobId = listContainerElement.dataset.jobId;
 | 
					    this.jobId = listContainerElement.dataset.jobId;
 | 
				
			||||||
    if (this.userId === undefined || this.jobId === undefined) {return;}
 | 
					    if (this.userId === undefined || this.jobId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@ nopaque.resource_lists.SpaCyNLPPipelineModelList = class SpaCyNLPPipelineModelLi
 | 
				
			|||||||
    this.userId = listContainerElement.dataset.userId;
 | 
					    this.userId = listContainerElement.dataset.userId;
 | 
				
			||||||
    if (this.userId === undefined) {return;}
 | 
					    if (this.userId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@ nopaque.resource_lists.TesseractOCRPipelineModelList = class TesseractOCRPipelin
 | 
				
			|||||||
    this.userId = listContainerElement.dataset.userId;
 | 
					    this.userId = listContainerElement.dataset.userId;
 | 
				
			||||||
    if (this.userId === undefined) {return;}
 | 
					    if (this.userId === undefined) {return;}
 | 
				
			||||||
    app.subscribeUser(this.userId).then((response) => {
 | 
					    app.subscribeUser(this.userId).then((response) => {
 | 
				
			||||||
      app.sockets.users.on('patch_user', (patch) => {
 | 
					      app.socket.on('patch_user', (patch) => {
 | 
				
			||||||
        if (this.isInitialized) {this.onPatch(patch);}
 | 
					        if (this.isInitialized) {this.onPatch(patch);}
 | 
				
			||||||
      });
 | 
					      });
 | 
				
			||||||
    });
 | 
					    });
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user