How to Simulate Blue-Green Deployment Switch in Python

import random
import time

class BlueGreenDeployment:
    def __init__(self, initial_env="blue"):
        self.environments = {"blue": "v1.0", "green": "v1.0"}
        self.active_env = initial_env
        self.running = True

    def deploy_new_version(self, version, target_env):
        if target_env == self.active_env:
            return f"Cannot deploy to active environment {target_env}"
        self.environments[target_env] = version
        return f"Deployed {version} to {target_env}"

    def health_check(self, env):
        return random.random() > 0.1  # 90% healthy

    def switch_traffic(self, target_env):
        if target_env == self.active_env:
            return f"Already on {target_env}"
        if not self.health_check(target_env):
            return f"Health check failed on {target_env}, aborting switch"
        self.active_env = target_env
        return f"Traffic switched to {target_env}"

    def rollback(self):
        previous = "blue" if self.active_env == "green" else "green"
        self.active_env = previous
        return f"Rolled back to {previous}"

    def status(self):
        return f"Active: {self.active_env} ({self.environments[self.active_env]}), " \
               f"Inactive: {'green' if self.active_env == 'blue' else 'blue'} ({self.environments['green' if self.active_env == 'blue' else 'blue']})"


if __name__ == "__main__":
    deploy = BlueGreenDeployment()
    print("Initial:", deploy.status())
    print(deploy.deploy_new_version("v2.0", "green"))
    print(deploy.switch_traffic("green"))
    print("After switch:", deploy.status())
    print(deploy.rollback())
    print("After rollback:", deploy.status())