fetch toggles

Author: calellowitz

connectid/settings.py

@@ -334,6 +334,7 @@ class APIVersion:
 COMMCARE_CONNECT_CLIENT_SECRET = env("COMMCARE_CONNECT_CLIENT_SECRET", default="")
 CONNECT_INVITED_USER_URL = "https://connect.dimagi.com/users/invited_user/"
 CONNECT_RESEND_INVITES_URL = "https://connect.dimagi.com/users/resend_invites/"
+CONNECT_TOGGLES_URL = "https://connect.dimagi.com/users/toggles/"
 
 # List of countries that are blacklisted from using PersonalID
 # Example: ["us", "ca", "gb"] (Alpha-2 country codes)

flags/utils.py

@@ -0,0 +1,11 @@
+from waffle.models import Switch
+
+from utils.connect import get_connect_toggles
+
+
+def get_user_toggles(username=None, phone_number=None):
+    switches = Switch.objects.all()
+    connect_switches = get_connect_toggles(username=username, phone_number=phone_number)
+    toggles = {switch.name: switch.active for switch in switches}
+    toggles.update(connect_switches)
+    return toggles

flags/views.py

@@ -1,10 +1,13 @@
 from django.http import JsonResponse
 from django.views import View
-from waffle.models import Switch
+
+from .utils import get_user_toggles
 
 
 class TogglesView(View):
     def get(self, request):
-        switches = Switch.objects.all()
-        toggles = {switch.name: switch.active for switch in switches}
+        username = None
+        if request.user.is_authenticated:
+            username = request.user.username
+        toggles = get_user_toggles(username=username)
         return JsonResponse({"toggles": toggles})

users/tests/test_views.py

@@ -1109,6 +1109,12 @@ def test_phone_mismatch(self, mock_verify_token, authed_client_token):
 
 @pytest.mark.django_db
 class TestStartConfigurationView:
+    @pytest.fixture(scope="class", autouse=True)
+    def mock_get_user_toggles(self):
+        with mock.patch("users.views.get_user_toggles") as mock_get_toggles:
+            mock_get_toggles.return_value = {}
+            yield mock_get_toggles
+
     @patch("utils.app_integrity.decorators.check_number_for_existing_invites")
     def test_no_integrity_token(self, check_number_mock, client):
         check_number_mock.return_value = False
@@ -1200,6 +1206,7 @@ def test_session_started(self, mock_country_code, client):
         assert session.gps_location == gps_location
         assert not session.is_phone_validated
         assert session.device_id == "device_id"
+        assert "toggles" in response.json()
 
     @skip_app_integrity_check
     @patch("users.models.ConfigurationSession.country_code", new_callable=PropertyMock)

users/views.py

@@ -23,6 +23,7 @@
 from rest_framework.decorators import api_view, authentication_classes, permission_classes
 from rest_framework.views import APIView
 
+from flags.utils import get_user_toggles
 from services.ai.ocs import OpenChatStudio
 from utils import get_ip, get_sms_sender, send_sms
 from utils.app_integrity.decorators import require_app_integrity
@@ -108,10 +109,12 @@ def start_device_configuration(request):
 
     token_session.save()
 
+    toggles = get_user_toggles(phone_number=data["phone_number"])
     response_data = {
         "required_lock": ConnectUser.get_device_security_requirement(data["phone_number"], request.invited_user),
         "demo_user": is_demo_user,
         "token": token_session.key,
+        "toggles": toggles,
     }
 
     if request.version == settings.API_VERSION.V1:

utils/connect.py

@@ -19,3 +19,16 @@ def resend_connect_invite(user):
         "name": user.name,
     }
     requests.post(url, auth=auth, data=data)
+
+
+def get_connect_toggles(username=None, phone_number=None):
+    url = settings.CONNECT_TOGGLES_URL
+    auth = (settings.COMMCARE_CONNECT_CLIENT_ID, settings.COMMCARE_CONNECT_CLIENT_SECRET)
+    params = {}
+    if username is not None:
+        params["username"] = username
+    elif phone_number is not None:
+        params["phone_number"] = phone_number
+    response = requests.get(url, auth=auth, params=params)
+    data = response.json()
+    return {toggle["name"]: toggle["active"] for toggle in data.get("toggles", {})}