Hosts: Add ChallengeHostTeam model getter functions. (#1061)

Cloud-CV · Jun 25, 2017 · f86a39f · f86a39f
1 parent 588fb8c
commit f86a39f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/apps/hosts/utils.py b/apps/hosts/utils.py
@@ -1,6 +1,10 @@
-from .models import ChallengeHost
+from base.utils import get_model_object
+
+from .models import ChallengeHost, ChallengeHostTeam
 
 
 def get_challenge_host_teams_for_user(user):
     """Returns challenge host team ids for a particular user"""
     return ChallengeHost.objects.filter(user=user).values_list('team_name', flat=True)
+
+get_challenge_host_team_model = get_model_object(ChallengeHostTeam)