Skip to content

Commit

Permalink
Merge 3a3017e into a149ed8
Browse files Browse the repository at this point in the history
  • Loading branch information
yjmrobert authored Oct 3, 2023
2 parents a149ed8 + 3a3017e commit f81f2d6
Showing 1 changed file with 57 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -65,4 +65,60 @@ def sync_workspace_users_function(workspace_definition):

# Cleanup users in workspace that aren't in AAD Graph
remove_deleted_users_in_workspace(workspace_client)
synchronize_workspace_users(workspace_definition, workspace_client)
synchronize_workspace_users(workspace_definition, workspace_client)





####################################################################################
# Temporary function to run the sync function in INT and POC environments
####################################################################################



@app.function_name(name="SynchronizeWorkspaceUsersQueueTrigger")
@app.queue_trigger(arg_name="msg", queue_name="user-run-request",
connection="TempIntConnectionString") # Queue Trigger

def queue_sync_workspace_users_function(msg: func.QueueMessage) -> None:
"""
Synchronizes the users in the Databricks workspace with the users in the definition file.
Args:
workspace_definition (QueueMessage): The workspace definition file.
Returns:
None
"""
workspace_definition = msg.get_json()
logging.info("Synchronizing workspace users.")

sync_workspace_users_function(workspace_definition)

logging.info("Successfully synchronized workspace users.")
return None

@app.function_name(name="SynchronizeWorkspaceUsersQueueTrigger")
@app.queue_trigger(arg_name="msg", queue_name="user-run-request",
connection="TempPocConnectionString") # Queue Trigger

def queue_sync_workspace_users_function(msg: func.QueueMessage) -> None:
"""
Synchronizes the users in the Databricks workspace with the users in the definition file.
Args:
workspace_definition (QueueMessage): The workspace definition file.
Returns:
None
"""
workspace_definition = msg.get_json()
logging.info("Synchronizing workspace users.")

sync_workspace_users_function(workspace_definition)

logging.info("Successfully synchronized workspace users.")
return None

0 comments on commit f81f2d6

Please sign in to comment.