【Django】使用信号机制保证Model的字段规则

使用信号机制保证Model的字段规则

• 需求描述
• 实现方式

需求描述

最近在写Django DRM，我这里有一个Model，是记录节点信息的，几个相关字段如下：

字段名	字段描述	备注
node_id	节点的ID	主键，但是根据IP生成
node_ip	节点的IP	节点的IP地址
web_url	节点服务的URL	和IP也是关联的

现在的需求就是，在创建、更改一条节点信息时，node_id和web_url同步做校验修改，如下：

实现方式

这里采用Django的信号机制来实现这个需求。编写函数到signals_receiver.py：

from django.db.models import signals
from django.dispatch import receiver

from prometheus.models import PrometheusStatus


@receiver(signals.post_save, sender=PrometheusStatus)
def monitor_prometheus_create(instance, **kwargs):
    node_id = 'P'
    for num in instance.node_ip.split('.'):
        node_id += num.zfill(3)
    web_url = "{protocol}://{node_ip}:{port}".format(protocol='http', node_ip=instance.node_ip, port='9090')
    if instance.node_id == node_id and instance.web_url == web_url:
        return
    PrometheusStatus.objects.filter(node_ip=instance.node_ip).update(node_id=node_id, web_url=web_url)

说明一下，node_id的规则是P+IP补零后的12位写法，比如192.168.1.1的ID就是P192168001001；
这里的信号网上有些解决方案是使用post_init，我个人认为post_save更合适，这样无论是新建还是后续进行变更，都会更新对应的值；
这里我修改一下IP：

保存以后查看结果：

另外，因为我的signals_receiver.py文件是单独放在app的details目录下的，所以我还在urls.py里面import了一下：

from .details import signals_receiver