Tony Mack | a7dbd42 | 2015-01-05 22:48:11 -0500 | [diff] [blame^] | 1 | from core.models import * |
2 | |||||
3 | def handle(site): | ||||
4 | # make sure site has a ControllerSite record for each controller | ||||
5 | ctrl_sites = ControllerSite.objects.filter(site=site) | ||||
6 | existing_controllers = [cs.controller for cs in ctrl_sites] | ||||
7 | |||||
8 | all_controllers = Controller.objects.all() | ||||
9 | for ctrl in all_controllers: | ||||
10 | if ctrl not in existing_controllers: | ||||
11 | ctrl_site = ControllerSite(controller=ctrl, site=site) | ||||
12 | ctrl_site.save() |