具有脚本语言的动态数据库

Question

我有一组要处理的.csv文件。使用SQL查询来处理它会容易得多。我想知道是否有某种方式可以加载.csv文件，并使用SQL语言通过python或ruby之类的脚本语言对其进行查询。用类似于ActiveRecord的东西加载它会很棒。

问题是我不想在运行脚本之前在某个地方运行数据库。在脚本语言和某些模块之外，我不需要其他安装。

我的问题是该任务应使用哪种语言以及使用哪些模块。我环顾四周，找不到适合我的任何东西。可能吗？

Answer 1

有sqlite3，包含在python中。使用它，您可以创建一个数据库（
在内存上 ）并向其中添加行，并执行SQL查询。

如果您想要类似ActiveRecord的功能，则应该添加一个外部ORM，例如sqlalchemy。那是一个单独的下载

使用sqlalchemy的快速示例：

from sqlalchemy import create_engine, Column, String, Integer, MetaData, Table
from sqlalchemy.orm import mapper, create_session
import csv
CSV_FILE = 'foo.csv'
engine = create_engine('sqlite://') # memory-only database

table = None
metadata = MetaData(bind=engine)
with open(CSV_FILE) as f:
    # assume first line is header
    cf = csv.DictReader(f, delimiter=',')
    for row in cf:
        if table is None:
            # create the table
            table = Table('foo', metadata, 
                Column('id', Integer, primary_key=True),
                *(Column(rowname, String()) for rowname in row.keys()))
            table.create()
        # insert data into the table
        table.insert().values(**row).execute()

class CsvTable(object): pass
mapper(CsvTable, table)
session = create_session(bind=engine, autocommit=False, autoflush=True)

现在，您可以查询数据库，按任何字段进行过滤等。

假设您在此csv上运行以上代码：

name,age,nickname
nosklo,32,nosklo
Afila Tun,32,afilatun
Foo Bar,33,baz

这将创建并与领域内存填充一个表name，age，nickname。然后，您可以查询表：

for r in session.query(CsvTable).filter(CsvTable.age == '32'):
    print r.name, r.age, r.nickname

这将自动创建并运行SELECT查询并返回正确的行。

使用sqlalchemy的另一个优点是，如果您决定将来使用另一个功能更强大的数据库，则可以在不更改代码的情况下实际使用。