Changeset 9140 for src/django_gheat/gheat/management

Timestamp:

May 3, 2011, 9:24:27 PM (14 years ago)

Author:

rick

Message:

Import DroidStumbler Files.

File:

• src/django_gheat/gheat/management/commands/import_droidstumbler.py (copied) (copied from src/django_gheat/gheat/management/commands/import_csv.py ) (5 diffs)

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

@@ -1 +1 @@
 #!/usr/bin/env python
+# -*- coding: utf-8 -*-
 #
 ###########################################
 #
-# Script for importing .csv files
+# Script for importing DroidStumbler .csv files
 #
 # In theory, only the -f option is needed, but for overview's sake, please use the others aswell.
@@ -12 +13 @@
 #
 # (Run from project root)
-# ./manage.py import_csv -f <file location> -m <dataset name> -g <username> -e <email>
+# ./manage.py import_droidstumbler -f <file location> -m <dataset name> -g <username> -e <email>
 #
 # Make sure the variables in this script match the column numbers in your file e.g.;
@@ -33 +34 @@
 import csv
 
-def import_file(location, meetrondje, gebruiker, email):
+def import_droidstumbler(location, meetrondje, gebruiker, email):
 
   g, created = Gebruiker.objects.get_or_create(naam=gebruiker , email=email)
@@ -39 +40 @@
   mr = MeetRondje.objects.create(datum=datetime.datetime.now() , naam=meetrondje , gebruiker=g , apparatuur=a)
 
-  csvfile = csv.reader(open(location, 'rb'), delimiter='\t')
+  csvfile = csv.reader(open(location, 'rb'), delimiter=',')
   for row in csvfile:
-    lat = row[0].replace('N ', '')
-    lon = row[1].replace('E ', '')
-    ssid = row[2].strip('( )')
-    bssid = row[4].strip('( )')
-    enc = row[8]
-    if enc[2] == '1': enc = True
-    else: enc = False
-    sig = 100 # Get's fix soon
-    print lat, lon, ssid, bssid, enc, sig
-
-    ap, created = Accespoint.objects.get_or_create(mac=bssid, ssid=ssid, encryptie=enc)
-    m = Meting.objects.create(meetrondje=mr, accespoint=ap, latitude=lat, longitude=lon, signaal=sig)
-
+    epoch, msg_type, lat, lon, accuracy, ssid, bssid, level, frequency, capabilities, empty = row
+    if msg_type == "data" and lat and lon:
+      ap, created = Accespoint.objects.get_or_create(mac=bssid, ssid="ap-WirelessLeiden-" + ssid, encryptie=capabilities)
+      if created:
+        print "# INFO: New AccessPoint %s (%s)" % (ap.ssid, ap.mac)
+      m = Meting.objects.create(meetrondje=mr, accespoint=ap, latitude=lat, longitude=lon, signaal=(100 + int(level)))
+      print "# INFO: New Meeting: %s" % m
 
 class Command(BaseCommand):
@@ -64 +59 @@
 
   def handle(self, *args, **options):
-    import_file(options['location'],options['meetrondje'],options['gebruiker'],options['email'])
+    import_droidstumbler(options['location'],options['meetrondje'],options['gebruiker'],options['email'])