source: src/django_gheat/gheat/management/commands/dataimport.py@ 9369

#!/usr/bin/env python
#
###########################################
#
# Script for importing .netxml files (Kismet output)
#
# In theory, only the -f option is needed, but for overview's sake, please use the others aswell.
# -f = location of the .netxml, e.g. '/home/test.netxml'
# -m = name of the dataset, e.g. 'Walk in park' or 'Trip with boat'
# -g = your name
# -e = your email address
#
# (Run from project root)
# ./manage.py dataimport -f <file location> -m <dataset name> -g <username> -e <email>
#
# Make sure your file has atleast the following elements:
#
# <wireless-network>
#   <SSID>
#     <encryption> </encryption>
#     <essid cloaked=""> </essid>
#   </SSID>
#   <BSSID> </BSSID>
#   <gps-info>
#     <min-lat> </min-lat>
#     <min-lon> </min-lon>
#   </gps-info>
# </wireless-network>
#
#
# Dennis Wagenaar
# d.wagenaar@gmail.com
#
###########################################

from django.core.management import setup_environ
from django.core.management.base import BaseCommand
from optparse import OptionParser, make_option
import settings
setup_environ(settings)
from gheat.models import *
from lxml import etree
import datetime

def import_file(location, meetrondje, gebruiker, email):

  datasource = etree.parse(location)
  doc = datasource.findall('wireless-network')
  kaart = datasource.find('card-source/card-source')
  kaart = kaart.text
  g, created = Gebruiker.objects.get_or_create(naam=gebruiker , email=email)
  a, created = Apparatuur.objects.get_or_create(antenne='test' , kaart=kaart)
  mr = MeetRondje.objects.create(datum=datetime.datetime.now() , naam=meetrondje , gebruiker=g , apparatuur=a)

  for wnetwork in doc:
    enc = wnetwork.find('SSID/encryption')
    if enc != None: enc = True
    else: enc = False
    ssid = wnetwork.find('SSID/essid')
    if ssid != None: ssid = ssid.text
    else: ssid = 'hidden'
    bssid = wnetwork.find('BSSID')
    if bssid != None: bssid = bssid.text
    min_sig = wnetwork.find('snr-info/min_signal_dbm')
    if min_sig != None: min_sig = int(min_sig.text)
    else: min_sig = (-99)
    max_sig = wnetwork.find('snr-info/max_signal_dbm')
    if max_sig != None: max_sig = int(max_sig.text)
    else: max_sig= (-99)
    lat = wnetwork.find('gps-info/min-lat')
    if lat != None: lat = lat.text
    lon = wnetwork.find('gps-info/min-lon')
    if lon != None: lon = lon.text
    if ssid is None: continue
    
    avg_sig = (min_sig+max_sig)/2
    sig = (1.67*avg_sig)+166.67
    if sig > 100: sig = 100
    if sig < 0: sig = 0

    print enc, ssid, bssid, lat, lon, 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)


class Command(BaseCommand):
  option_list = BaseCommand.option_list + (
    make_option('-f', '--location', dest='location', default='location'),
    make_option('-m', '--meetrondje', dest='meetrondje', default='rondje'),
    make_option('-g', '--gebruiker', dest='gebruiker', default='username'),
    make_option('-e', '--email', dest='email', default='foo@bar.org'),
    )

  def handle(self, *args, **options):
    import_file(options['location'],options['meetrondje'],options['gebruiker'],options['email'])