Changeset 9592 for src/django_gheat/gheat

Timestamp:

Aug 28, 2011, 1:00:28 PM (13 years ago)

Author:

rick

Message:

Organization needs to be a special Model to make fudgings around with it more easy.

Location:

src/django_gheat/gheat

Files:

• admin.py (modified) (2 diffs)
• management/commands/import_droidstumbler.py (modified) (2 diffs)
• management/commands/import_kismet.py (modified) (2 diffs)
• models.py (modified) (3 diffs)
• sql (added)
• sql/organization.sql (added)

src/django_gheat/gheat/admin.py

@@ -5 +5 @@
   list_display = ['ssid','mac', 'organization']
   search_fields = ['ssid', 'mac', 'organization']
+  ordering = ['ssid']
 admin.site.register(Accespoint, AccespointAdmin)
 
@@ -18 +19 @@
 admin.site.register(Node, NodeAdmin)
 
+class OrganizationAdmin(admin.ModelAdmin):
+  list_display = ['name']
+  ordering = ['name']
+admin.site.register(Organization, OrganizationAdmin)
+
 admin.site.register(Apparatuur)
 admin.site.register(Gebruiker)

src/django_gheat/gheat/management/commands/import_droidstumbler.py

@@ -43 +43 @@
   return count
 
+organizations = dict(Organization.objects.all().values_list('name','id'))
+def get_organization_id_by_ssid(ssid):
+  name = Organization.get_name_by_ssid(ssid)
+  if not name:
+    return 'NULL'
+  else:
+    return int(organizations[name])
 
 def import_droidstumbler(filename, meetrondje):
@@ -96 +103 @@
       ssid, encryption = ap_pool[bssid]
       # Special trick in SSID ts avoid escaping in later stage
-      item = str((bssid.upper(),ssid.replace('%','%%'),encryption,Accespoint.get_organization(ssid)))
+      item = str((bssid.upper(),ssid.replace('%','%%'),encryption,get_organization_id_by_ssid(ssid)))
       sql_values.append(item)
-    counters['ap_added'] = bulk_sql('gheat_accespoint (`mac`, `ssid`, `encryptie`, `organization`)',sql_values)
+    counters['ap_added'] = bulk_sql('gheat_accespoint (`mac`, `ssid`, `encryptie`, `organization_id`)',sql_values)
 
   # Build mapping for meting import

src/django_gheat/gheat/management/commands/import_kismet.py

@@ -19 +19 @@
 from collections import defaultdict
 
-from import_droidstumbler import bulk_sql
+from import_droidstumbler import bulk_sql,get_organization_id_by_ssid
 
 logger = logging.getLogger(__name__)
@@ -76 +76 @@
       ssid, encryption = ap_pool[bssid]
       # Special trick in SSID ts avoid escaping in later stage
-      item = str((bssid.upper(),ssid.replace('%','%%'),encryption,Accespoint.get_organization(ssid)))
+      item = str((bssid.upper(),ssid.replace('%','%%'),encryption,get_organization_id_by_ssid(ssid)))
       sql_values.append(item)
-    counters['ap_added'] = bulk_sql('gheat_accespoint (`mac`, `ssid`, `encryptie`, `organization`)',sql_values)
+    counters['ap_added'] = bulk_sql('gheat_accespoint (`mac`, `ssid`, `encryptie`, `organization_id`)',sql_values)
 
   # Determine which Wireless Clients to add

src/django_gheat/gheat/models.py

@@ -58 +58 @@
 
 
-ORGANIZATION_CHOICES = (
-  ('WirelessLeiden', 'Wireless Leiden'),
-)
+class Organization(models.Model):
+  fullname = models.CharField(max_length=100,unique=True)
+  name = models.CharField(max_length=100,unique=True)
+  def __unicode__(self):
+    return self.fullname
+
+  @staticmethod
+  def get_name_by_ssid(ssid):
+    """ Try to determine the organization via the SSID """
+    name = None
+    if ssid.startswith('ap') and ssid.endswith('wleiden.net'):
+      name = 'WirelessLeiden'
+    elif ssid.startswith('ap') and 'WirelessLeiden' in ssid:
+      name = 'WirelessLeiden'
+    return name
 
 class Node(models.Model):
@@ -66 +78 @@
   latitude = models.FloatField(null=True,blank=True)
   longitude = models.FloatField(null=True,blank=True)
-  organization = models.CharField(max_length="50",choices=ORGANIZATION_CHOICES,null=True,blank=True)
+  organization = models.ForeignKey(Organization,null=True, blank=True)
+  class Meta:
+    unique_together = ('name', 'organization')
+
+  def __unicode__(self):
+    return "%s - %s" % (self.name, self.organization)
 
 class Accespoint(models.Model):
   mac = models.CharField(max_length=17)
   ssid = models.CharField(max_length=64)
-  organization = models.CharField(max_length="50",choices=ORGANIZATION_CHOICES,null=True,blank=True)
+  organization = models.ForeignKey(Organization,null=True, blank=True)
   encryptie = models.BooleanField()
   class Meta:
@@ -78 +95 @@
     return "%s - %s" % (self.mac, self.ssid)
 
-  @staticmethod
-  def get_organization(ssid):
-    """ Try to determine the organization via the SSID """
-    organization = ''
-    if ssid.startswith('ap') and ssid.endswith('wleiden.net'):
-      organization = 'WirelessLeiden'
-    elif ssid.startswith('ap') and 'WirelessLeiden' in ssid:
-      organization = 'WirelessLeiden'
-    return organization
 
   def save(self, *args, **kwargs):