Как конвертировать выбор строки Джанго Modelfield дорожить

Question

Предположим, у меня есть следующие модели Django:

class myModel(models.Model): 
    C = "myC" 
    D = "myD" 

    A = models.CharField(max_length=254, null=False, blank=False,) 
    B = models.CharField(
     max_length=254, 
     null=False, 
     blank=False, 
     choices=((C, "C"), (D, "D")), 
    ) 

    def set_B(self, choice_val): 
     self.B = #WHAT GOES HERE?? 
     self.save() 

Теперь предположим, что я создаю объект X типа MyModel. Затем я звоню X.set_B("D"). Я хочу, чтобы метод myModel set_B() устанавливал значение B на myModel.D. Как я могу это сделать? Какой код мне нужно поставить в set_B() так, чтобы он преобразует человекочитаемое значение «D» в myModel.D?

Answer 1

Вы можете просто использовать getattr Funtion.

class myModel(models.Model): 

A = models.CharField(max_length=254, null=False, blank=False,) 
B = models.CharField(
    max_length=254, 
    null=False, 
    blank=False, 
    choices=((C, "C"), (D, "D")), 
) 
C = #Something 
D = #Something 

def set_B(self, choice_val): 
    self.B = getattr(self,choice_val) 
    self.save() 

getattr функция возвращает значение атрибута choice_val.

Answer 2

Вы можете построить просмотровые dict на уровне класса и использовать его:

class MyModel(models.Model): 

    B_CHOICES = ((C, "C"), (D, "D")) 
    B_VALUES_FROM_NAME = {name: val for val, name in B_CHOICES} 
    B = models.CharField(
     max_length=254, 
     null=False, 
     blank=False, 
     choices=B_CHOICES, 
    ) 

    def set_B(self, name): 
     self.B = MyModel.B_VALUES_FROM_NAME[name] 
     self.save()