2017-02-19 61 views
2

現在我正在用Django中的ManyToManyField設計數據建模。ManyToMany中的不工作.add()函數Python3的關係Django

我想使每個會話具有各種儀器,

即)

一個有低音。

B有一個鼓。

所以,我測試下一碼

from django.test import TestCase 
from Session.models import Instrument 
from Applicants.models import MyUser 
from Session.models import Session 

class InstTest(TestCase): 
    def setUp(self): 
     i1 = Instrument.objects.create(name='bass') 
     i1.save() 
     i2 = Instrument.objects.create(name='guitar') 
     i2.save() 
     i3 = Instrument.objects.create(name='drum') 
     i3.save() 

     u1 = MyUser.objects.create(email='[email protected]', nickname='A') 
     u1.save() 
     u2 = MyUser.objects.create(email='[email protected]', nickname='B') 
     u2.save() 

     s1 = Session.objects.create(players=u1) 
     s1.save() 
     s1.instruments.add(i1) 

     s2 = Session.objects.create(players=u2) 
     s2.save() 
     s2.instruments.create(name='drum') 

    def test_example(self): 
     bass = Instrument.objects.get(name='bass') 
     guitar = Instrument.objects.get(name='guitar') 
     user = MyUser.objects.get(nickname='A') 
     session = Session.objects.get(players=user) 
     self.assertEqual('bass', bass.name) 
     self.assertEqual('A', session.players.nickname) 

     # My problem is in this line 
     self.assertEqual('bass', session.instruments.name) 

,這是導致Pycharm

/home/user/vEnv/bin/python /home/user/pycharm-2016.3.2/helpers/pydev/pydevd.py --multiproc --qt-support --client 127.0.0.1 --port 40507 --file /home/user/pycharm-2016.3.2/helpers/pycharm/django_test_manage.py test BandMaker.tests.InstTest.test_example /home/user/PycharmProjects/Project 
Testing started at --:-- PM ... 
warning: Debugger speedups using cython not found. Run '"/home/user/vEnv/bin/python" "/home/user/pycharm-2016.3.2/helpers/pydev/setup_cython.py" build_ext --inplace' to build. 
pydev debugger: process 4984 is connecting 

Connected to pydev debugger (build 163.10154.50) 
Creating test database for alias 'default'... 

Failure 
Traceback (most recent call last): 
    File "/home/user/PycharmProjects/project/BandMaker/tests.py", line 31, in test_example 
    self.assertEqual('bass', player.instruments.name) 
AssertionError: 'bass' != None 

Destroying test database for alias 'default'... 

Process finished with exit code 1 

我嘗試用其他方式來測試像s1.instruments.create(name='bass'),但我能看到的只是'bass' != None

我不明白爲什麼s1.instruments.add(i1)不起作用。 所以我需要誰來幫助我。

而且,這裏是其他模型類。

/Session/models.py

from django.db import models 
from Applicants.models import MyUser 

# Create your models here. 
class Session(models.Model): 
    players = models.ForeignKey(
     MyUser, 
     verbose_name='own user info', 
     related_name='%(app_label)s_%(class)s_related' 
    ) 

    instruments = models.ManyToManyField(
     'Session.Instrument', 
     verbose_name='Session Instrument List', 
     related_name='%(app_label)s_%(class)s_instruments' 
    ) 

class Instrument(models.Model): 
    name = models.CharField(max_length=50, verbose_name='instrument name') 

自定義用戶類

from django.db import models 
from django.contrib.auth.models import (
    BaseUserManager, AbstractBaseUser 
) 

class MyUserManager(BaseUserManager): 
    def create_user(self, email, nickname, password=None): 
     """ 
     Creates and saves a User with the given email, date of 
     birth and password. 
     """ 
     if not email: 
      raise ValueError('Users must have an email address') 

     user = self.model(
      email=self.normalize_email(email), 
      nickname=nickname, 
     ) 

     user.set_password(password) 
     user.save(using=self._db) 
     return user 

    def create_superuser(self, email, nickname, password): 
     """ 
     Creates and saves a superuser with the given email, date of 
     birth and password. 
     """ 
     user = self.create_user(
      email=email, 
      password=password, 
      nickname=nickname, 
     ) 
     user.is_admin = True 
     user.save(using=self._db) 
     return user 


class MyUser(AbstractBaseUser): 
    email = models.EmailField(
     verbose_name='email', 
     max_length=255, 
     unique=True, 
    ) 
    nickname = models.CharField(
     verbose_name='nickname', 
     max_length=50, 
     blank=False, 
     unique=False, 
     default='unknown' 
    ) 
    is_active = models.BooleanField(default=True) 
    is_admin = models.BooleanField(default=False) 

    objects = MyUserManager() 

    USERNAME_FIELD = 'email' # When username is required, You must use this field! 
    REQUIRED_FIELDS = ['nickname'] 

    def get_full_name(self): 
     # The user is identified by their email address 
     return self.email 

    def get_short_name(self): 
     # The user is identified by their email address 
     return self.email 

    def __str__(self):    # __unicode__ on Python 2 
     return self.email 

    def has_perm(self, perm, obj=None): 
     "Does the user have a specific permission?" 
     # Simplest possible answer: Yes, always 
     return True 

    def has_module_perms(self, app_label): 
     "Does the user have permissions to view the app `app_label`?" 
     # Simplest possible answer: Yes, always 
     return True 

    @property 
    def is_staff(self): 
     "Is the user a member of staff?" 
     # Simplest possible answer: All admins are staff 
     return self.is_admin 

回答

2

問題是你測試的方式,如果會話具有儀器:

self.assertEqual('bass', session.instruments.name)

session.instruments沒有屬性名稱,它是一個ManyToMany對象。

我覺得這個測試應該工作:

self.assertEqual('bass', session.instruments.all()[0].name)

+0

謝謝您的回答! =) – Irrationnelle