fix: 更新账号后 立即推送 (#9462)

Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-02-08 13:36:32 +08:00
committed by GitHub
parent 32afa214fd
commit d308efc63b
2 changed files with 19 additions and 3 deletions

View File

@@ -39,11 +39,22 @@ class AccountSerializerCreateValidateMixin:
attrs = super().validate(attrs)
return self.set_secret(attrs)
@staticmethod
def push_account(instance, push_now):
if not push_now:
return
push_accounts_to_assets.delay([instance.id], [instance.asset_id])
def create(self, validated_data):
push_now = validated_data.pop('push_now', None)
instance = super().create(validated_data)
if push_now:
push_accounts_to_assets.delay([instance.id], [instance.asset_id])
instance = super().create(validated_data, push_now)
self.push_account(instance, push_now)
return instance
def update(self, instance, validated_data):
push_now = validated_data.pop('push_now', None)
instance = super().update(instance, validated_data)
self.push_account(instance, push_now)
return instance