| def handle(user): | |
| from core.models import Controller, ControllerSite, ControllerUser, User | |
| from collections import defaultdict | |
| # user = User.get(user_id) | |
| controller_users = ControllerUser.objects.filter(user=user) | |
| existing_controllers = [cu.controller for cu in controller_users] | |
| all_controllers = Controller.objects.all() | |
| for controller in all_controllers: | |
| if controller not in existing_controllers: | |
| ctrl_user = ControllerUser(controller=controller, user=user) | |
| ctrl_user.save() | |