You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
141 lines
5.1 KiB
141 lines
5.1 KiB
import os
|
|
from typing import Optional, AnyStr
|
|
from subprocess import run
|
|
from django.db import models
|
|
from django.conf import settings
|
|
from django.utils.translation import gettext_lazy as _
|
|
from jsonfield import JSONField
|
|
|
|
from djing.fields import MACAddressField, MyGenericIPAddressField
|
|
from djing.lib import MyChoicesAdapter
|
|
from group_app.models import Group
|
|
from . import dev_types
|
|
from .base_intr import DevBase
|
|
|
|
|
|
class DeviceDBException(Exception):
|
|
pass
|
|
|
|
|
|
class DeviceMonitoringException(Exception):
|
|
pass
|
|
|
|
|
|
class Device(models.Model):
|
|
_cached_manager = None
|
|
|
|
ip_address = MyGenericIPAddressField(verbose_name=_('Ip address'), null=True, blank=True)
|
|
mac_addr = MACAddressField(verbose_name=_('Mac address'), null=True, blank=True, unique=True)
|
|
comment = models.CharField(_('Comment'), max_length=256)
|
|
DEVICE_TYPES = (
|
|
('Dl', dev_types.DLinkDevice),
|
|
('Pn', dev_types.OLTDevice),
|
|
('On', dev_types.OnuDevice),
|
|
('Ex', dev_types.EltexSwitch),
|
|
('Zt', dev_types.Olt_ZTE_C320),
|
|
('Zo', dev_types.ZteOnuDevice)
|
|
)
|
|
devtype = models.CharField(_('Device type'), max_length=2, default=DEVICE_TYPES[0][0],
|
|
choices=MyChoicesAdapter(DEVICE_TYPES))
|
|
man_passw = models.CharField(_('SNMP password'), max_length=16, null=True, blank=True)
|
|
group = models.ForeignKey(Group, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('Device group'))
|
|
parent_dev = models.ForeignKey('self', verbose_name=_('Parent device'), blank=True, null=True,
|
|
on_delete=models.SET_NULL)
|
|
|
|
snmp_extra = models.CharField(_('SNMP extra info'), max_length=256, null=True, blank=True)
|
|
extra_data = JSONField(verbose_name=_('Extra data'),
|
|
help_text=_('Extra data in JSON format. You may use it for your custom data'),
|
|
blank=True, null=True)
|
|
|
|
NETWORK_STATES = (
|
|
('und', _('Undefined')),
|
|
('up', _('Up')),
|
|
('unr', _('Unreachable')),
|
|
('dwn', _('Down'))
|
|
)
|
|
status = models.CharField(_('Status'), max_length=3, choices=NETWORK_STATES, default='und')
|
|
|
|
is_noticeable = models.BooleanField(_('Send notify when monitoring state changed'), default=False)
|
|
|
|
class Meta:
|
|
db_table = 'dev'
|
|
permissions = (
|
|
('can_view_device', _('Can view device')),
|
|
)
|
|
verbose_name = _('Device')
|
|
verbose_name_plural = _('Devices')
|
|
ordering = ('id',)
|
|
|
|
def get_abons(self):
|
|
pass
|
|
|
|
def get_status(self):
|
|
return self.status
|
|
|
|
def get_manager_klass(self):
|
|
klasses = tuple(kl for kl in self.DEVICE_TYPES if kl[0] == self.devtype)
|
|
if len(klasses) > 0:
|
|
res = klasses[0][1]
|
|
if issubclass(res, DevBase):
|
|
return res
|
|
raise TypeError('one of types is not subclass of DevBase. '
|
|
'Or implementation of that device type is not found')
|
|
|
|
def get_manager_object(self) -> DevBase:
|
|
man_klass = self.get_manager_klass()
|
|
if self._cached_manager is None:
|
|
self._cached_manager = man_klass(self)
|
|
return self._cached_manager
|
|
|
|
# Can attach device to subscriber in subscriber page
|
|
def has_attachable_to_subscriber(self) -> bool:
|
|
mngr = self.get_manager_object()
|
|
return mngr.has_attachable_to_subscriber()
|
|
|
|
def __str__(self):
|
|
return "%s: (%s) %s %s" % (self.comment, self.get_devtype_display(), self.ip_address or '', self.mac_addr or '')
|
|
|
|
def update_dhcp(self, remove=False):
|
|
# FIXME: Remove static list of registrable device types
|
|
if self.devtype not in ('On', 'Dl', 'Zo'):
|
|
return
|
|
if self.group is None:
|
|
raise DeviceDBException(_('Device does not have a group, please fix that'))
|
|
code = self.group.code
|
|
newmac = str(self.mac_addr)
|
|
filepath = os.path.join(settings.BASE_DIR, 'devapp', 'onu_register.sh')
|
|
if remove:
|
|
param = 'del'
|
|
else:
|
|
param = 'update'
|
|
run((filepath, param, newmac, code))
|
|
|
|
def generate_config_template(self) -> Optional[AnyStr]:
|
|
mng = self.get_manager_object()
|
|
return mng.monitoring_template()
|
|
|
|
def register_device(self):
|
|
mng = self.get_manager_object()
|
|
if self.extra_data is None:
|
|
if self.parent_dev and self.parent_dev.extra_data is not None:
|
|
return mng.register_device(self.parent_dev.extra_data)
|
|
return mng.register_device(self.extra_data)
|
|
|
|
|
|
class Port(models.Model):
|
|
device = models.ForeignKey(Device, models.CASCADE, verbose_name=_('Device'))
|
|
num = models.PositiveSmallIntegerField(_('Number'), default=0)
|
|
descr = models.CharField(_('Description'), max_length=60, null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return "%d: %s" % (self.num, self.descr)
|
|
|
|
class Meta:
|
|
db_table = 'dev_port'
|
|
unique_together = (('device', 'num'),)
|
|
permissions = (
|
|
('can_toggle_ports', _('Can toggle ports')),
|
|
)
|
|
verbose_name = _('Port')
|
|
verbose_name_plural = _('Ports')
|
|
ordering = ('num',)
|