vcs.maemo.org Git - wifihood/blob - wifiscanner/wifimap/scanner.py

   1
   2 try :
   3     import osso
   4 except :
   5     import osso_wrapper as osso
   6
   7 import time
   8
   9 import config , db
  10 import gps , wifiscan
  11
  12 import gobject
  13
  14 import os
  15
  16 class Scanner ( gps.GPSObject , wifiscan.WifiScanner ) :
  17
  18     def __init__ ( self , ifname="wlan0" ) :
  19         gps.GPSObject.__init__( self )
  20         wifiscan.WifiScanner.__init__( self , ifname )
  21         conf = config.Configuration( 'scanner' )
  22         self.homedir = conf.homedir
  23         self.db = db.database( os.path.join( self.homedir , conf.dbname ) )
  24
  25         self.write = False
  26
  27         # Values to be set by wireless scans
  28         self.newap = 0
  29         self.newaps = False
  30         self.aps = {}
  31         self.oldpos = {}
  32
  33     def start ( self , timeout=5000 , writelog=False ) :
  34         gps.GPSObject.start( self )
  35         wifiscan.WifiScanner.start( self , timeout )
  36         self.db.open()
  37         self.write = writelog
  38
  39     def stop ( self ) :
  40         gps.GPSObject.stop( self )
  41         wifiscan.WifiScanner.stop( self )
  42         self.db.close()
  43         self.write = False
  44
  45     def scan ( self ) :
  46         wifiscan.WifiScanner.scan( self )
  47         newap = 0
  48         self.newaps = False
  49         for mac,max_rss in self.scanlist.iteritems() :
  50             stored = self.db.db.execute( "SELECT rss, n, lat, lon FROM ap WHERE mac='%s'" % mac ).fetchone()
  51             if stored :
  52                 if not self.aps.has_key( mac ) :
  53                     self.aps[ mac ] = list(stored[1:])
  54                 if stored[0] > max_rss :
  55                     max_rss = stored[0]
  56                 self.db.update( mac , max_rss , self.tstamp , self.info[4:] )
  57             else :
  58                 newap += 1
  59                 if not self.aps.has_key( mac ) :
  60                     self.aps[ mac ] = [ 0 , 0 , 0 ]
  61                 self.db.add( mac , max_rss , self.tstamp , self.info[4:] )
  62             if self.aps[mac][0] != 0 :
  63                 self.oldpos[mac] = self.aps[mac][1] / self.aps[mac][0] , self.aps[mac][2] / self.aps[mac][0]
  64             self.aps[mac][0] += 1
  65             self.aps[mac][1] += self.info[4]
  66             self.aps[mac][2] += self.info[5]
  67         if newap :
  68             self.newap += newap
  69             self.newaps = True
  70         if self.write :
  71             self.write_logs()
  72
  73     def write_logs ( self ) :
  74             fd = open( os.path.join( self.homedir , "wiscan_gui.info" ) , 'a' )
  75             fd.write( "%s %s %s\n" % ( self.tstamp , " ".join(self.info) , " ".join(self.scanlist) ) )
  76             fd.close()
  77             if self.satellites :
  78                 loclist = open( os.path.join( self.homedir , "location.info" ) , 'a' )
  79                 loclist.write ( "%s\n" % ( self.satellites ,) )
  80                 loclist.close()
  81             if self.cells :
  82                 celllist = open( os.path.join( self.homedir , "cell.info" ) , 'a' )
  83                 celllist.write ( "%s\n" % ( self.cells ,) )
  84                 celllist.close()
  85
  86     def report ( self ) :
  87         # BUG : if report is called after close, db.nrows() will produce an exception
  88         return "%s\t%s\t%d ap\t%d total ap" % ( gps.GPSObject.report(self) , wifiscan.WifiScanner.report(self) , self.newap , self.db.nrows() )
  89
  90
  91 gobject.type_register(Scanner)
  92
  93 if __name__ == "__main__" :
  94     loop = gobject.MainLoop()
  95     sample = Scanner()
  96     def on_stop(control, mainloop):
  97         mainloop.quit()
  98     sample.control.connect("gpsd-stopped", on_stop, loop)
  99     def show_scan(wifiscanner):
 100         gobject.timeout_add( 5000 , show_scan , sample )
 101         print "scan results %s" % wifiscanner.report()
 102         print "  tstamp %s" % wifiscanner.tstamp
 103         c = 0
 104         for k,v in wifiscanner.scanlist.iteritems() :
 105            c += 1
 106            print "    %s %s" % ( k , v )
 107            if c > 5 :
 108                print "    ..."
 109                break
 110         print
 111     sample.start()
 112     sample.scan()
 113     gobject.timeout_add( 5100 , show_scan , sample )
 114     loop.run()
 115     sample.stop()
 116