source: src/django_gheat/gheat/management/commands/import_kismet.py@ 9158

#!/usr/bin/env python
#
# Script for importing .gpsxml and .netxml files (Kismet output)
#

from django.core.management.base import BaseCommand,CommandError
from optparse import OptionParser, make_option
from gheat.models import *
from lxml import etree
import datetime
import os
import sys

def import_file(gpsxml_file, netxml_file, meetrondje, gebruiker, email):
  gpsxml_doc = etree.parse(gpsxml_file)
  netxml_doc = etree.parse(netxml_file)

  points = gpsxml_doc.findall('gps-point')
  wnetworks = netxml_doc.findall('wireless-network')

  # TODO: Source source is variable entitity, based on mesurement
  kaart = 'deadcode'
  gebruiker, created = Gebruiker.objects.get_or_create(naam=gebruiker , email=email)
  apparatuur, created = Apparatuur.objects.get_or_create(antenne='test' , kaart=kaart)
  # TODO: Date is set to import date, but should pick the date from the netxml file
  mr = MeetRondje.objects.create(datum=datetime.datetime.now(),
    naam=meetrondje , gebruiker=gebruiker , apparatuur=apparatuur)

  # Create all accesspoints and for caching validation purposes store them
  # locally as well
  ap_cache = {}
  ap_ignore = []
  print "#INFO: Going to import %s accesspoints" % len(wnetworks)
  for wnetwork in wnetworks:
    bssid = wnetwork.find('BSSID').text
    # Only store access points
    if wnetwork.attrib['type'] != "infrastructure":
      ap_ignore.append(bssid)
      continue

    enc = (wnetwork.find('SSID/encryption') != None)
    ssid_node = wnetwork.find('SSID/essid[@cloaked="false"]')
    ssid = ssid_node.text if ssid_node != None else 'hidden'

    ap, created = Accespoint.objects.get_or_create(mac=bssid, ssid=ssid, encryptie=enc)
    ap_cache[bssid] = ap

  count = 0
  #XXX: This is not effient at all, try to wrap it into a a bulk insert would
  # be much more effient as for example: http://djangosnippets.org/snippets/2362/
  print "#INFO: Going to import %s points" % len(points)
  for point in points:
    # XXX: Filter this in the beginning with XPath, but etree does not support that (yet).
    if point.attrib['bssid'] in ['GP:SD:TR:AC:KL:OG','00:00:00:00:00:00']:
      continue
    if point.attrib['bssid'] in ap_ignore:
      continue
    ap = ap_cache[point.attrib['bssid']]

    # XXX: Signal need properly be a relation of signal_dbm and noice_dbm
    signaal = 100 + int(point.attrib['signal_dbm'])

    meting,created = Meting.objects.get_or_create(meetrondje=mr, accespoint=ap,
      latitude=point.attrib['lat'], longitude=point.attrib['lon'],
      signaal=signaal)
    # Give some feedback to the user
    if created:
      count += 1
      if (count % 1000) == 0:
        print count,
      elif (count % 100) == 0:
        print ".",

class Command(BaseCommand):
  args = '<gpsxml> <netxml>'
  option_list = BaseCommand.option_list + (
    make_option('-m', '--meetrondje', dest='meetrondje', default='rondje',help='Naam van het meetrondje'),
    make_option('-g', '--gebruiker', dest='gebruiker', default='username',help='Naam van de persoon die de meting uitgevoerd heeft'),
    make_option('-e', '--email', dest='email', default='foo@bar.org',help='Email van de persoon die de meting uitgevoerd heeft'),
    )

  # TODO: Condider netxml file to be the gpsxml file with diffent suffix.
  def handle(self, *args, **options):
    try:
      (gpsxml_file, netxml_file) = args
    except ValueError:
      self.print_help(sys.argv[0],sys.argv[1])
      raise CommandError("Not all arguments are provided")
    if not os.path.isfile(gpsxml_file):
      raise CommandError("gpsxml file '%s' does not exists" % gpsxml_file)
    if not os.path.isfile(netxml_file):
      raise CommandError("netxml file '%s' does not exists" % netxml_file)

    import_file(gpsxml_file, netxml_file ,options['meetrondje'],options['gebruiker'],options['email'])