使用 Python 将 CSV 输出导出并映射到 MySQL 表

Question

我有多个客户端到单个服务器的双向 iperf 设置，用于网络监控。 iperf 服务器运行良好，并根据客户端编写的 cron 作业以 CSV 格式显示输出。

我希望编写一个 python 脚本来自动化将这些 CSV 输出映射到 MySQL 数据库的过程；反过来，它会定期更新和保存，无需人工干预。

我使用 Ubuntu 13.10 机器作为 iperf 服务器。以下是我得到的示例 CSV 输出。这不会存储到文件中，只是显示在屏幕上。

s1:~$ iperf -s -y C  
20140422105054,172.16.10.76,41065,172.16.10.65,5001,6,0.0-20.0,73138176,29215083
20140422105054,172.16.10.76,5001,172.16.10.65,56254,4,0.0-20.0,46350336,18502933
20140422105100,172.16.10.76,54550,172.16.10.50,5001,8,0.0-20.0,67895296,27129408
20140422105100,172.16.10.76,5001,172.16.10.50,58447,5,0.0-20.1,50937856,20292796
20140422105553,172.16.10.76,5001,172.16.10.65,47382,7,0.0-20.1,51118080,20358083
20140422105553,172.16.10.76,41067,172.16.10.65,5001,5,0.0-20.1,76677120,30524007
20140422105600,172.16.10.76,5001,172.16.10.50,40734,4,0.0-20.0,57606144,23001066
20140422105600,172.16.10.76,54552,172.16.10.50,5001,8,0.0-20.0,70123520,28019115
20140422110053,172.16.10.76,41070,172.16.10.65,5001,5,0.0-20.1,63438848,25284066
20140422110053,172.16.10.76,5001,172.16.10.65,46462,6,0.0-20.1,11321344,4497094

我想将它们映射到的字段是：timestamp、server_ip、server_port、client_ip、client_port、tag_id、interval、transferred、bandwidth

我想定期将此 CSV 输出映射到 MySQL 数据库，为此我明白我必须编写 Python 脚本（在 cron 作业中）查询并存储在 MySQL 数据库中。我是 Python 脚本和数据库查询的初学者。

我在 [https://serverfault.com/questions/566737/iperf-csv-output-format]; 进行了另一个关于服务器故障的讨论；并希望以此为基础构建我的查询。

Answer 1

生成 SQL 脚本，然后运行它

如果您不想使用像sqlalchemy 这样的复杂解决方案，可以使用以下方法。

1. 拥有您的 csv 数据，将它们转换为 SQL 脚本
2. 使用mysql命令行工具运行此脚本

在你第一次做之前，确保你在数据库中创建了需要的数据库结构（这个我留给你）。

我的以下示例使用（只是为了方便）包docopt，因此您需要安装它：

$ pip install docopt

CSV 到 SQL 脚本转换实用程序

csv2sql.py:

"""
Usage:
    csv2sql.py [--table <tablename>] <csvfile>

Options:
    --table <tablename>  Name of table in database to import into [default: mytable]

Convert csv file with iperf data into sql script for importing
those data into MySQL database.
"""
from csv import DictReader
from docopt import docopt

if __name__ == "__main__":
    args = docopt(__doc__)
    fname = args["<csvfile>"]
    tablename = args["--table"]
    headers = ["timestamp",
               "server_ip",
               "server_port",
               "client_ip",
               "client_port",
               "tag_id",
               "interval",
               "transferred",
               "bandwidth"
              ]
    sql = """insert into {tablename}
    values ({timestamp},"{server_ip}",{server_port},"{client_ip}",{client_port},{tag_id},"{interval}",{transferred},{bandwidth});"""
    with open(fname) as f:
        reader = DictReader(f, headers, delimiter=",")
        for rec in reader:
            print(sql.format(tablename=tablename, **rec)) # python <= 2.6 will fail here

将 CSV 转换为 SQL 脚本

首先介绍一下转换工具：

$ python csv2sql.py -h
Usage:
    csv2sql.py [--table <tablename>] <csvfile>

Options:
    --table <tablename>  Name of table in database to import into [default: mytable]

Convert csv file with iperf data into sql script for importing
those data into MySQL database.

将您的数据保存在文件data.csv：

$ python csv2sql.py data.csv 
insert into mytable
    values (20140422105054,"172.16.10.76",41065,"172.16.10.65",5001,6,"0.0-20.0",73138176,29215083);
insert into mytable
    values (20140422105054,"172.16.10.76",5001,"172.16.10.65",56254,4,"0.0-20.0",46350336,18502933);
insert into mytable
    values (20140422105100,"172.16.10.76",54550,"172.16.10.50",5001,8,"0.0-20.0",67895296,27129408);
insert into mytable
    values (20140422105100,"172.16.10.76",5001,"172.16.10.50",58447,5,"0.0-20.1",50937856,20292796);
insert into mytable
    values (20140422105553,"172.16.10.76",5001,"172.16.10.65",47382,7,"0.0-20.1",51118080,20358083);
insert into mytable
    values (20140422105553,"172.16.10.76",41067,"172.16.10.65",5001,5,"0.0-20.1",76677120,30524007);
insert into mytable
    values (20140422105600,"172.16.10.76",5001,"172.16.10.50",40734,4,"0.0-20.0",57606144,23001066);
insert into mytable
    values (20140422105600,"172.16.10.76",54552,"172.16.10.50",5001,8,"0.0-20.0",70123520,28019115);
insert into mytable
    values (20140422110053,"172.16.10.76",41070,"172.16.10.65",5001,5,"0.0-20.1",63438848,25284066);
insert into mytable
    values (20140422110053,"172.16.10.76",5001,"172.16.10.65",46462,6,"0.0-20.1",11321344,4497094);

全部放入文件data.sql:

$ python csv2sql.py data.csv > data.sql

将data.sql 应用于您的 MySQL 数据库

最后使用mysql命令（MySQL提供）导入数据库：

$ myslq --user username --password password db_name < data.sql 

Answer 2

这可能不是您正在寻找的答案，但是如果您通过在线学习基础教程来学习一点 sqlite3（一个原生 Python 模块 - “import sqlite3”），您将意识到你的问题并不难解决。然后只需使用标准计时器，例如 time.sleep() 来重复该过程。

Answer 3

如果您打算使用 Python，那么我建议您使用 sqlalchemy

一般做法是：

• 定义类，包含所有要存储的属性
• 将类的所有属性映射到数据库列和类型
• 从 csv 读取您的数据（使用例如csv 模块），为每一行创建相应的对象作为之前准备的类，并将其存储。

sqlalchemy 会为您提供更多细节和说明，您的要求似乎很简单。

另一种选择是找出现有的 csv 导入工具，其中一些已经在 MySQL 中可用，还有很多其他的。