Feature flag for which repo to use in environment

alphagov · May 7, 2014 · c64ebb8 · c64ebb8
1 parent 212c3e2
commit c64ebb8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/backdrop/core/repository.py b/backdrop/core/repository.py
@@ -1,8 +1,8 @@
-
 import json
 import logging
 
 import requests
+import os
 
 from backdrop.core.data_set import DataSetConfig
 from backdrop.core.user import UserConfig
@@ -155,7 +155,11 @@ class UserConfigRepository(object):
 
     def __init__(self, db):
         self._db = db
-        self._repository = _Repository(db, UserConfig, "users", "email")
+        if 'USERS_USE_HTTP_REPO' in os.environ \
+           and os.environ['USERS_USE_HTTP_REPO']:
+            self._repository = HttpRepository(db, UserConfig, "users", "email")
+        else:
+            self._repository = _Repository(db, UserConfig, "users", "email")
 
     def save(self, user_config):
         self._repository.save(user_config)