从 GraphQL 计算一个值并将其保存在 Django 模型中

Question

我正在尝试从 GraphQL 计算一个值。我正在向 Django 模型发送突变，但在保存之前我想用 if 语句计算这个值（如果值大于 10 除以 2，如果小于 10 则乘以 2）。

不知道在哪里添加这个功能

这是我在 schema.py

中的突变

class CreatePrice(graphene.Mutation):
    price = graphene.Field(PriceType)

    class Arguments:
        price_data = PriceInput(required=True)

    @staticmethod
    def mutate(root, info, price_data):
        price = Price.objects.create(**price_data)
        return CreatePrice(price=price)

class Mutation(graphene.ObjectType):
    create_product = CreateProduct.Field()
    create_price = CreatePrice.Field()

schema = graphene.Schema(query = Query, mutation=Mutation)

这是我的 Django 模型。基价为计算值，函数名有两个选项（*2或/2取决于初始值）。

class Price(models.Model):
    base_price = models.CharField(max_length = 20)
    function_name = models.CharField(max_length = 20, choices = PROMO_FUNCTION)

    def __str__(self):
        return self.price_name

P.S。抱歉英语不好。谢谢！

Answer 1

我不知道你为什么用 CharField 代替 base_price。所以，我建议你这样做：

@staticmethod
def mutate(root, info, price_data):
    if int(price_data.base_price) >= 10:
        price_data.base_price = str(int(price_data.base_price) / 2)
    else:
        price_data.base_price = str(int(price_data.base_price) * 2)
    price = Price(base_price=price_data.base_price, function_name=price_data.function_name)
    price.save()
    return CreatePrice(price=price)

您还可以通过创建对象并在其上使用 save 方法在数据库中创建记录。

从 GraphQL 计算一个值并将其保存在 Django 模型中

Calculate a value from GraphQL and save it in Django models

python

django

django-models

graphql

graphene-python