jumpserver/apps/assets/models/asset/host.py

18 lines
478 B
Python

from django.db import models
from django.utils.translation import gettext_lazy as _
from assets.const import Category
from .common import Asset
from .device_info import DeviceInfo
class Host(Asset):
device_info = models.ForeignKey(DeviceInfo, on_delete=models.SET_NULL,
null=True, verbose_name=_("Device info"))
def save(self, *args, **kwargs):
self.category = Category.HOST
return super().save(*args, **kwargs)