from core.models import *

def handle(site):
    """ 
    ensure ControllerSiteDeployment object exists for each of the site's SiteDeployment objects
    """
    from core.models import SiteDeployments, Controller, ControllerSiteDeployments 
    from collections import defaultdict
    
    # get current controller site deployments	
    ctrl_site_deployments = ControllerSiteDeployments.objects.filter(site_deployment__site = site)
    ctrl_site_deployments_dict = {} 
    for ctrl_site_depl in controller_site_deployments:
        ctrl_site_deployments_dict[ctrl_site_depl.site_deployment] = ctrl_cite_depl

    # get current site deployments
    site_deployments = SiteDeployments.objects.filter(site=site)
 
    # for each site deployment, if there is no controller site deployment create one 
    for site_deployment in site_deployments:
        if site_deployment not in ctrl_site_deployments_dict:
            ctrl_site_deployment = ControllerSiteDeployment(
                site_deployment=site_deployment,
                controller=site_deployment.controller,
            )
            ctrl_site_deployment.save()