如何通过Alembic（从CSV文件）导入数据？

Question

我尝试设计数据库架构并使用Alembic（如果重要的话，请使用virtualenv）。我用testdata制作了一些CSV文件。目前我通过以下方式将它们复制到交互式postgresql shell的db中：

/copy table_name FROM '~/path/to/file.csv' WITH CSV;

我想自动执行此操作，因此在通过Alembic进行分级升级时，不必手动复制每个表。我尝试了以下方法：

在方法的我的Alembic版本文件中upgrade()，在表的生成下方，添加了：

op.execute("copy table_name FROM '~/path/to/file.csv' WITH CSV;", execution_options=None)

但是它永远找不到文件。这是错误：

File "/Users/me/Workspace/project/venv/lib/python3.4/site-packages/SQLAlchemy-0.9.4-py3.4-macosx-10.6-intel.egg/sqlalchemy/engine/default.py", line 435, in do_execute
    cursor.execute(statement, parameters)
sqlalchemy.exc.OperationalError: (OperationalError) could not open file "~/Workspace/project/path/to/file.csv" for reading: No such file or directory
 "copy table_name FROM '~/Workspace/project/path/to/file.csv' WITH CSV;" {}

我究竟做错了什么？

我尝试运行只能使用sqlalchemy命令的postgresql命令是我的问题吗？如果是这样，我将如何通过sqlalchemy做到这一点？

我知道alembic中的批量导入选项，但我不希望重新格式化所有内容。

还有其他选项可以自动执行复制操作吗？有更好的工作流程吗？

Answer 1

Postgres 不会以“您”的身份运行，因此主目录会有所不同。在 postgres 中输入文件路径时，请始终使用文件的完整路径。如果将此作为 Alembic 升级的一部分，您应该执行以下操作：

import os
dirname = os.path.dirname(__file__)
filename = os.path.join(dirname, 'relative/path/to/file/you/want')
op.execute(f"copy table_name FROM '{file_name}' WITH CSV;", execution_options=None)