def test_metadata(self): net = ParseNetxml(get_asset_dic() + "/network.netxml") meta = net.get_metadata() self.assertDictEqual(meta, { "kismet_version": "2013.03.R0", "start_time": "2014-11-15 13:00:00" })
def insert(self): print(json.dumps(self._files, indent=4)) for file in self._files: gpsxml = file + ".gpsxml" netxml = file + ".netxml" meta = {} gps = ParseGpsxml(file=gpsxml, date_format="%Y-%m-%dT%H:%M:%S.0") meta.update(gps.get_metadata()) elk_meta = self._elk.search(doc_type="files", size=1, q="file:\"%s\"" % (meta["file"],)) if elk_meta["hits"]["total"] > 0: print("File %s exists in elasticsearch!" % (netxml,)) continue try: net = ParseNetxml(file=netxml, date_format="%Y-%m-%dT%H:%M:%S.0") meta.update(net.get_metadata()) meta["time"] = meta["start_time"] del meta["start_time"] file_id = self._elk.insert("files", meta)["_id"] print("Inserting networks to database from " + netxml) net = InsertNetxmlElk(self._elk, netxml, file_id) net.insert() print("Inserting gpspoints to database from " + gpsxml) gps = InsertGpsxmlElk(self._elk, gpsxml, file_id) gps.insert() except Exception as e: print("Error :( " + str(e))