def with_assumed_user_id(process: ProcessQuery) -> ProcessQuery:
# The ssh_process must have an associated user id
(
UserIdAssumptionQuery()
.with_assuming_process(process)
.with_user_id()
)
return process
def get_user_id(process: ProcessView) -> Optional[int]:
user_assumption = (UserIdAssumptionQuery().with_assuming_process(
ProcessQuery().with_node_key(
process.node_key)).with_user_id().query_first(
process.dgraph_client)
) # type: Optional[UserIdAssumptionView]
if user_assumption:
return user_assumption.get_used_id()
return None
def with_assumed_user_id(process: ProcessQuery) -> ProcessQuery:
(UserIdAssumptionQuery().with_assuming_process(process).with_user_id())
return process