def get_user_profile(user): """ This function should be used to 'get or create' a user's profile """ try: profile = user.get_profile() except ObjectDoesNotExist: profile = UserProfile() profile.user = request.user profile.save() return profile
def setUp(self): """ Required Setup for testing: -Create a user with permission to add favourites -Create a user with default premissions -Create a show with dummy data -Create blank user profile so favourites can be added """ user = User.objects.create_user('john', '*****@*****.**', 'password') perm = Permission.objects.get(codename="can_have_favs") user.save() user.user_permissions.add(perm) profile = UserProfile() profile.user = user profile.save() user = User.objects.create_user('jane', '*****@*****.**', 'password') user.save() profile = UserProfile() profile.user = user profile.save() new_show = Show() new_show.name = "Some test show" new_show.id = 1234 new_show.save() new_entry = GuideEntry() #Compile a date/time stamp for the show year = datetime.now().strftime("%Y") month = datetime.now().strftime("%m") day = datetime.now().strftime("%d") date_object = datetime.strptime( year + '-' + month + '-' + day + ' 5:00 pm', '%Y-%m-%d %I:%M %p') new_entry.start = date_object new_entry.network = "SPIKE TV" new_entry.show = new_show new_entry.save()
def setUp(self): """ Required Setup for testing: -Create a user with permission to add favourites -Create a user with default premissions -Create a show with dummy data -Create blank user profile so favourites can be added """ user = User.objects.create_user('john', '*****@*****.**', 'password') perm = Permission.objects.get(codename="can_have_favs") user.save() user.user_permissions.add(perm) profile = UserProfile() profile.user = user profile.save() user = User.objects.create_user('jane', '*****@*****.**', 'password') user.save() profile = UserProfile() profile.user = user profile.save() new_show = Show() new_show.name = "Some test show" new_show.id = 1234 new_show.save() new_entry = GuideEntry() #Compile a date/time stamp for the show year = datetime.now().strftime("%Y") month = datetime.now().strftime("%m") day = datetime.now().strftime("%d") date_object = datetime.strptime(year + '-' + month + '-' + day + ' 5:00 pm', '%Y-%m-%d %I:%M %p') new_entry.start = date_object new_entry.network = "SPIKE TV" new_entry.show = new_show new_entry.save()