source: branches/exodus-admin/models.py@ 6390

# (c) Roland van Laar 2006
# vi:et:ts=4
from django.db import models
from django.contrib import admin

# Create your models here.

#No need to formalize it; CHOICES don't change much

STATUS_CHOICES = ( 
                        ('up', 'up'),
                        ('dw', 'down'),
                        ('pl', 'planned'),
                        )

POLAR_CHOICES = (
                        ('hr', 'horizontal'),
                        ('vr', 'vertical'),
                        )

INTERFACE_TYPE_CHOICES = (
                        ('eth', 'eth'),
                        ('11a', '11a'),
                        ('11b', '11b'),
                        ('11g', '11g'),
                        )
WIFI_MODE_CHOICES = (
                        ('ms', 'master'),
                        ('mn', 'managed'),
                        )

class ExtendedModel(models.Model):
    class Meta:
        abstract = True
        verbose_name = 'ExtendedModel'

    def as_list(self):
        meta = self._meta
        values = [(f.verbose_name, getattr(self, f.attname))  \
                for f in meta.local_fields if f.verbose_name not in ( 'ID')]
        return values

class Antenna(ExtendedModel):
    type = models.CharField(max_length=20, unique=True)
    gain = models.IntegerField(max_length=3)

    def __str__(self):
        return self.type

class Location(ExtendedModel):
    description = models.CharField(max_length=200, unique=True)
    longitude = models.DecimalField(max_digits=8,decimal_places=6)
    latitude = models.DecimalField(max_digits=8,decimal_places=6)

    class Meta:
        verbose_name = 'Location'

    def __str__(self):
        return self.description


#XXX: Name Domain perhaps?
class DnsServer(ExtendedModel):
    ipaddress = models.IPAddressField()
    domainname = models.CharField(max_length=50, unique=True)
    
    class Meta:
        verbose_name = 'Dns Server'
    
    def __str__(self):
        return self.ipaddress

class Network(ExtendedModel):
        name = models.CharField(max_length=50, unique=True)
        ipspacestart = models.IPAddressField()
        netmask = models.IntegerField()

    class Meta:
        verbose_name = 'Network'

        def __str__(self):
                return self.name


class Node(ExtendedModel):
    name = models.CharField(max_length=30, unique=True)
    location = models.ForeignKey(Location)
    status = models.CharField(max_length=10, choices=STATUS_CHOICES, default=1)
    masterip = models.IPAddressField(unique=True)
    network = models.ForeignKey(Network,default=1)
 
    class Meta:
        ordering = ['name']
        verbose_name = 'Node'

    def __str__(self):
        return self.name

class Interface(ExtendedModel):
        node = models.ForeignKey(Node)
        type = models.CharField(max_length=10, choices=INTERFACE_TYPE_CHOICES, \
            default=1)
        iface = models.CharField(max_length=10, verbose_name="interface", \
            default='eth0')
        ip = models.IPAddressField(unique=True)
        netmask = models.IntegerField(default=30)
        polar = models.CharField(blank=True, max_length=10, choices=POLAR_CHOICES)
        ssid = models.CharField(max_length=30, blank=True, null=True)
        mode = models.CharField(max_length=10, choices=WIFI_MODE_CHOICES)
        channel = models.IntegerField(blank=True, null=True)
        antenna = models.ForeignKey(Antenna, blank=True, null=True)
        shortdesc = models.CharField(blank=True, max_length=10)
        desc = models.CharField(blank=True, max_length=100)
        link = models.ForeignKey('self', blank=True, null=True)
        
        class Meta:
                unique_together = ('node', 'iface')
                verbose_name = 'Interface'
        
        def __str__(self):
                return "%s/%s" % (self.node, self.iface)

class InterfaceAlias(ExtendedModel):
        iface = models.ForeignKey(Interface,related_name='child')
        ip = models.IPAddressField(unique=True)
        netmask = models.IntegerField(default=30)
        shortdesc = models.CharField(blank=True, max_length=10)
        desc = models.CharField(blank=True, max_length=100)
        linkAlias = models.ForeignKey('self', blank=True, null=True)
        link = models.ForeignKey(Interface, blank=True, null=True)
        
        def __str__(self):
                return "Alias %s" % (self.iface)

#
# XXX: Needs rethinking, dhcp[start|stop] could be removed if whole publicAP is
# going to be activated as DHCP range, or only number of fixed clients needs to
# be defined, all other could be calculated on the fly if needed

class PublicAP(ExtendedModel):
    iface = models.ForeignKey(Interface,related_name='ap', verbose_name="interface")
    ip = models.IPAddressField(unique=True)
    netmask = models.IntegerField(default=28)
    dhcpstart = models.IPAddressField()
    dhcpstop = models.IPAddressField()
    shortdesc = models.CharField(max_length=10)
    desc = models.CharField(max_length=100)

    class Meta:
        verbose_name = "Public Access Point"

    def __str__(self):
        return "%s:%s" % (self.iface,self.pk)

class DhcpStatic(ExtendedModel):
    hostname = models.CharField(max_length=10,unique=True)
    macaddress = models.CharField(max_length=17)
    address = models.IPAddressField()
    PublicAP = models.ForeignKey(PublicAP)

    class Meta:
        verbose_name = "Static host"

    def __str__(self):
        return self.hostname