Add csv example.

This commit is contained in:
aouinizied 2020-05-06 20:04:33 +02:00
parent a2a2f1f7be
commit 6f1b1efe04

33
examples/csv_generator.py Normal file
View file

@ -0,0 +1,33 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
file: csv_generator.py
This file is part of nfstream.
Copyright (C) 2019-20 - Zied Aouini <aouinizied@gmail.com>
nfstream is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License
as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
nfstream is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty
of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with nfstream.
If not, see <http://www.gnu.org/licenses/>.
"""
from nfstream import NFStreamer
import sys
import datetime
path = sys.argv[1]
output_file_name = path + ".csv"
print("nfstream processing started. Use Ctrl+C to interrupt and save.")
start = datetime.datetime.now()
df = NFStreamer(source=path, statistics=True, idle_timeout=1).to_pandas()
end = datetime.datetime.now()
df.to_csv(output_file_name)
print("nfstream processed {} flows and saved them in file: {}".format(df.shape[0], output_file_name))
print("Processing time: {}".format(end - start))