Opportunistically migrate Workspace to new Bundle

Hey, here's a high-level approach to achieving this automation, along with some pseudocode to give you an idea of how it could be implemented.

Goals

1. Monitoring Workspace Status

Here you need a monitoring service or script that continuously checks the status of each Workspace. This service should be able to determine:

• If the user is an hourly or always-on user.
• And the last active time of the user (to calculate downtime for always-on users).

2. Determining Update Opportunity

Based on the Workspace status, the service decides when it's an opportune moment to update the Workspace bundle. The logic can be something like:

• For hourly users: Trigger the update process as soon as the machine is turned off.
• For always-on users: Trigger the update process when the machine has been disconnected or inactive for 12 hours.

3.Performing the Update

Once an update opportunity is identified, the service should execute the update process. This could involve running update scripts, applying new configurations, and restarting the Workspace with the new bundle.

Implementation

def check_workspace_status():
    # Example function to check the status of all Workspaces
    workspaces = get_all_workspaces()
    for workspace in workspaces:
        user_type = workspace.user_type  # hourly or always-on
        last_active = workspace.last_active_time
        current_time = get_current_time()
        
        if user_type == "hourly" and not workspace.is_active:
            perform_update(workspace)
        elif user_type == "always-on" and (current_time - last_active) >= 12 * HOUR:
            perform_update(workspace)

def perform_update(workspace):
    # Example function to perform the update
    update_bundle(workspace)
    restart_workspace(workspace)

def update_bundle(workspace):
    # Implement the bundle update logic here
    pass

def restart_workspace(workspace):
    # Implement the Workspace restart logic here
    pass

def main():
    while True:
        check_workspace_status()
        sleep(CHECK_INTERVAL)  # Wait for a certain interval before checking again

if __name__ == "__main__":
    main()

With this Pseudocode example and the previous link shared by Sidra you can try a demo implementation https://repost.aws/questions/QU2-VcSQGfRyK9oPyKza18WA/opportunistically-migrate-workspace-to-new-bundle#ANvtWIUPe6Tyqz-znzt08GWg

Additionally, automate Workspace updates, utilizing cron jobs for scheduling the execution of your monitoring and updating script.

https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/workspaces.html You can use this documentation to make your own script , preferably you will need either the SDK or CLI to trigger your bundle update process for your respective Workspace.

I have gotten as far as the code to do the workspace migration to a new bundle and works in a stand-alone script

import boto3
aws = boto3.session.Session()
client = aws.client('workspaces')

response = client.migrate_workspace(SourceWorkspaceId='ws-abcde1234',BundleId='wsb-9876xyz')

I think some of the code has to be changed. For example user_type is probably RunningMode

        if RunningMode == "AUTO_STOP" and not workspace.is_active:
          perform_update(workspace)
        elif RunningMode == "ALWAYS_ON" and (current_time - last_active) >= 12 * HOUR:
          perform_update(workspace)

I haven't been able to find information about workspace.last_active_time. Closest has been describe-workspaces-connection-status which includes a value for LastKnownUserConnectionTimestamp but I am not sure how to make it work in the script you provided.