Changeset 9818 for src/django_gheat/gheat/models.py

Timestamp:

Dec 23, 2011, 8:28:57 AM (13 years ago)

Author:

rick

Message:

Type is needed to start playing around with filters based on the function/type of a unit.

File:

• src/django_gheat/gheat/models.py (modified) (3 diffs)

src/django_gheat/gheat/models.py

@@ -7 +7 @@
 MAX_SIGNAL = 100
 
+ACCESPOINT_TYPE_CHOICES= (
+  ('accespoint', 'Acess Point'),
+  ('interlink', 'InterLink'),
+)
 
 # http://stackoverflow.com/questions/849142/how-to-limit-the-maximum-value-of-a-numeric-field-in-a-django-model/849426#849426
@@ -71 +75 @@
   mac = models.CharField(max_length=17)
   ssid = models.CharField(max_length=64)
+  type = models.CharField(max_length=20, null=True, blank=True, choices=ACCESPOINT_TYPE_CHOICES)
   organization = models.ForeignKey(Organization,null=True, blank=True)
   encryptie = models.BooleanField()
@@ -78 +83 @@
     return "%s - %s" % (self.mac, self.ssid)
 
+  @staticmethod
+  def get_type_by_ssid(ssid):
+    if ssid.startswith('ap'):
+      type = 'accespoint'
+    elif ssid.startswith('il'):
+      type = 'interlink'
+    else:
+      type = None
+    return type
 
   def save(self, *args, **kwargs):
-    self.organization = self.get_organization(self.ssid)
     super(Accespoint, self).save(*args, **kwargs)