Changeset 9592

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

Files:

• gheat/admin.py (modified) (2 diffs)
• gheat/management/commands/import_droidstumbler.py (modified) (2 diffs)
• gheat/management/commands/import_kismet.py (modified) (2 diffs)
• gheat/models.py (modified) (3 diffs)
• gheat/sql (added)
• gheat/sql/organization.sql (added)
• wlheatmap/filters.py (modified) (1 diff)
• wlheatmap/nodelist.py (modified) (1 diff)
• wlheatmap/templates/js/LayerBase.js (modified) (1 diff)

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):

src/django_gheat/wlheatmap/filters.py

@@ -20 +20 @@
       # Get list if accespoints found in the specific 'Meting', and make this is list.
       wirelessleiden_ssid = Meting.objects.filter(meetrondje=mr,
-        accespoint__organization='WirelessLeiden').\
+        accespoint__organization__name='WirelessLeiden').\
         values_list('accespoint__ssid',flat=True).\
         order_by('accespoint__ssid').distinct()

src/django_gheat/wlheatmap/nodelist.py

@@ -40 +40 @@
   filter = {}
   filter.update({
-    'organization' : 'WirelessLeiden',
+    'organization__name' : 'WirelessLeiden',
     'meting__latitude__lt' : maxlat,
     'meting__longitude__lt' : maxlon,

src/django_gheat/wlheatmap/templates/js/LayerBase.js

@@ -184 +184 @@
     initialize: function(name, options) {
         var url = get_balanced_urls('http://', {{ settings.DJANGO_CDN_DOMAINS|safe }},
-          '{{ settings.DJANGO_PREFIX }}/wlheatmap/tile/${z}/${x},${y}.png?colour=255,0,0&accespoint__organization=WirelessLeiden');
+          '{{ settings.DJANGO_PREFIX }}/wlheatmap/tile/${z}/${x},${y}.png?colour=255,0,0&accespoint__organization__name=WirelessLeiden');
         options = OpenLayers.Util.extend({ numZoomLevels: 21 }, options);
         var newArguments = [name, url, options];