CORD-1010 Fix Controller, Image, Network, and User model_policies
Change-Id: I42f8771048090666acd9d75ca32b6081a72ce880
diff --git a/xos/synchronizer/model_policies/model_policy_User.py b/xos/synchronizer/model_policies/model_policy_User.py
index cc37db1..448b58f 100644
--- a/xos/synchronizer/model_policies/model_policy_User.py
+++ b/xos/synchronizer/model_policies/model_policy_User.py
@@ -1,13 +1,12 @@
from synchronizers.new_base.modelaccessor import *
def handle(user):
- # user = User.get(user_id)
-
- controller_users = ControllerUser.objects.filter(user=user)
+ controller_users = ControllerUser.objects.filter(user_id=user.id)
existing_controllers = [cu.controller for cu in controller_users]
+ existing_controller_ids = [c.id for c in existing_controllers]
all_controllers = Controller.objects.all()
for controller in all_controllers:
- if controller not in existing_controllers:
+ if controller.id not in existing_controller_ids:
ctrl_user = ControllerUser(controller=controller, user=user)
ctrl_user.save()