コード例 #1
0
ファイル: views.py プロジェクト: polmuz/weekkit
def add_friend(request):
    if request.method == 'POST':
        username = request.POST['username']
        try:
            friend = User.objects.get(username=username)
        except User.DoesNotExist:
            return redirect("/friends/?error=no_friend")
        Friendship.create_friendship(request.user, friend)
        return redirect("/friends/")
コード例 #2
0
ファイル: tests.py プロジェクト: polmuz/weekkit
    def setUp(self):
        user1 = User.objects.create_user('johndoe',
                                         '*****@*****.**',
                                         'johndoe')
        self.client.login(username='******', password='******')

        user2 = User.objects.create_user('janedoe',
                                         '*****@*****.**',
                                         'janedoe')

        Friendship.create_friendship(user1, user2)