def get_user(request):
userid = responses.get_userid(request)
if userid:
try:
return User.objects.get(id=userid)
except User.DoesNotExist:
pass
return None
def get_or_create_user(request):
userid = get_userid(request)
if userid:
return User.objects.get_or_create(id=userid)
else:
user = None
tries = 5
while user is None and tries >= 0:
tries -= 1
userid = create_userid(request)
user = User.objects.create(id=userid)
if user:
return user, True
raise Exception("Can't create unique user")
def get_or_create_user(request):
userid = responses.get_userid(request)
if userid:
user, _ = User.objects.get_or_create(id=userid)
return user
else:
user = None
tries = 5
while not user and tries >= 0:
tries -= 1
userid = responses.create_userid(request)
user = User.objects.create(id=userid)
if user:
return user
raise Exception("Can't create unique user")
def get_user(request):
userid = get_userid(request)
if userid:
return User.objects.get(id=userid)
def test_create_user_id(self):
user_id = responses.create_userid(self.request)
self.assertEqual(user_id, responses.get_userid(self.request))
def test_set_user_id(self):
user_id = 'Aasdfa'
self.assertEqual(responses.set_userid(self.request, user_id), user_id)
self.assertEqual(responses.get_userid(self.request), user_id)
def test_get_user_id(self):
user_id = 'kjhsdafA'
self.request.session = {USER_ID_KEY: user_id}
self.assertEqual(responses.get_userid(self.request), user_id)
