planetstack/model_policies/model_policy_User.py

   1 def handle(user):
   2     from core.models import Controller, ControllerSite, ControllerUser
   3     from collections import defaultdict
   4
   5     controller_users = ControllerUser.objects.filter(user=user)
   6     existing_controllers = [cu.controller for cu in controller_users]
   7     all_controllers = Controller.objects.all()
   8     for controller in all_controllers:
   9         if controller not in existing_controllers:
  10             ctrl_user = ControllerUser(controller=controller, user=user)
  11             ctrl_user.save()
  12