Django обмен мнениями между связанными предметами

Question

Рассмотрите систему бронирования номеров. У вас могут быть модели здания, пола, комнаты, а также бронирование. Мы даем комнате название основано на его здании и пол:

class Room(models.Model): 
    number = models.PositiveIntegerField() 
    name = models.CharField(..) 
    floor = models.ForeignKey('Floor') 

    def __str__(self): 
     return '%s, #%d Floor %d, %s' % (
      self.name, 
      self.number, 
      self.floor.number, 
      self.floor.building.name 
     ) 

Это крайне неэффективно, когда вы делаете сотни из них (например, список администратора, большие отчеты и т.д.), поэтому я принял в письменной форме менеджеров:

class RoomManager(models.Manager): 
    def get_queryset(self): 
     return super().get_queryset().annotate(
      roomname=Concat(
       'name', 
       V(', #'), 
       'number' 
       V(' Floor '), 
       'floor__number' 
       V(', '), 
       'floor__building__name', 
       output_field=models.CharField() 
      ), 
     ) 

И что работает. Он делает все, что я хотел. Это быстро, и я переработал __str__, чтобы сделать if hasattr(self, 'roomname'): return self.roomname, прежде чем он сделает ужасный многострочный построитель строк.

Но теперь вместо этого у меня есть Бронирование. Каждый экземпляр бронирования связан с одной комнатой. Есть много случаев, когда перечислять заказы, я на самом деле также перечисляю имена комнат.

Что я сделал это написать BookingManager:

class RoomManager(models.Manager): 
    def get_queryset(self): 
     return super().get_queryset().annotate(
      roomname=Concat(
       'room__name', 
       V(', #'), 
       'room__number' 
       V(' Floor '), 
       'room__floor__number' 
       V(', '), 
       'room__floor__building__name', 
       output_field=models.CharField() 
      ), 
     ) 

Но что, черт возьми? Я повторяюсь. Django - все о DRY, и здесь я копирую и вставляю огромную грязную аннотацию. Это отвратительно.

Мой вопрос ... Есть ли другой способ?

Answer 1

При написании этого у меня была идея, я мог бы написать метод, позволяющий передавать префикс для связи с комнатой. Таким образом, я мог бы назвать его от всего, что было связано с комнате, и было бы получить правильный материал:

class RoomManager(models.Manager): 
    def get_queryset(self): 
     return super().get_queryset().annotate(
      roomname=RoomManager.roomname() 
     ) 

    @staticmethod 
    def roomname(prefix=''): 
     return Concat(
      prefix + 'name', 
      V(', #'), 
      prefix + 'number' 
      V(' Floor '), 
      prefix + 'floor__number' 
      V(', '), 
      prefix + 'floor__building__name', 
      output_field=models.CharField() 
     ) 

И в BookingManager я могу только аннотацию на RoomManager.roomname('room__')

Это чище, и я буду использовать это в другом месте, но это не очень умно.

Answer 2

Как насчет чего-то подобного?

class Room(models.Model): 
    number = models.PositiveIntegerField() 
    name = models.CharField(..) 
    floor = models.ForeignKey('Floor') 

    def __str__(self): 
     return '%s, %s %s' % (
      self.name, 
      self.number, 
      self.floor 
     ) 

class Floor(models.Model): 
    number = models.PositiveIntegerField() 
    building = models.ForeignKey('Building') 

    def __str__(self): 
     return 'Floor %d, %s' % (
      self.number, 
      self.building 
     ) 

class Building(models.Model): 
    name = models.CharField(...) 

    def __str__(self): 
     return '%s' % (
      self.name 
     )