3445b5d800148a4a9f252ecbdf4cd54b2fbdf3c8,apps/domain/src/main/core/services/infra_service.py,,del_worker_msg,#Any#Any#Any#,203

Before Change

    // Owner / Admin
    if users.can_manage_infrastructure(user_id=_current_user_id):
        node.environments.delete_associations(environment_id=_worker_id)
        node.environments.delete(id=_worker_id)
    else:  // Env Owner
        envs = [
            int(env.id)
            for env in node.environments.get_environments(user=_current_user_id)

After Change

def del_worker_msg(
    msg: DeleteWorkerMessage, node: AbstractNode, verify_key: VerifyKey
) -> DeleteWorkerResponse:
    try:
        // Get Payload Content
        worker_id = msg.content.get("worker_id", None)
        _current_user_id = msg.content.get("current_user", None)

        users = node.users
        if not _current_user_id:
            _current_user_id = users.first(
                verify_key=verify_key.encode(encoder=HexEncoder).decode("utf-8")
            ).id

        is_admin = users.can_manage_infrastructure(user_id=_current_user_id)

        envs = [
            int(env.id)
            for env in node.environments.get_environments(user=_current_user_id)
        ]
        created_by_current_user = int(worker_id) in envs

        // Owner / Admin
        if not is_admin and not created_by_current_user:
            raise AuthorizationError("You"re not allowed to delete this worker!")

        env = node.environments.first(id=worker_id)
        _config = Config(provider=env.provider, app=Config(name="worker", id=worker_id))

        success = Provider(_config).destroy()
        if success:
            node.environments.set(
                id=worker_id, state=states["destroyed"], destroyed_at=datetime.now()
            )
        else:
            raise Exception("Worker deletion failed")

        return DeleteWorkerResponse(
            address=msg.reply_to,
            status_code=200,
            content={"msg": "Worker was deleted successfully!"},
        )
    except Exception as e:
        return DeleteWorkerResponse(
            address=msg.reply_to, status_code=False, content={"error": str(e)}
        )


// def update_worker_msg(
//     msg: UpdateWorkerMessage,
//     node: AbstractNode,
//     verify_key: VerifyKey,
// ) -> UpdateWorkerResponse:
//     // Get Payload Content
//     _worker_id = msg.content.get("worker_id", None)
//     _current_user_id = msg.content.get("current_user", None)
//
//     env_parameters = {
//         i: msg.content[i]
//         for i in msg.content.keys()
//         if i in list(Environment.__table__.columns.keys())
//     }
//
//     users = node.users
//     if not _current_user_id:
//         _current_user_id = users.first(
//             verify_key=verify_key.encode(encoder=HexEncoder).decode("utf-8")
//         ).id
//
//     _current_user = users.first(id=_current_user_id)
//
//     // Owner / Admin
//     if users.can_manage_infrastructure(user_id=_current_user_id):
//         node.environments.modify({"id": _worker_id}, env_parameters)
//     else:  // Env Owner
//         envs = [
//             int(env.id)
//             for env in node.environments.get_environments(user=_current_user_id)
//         ]
//         if int(_worker_id) in envs:
//             node.environments.modify({"id": _worker_id}, env_parameters)
//         else:
//             raise AuthorizationError(
//                 "You"re not allowed to update this environment information!"
//             )
//     return UpdateWorkerResponse(
//         address=msg.reply_to,
//         status_code=200,
//         content={"msg": "Worker was updated succesfully!"},
//     )


class DomainInfrastructureService(ImmediateNodeServiceWithReply):

    msg_handler_map = {
        CreateWorkerMessage: create_worker_msg,

Instances