[Feature] 添加signals 解耦代码

This commit is contained in:
ibuler
2017-12-12 12:19:45 +08:00
parent cbc000696e
commit 99b4c66b5e
41 changed files with 420 additions and 498 deletions

View File

@@ -96,12 +96,16 @@ class AdminUser(models.Model):
def become_pass(self, password):
self._become_pass = signer.sign(password)
def get_related_assets(self):
assets = []
for cluster in self.cluster_set.all():
assets.extend(cluster.assets.all())
assets.extend(self.asset_set.all())
return list(set(assets))
@property
def assets_amount(self):
amount = 0
for cluster in self.cluster_set.all():
amount += cluster.assets.all().count()
return amount
return len(self.get_related_assets())
class Meta:
ordering = ['name']
@@ -209,9 +213,14 @@ class SystemUser(models.Model):
'private_key_file': self.private_key_file,
}
def get_clusters_assets(self):
from .asset import Asset
clusters = self.cluster.all()
return Asset.objects.filter(cluster__in=clusters)
@property
def assets_amount(self):
return self.assets.count()
return len(self.get_clusters_assets())
def to_json(self):
return {