def test_integration_datasources_get_all(self):
here_dir = os.path.dirname(os.path.abspath(__file__))
test_db = os.path.join(here_dir, "../assets/testdata.kismet_4")
abstraction = kismetdb.DataSources(test_db)
all_sources = abstraction.get_all()
for source in all_sources:
assert isinstance(source["json"], type(""))
def test_integration_datasources_instantiate(self):
here_dir = os.path.dirname(os.path.abspath(__file__))
test_db = os.path.join(here_dir, "../assets/testdata.kismet_4")
abstraction = kismetdb.DataSources(test_db)
assert abstraction
def main():
parser = argparse.ArgumentParser(description="Kismet to CSV Log Converter")
parser.add_argument("--in",
action="store",
dest="infile",
help="Input (.kismet) file")
parser.add_argument("--out",
action="store",
dest="outfile",
help="Output CSV filename")
parser.add_argument("--table",
action="store",
dest="srctable",
help="Select the table to output")
results = parser.parse_args()
replacements = {}
if results.infile is None:
print("Expected --in [file]")
sys.exit(1)
if not os.path.isfile(results.infile):
print("Could not find input file \"{}\"".format(results.infile))
sys.exit(1)
if results.srctable is None:
results.srctable = "devices"
replacements["srctable"] = results.srctable
if results.srctable == "devices":
table_abstraction = kismetdb.Devices(results.infile)
column_names = [
"first_time", "last_time", "devkey", "phyname", "devmac",
"strongest_signal", "min_lat", "min_lon", "max_lat", "max_lon",
"avg_lat", "avg_lon", "bytes_data", "type"
]
elif results.srctable == "packets":
table_abstraction = kismetdb.Packets(results.infile)
column_names = [
"ts_sec", "ts_usec", "phyname", "sourcemac", "destmac", "transmac",
"frequency", "devkey", "lat", "lon", "packet_len", "signal",
"datasource", "dlt", "error"
]
elif results.srctable == "datasources":
table_abstraction = kismetdb.DataSources(results.infile)
column_names = [
"uuid", "typestring", "definition", "name", "interface"
]
elif results.srctable == "alerts":
table_abstraction = kismetdb.Alerts(results.infile)
column_names = [
"ts_sec", "ts_usec", "phyname", "devmac", "lat", "lon", "header"
]
else:
print("Invalid table entered, please retry with either devices, "
"packets, datasources or alerts.")
sys.exit(1)
if results.outfile is None:
results.outfile = "{}-{}.csv".format(results.infile,
replacements["srctable"])
csv_file_mode = "wb" if sys.version_info[0] < 3 else "w"
with open(results.outfile, csv_file_mode) as csvfile:
csvWriter = csv.DictWriter(csvfile,
delimiter="\t",
extrasaction="ignore",
fieldnames=column_names)
nrows = 0
csvWriter.writeheader()
for row in table_abstraction.yield_meta():
csvWriter.writerow(row)
nrows = nrows + 1
if nrows % 1000 == 0:
print("Wrote {} rows".format(nrows))