在luigi Python中是否可以将数据库行用作输出

时间:2016-05-05 03:25:14

标签: python luigi

我正在学习luigi并且看到输出主要是文件系统上的文件。

如果该文件存在,那么luigi认为该任务已经完成。

在我的情况下,而不是写文件我想在postgres中更新状态为DONE的数据库erecord。

我想知道在luigi有可能吗

1 个答案:

答案 0 :(得分:2)

看看Luigi contribs。您可以对以下至少3个软件包感兴趣:mssqldbmysqldbrdbmssqlaredshift。如果您打开其中任何一个,请查找luigi.Target子类。

例如,在luigi.contribs.mysqldb中您有MySqlTarget。它为您提供了与您上面谈到的功能类似的功能:任务检查"标记表"存在(如果不存在,此任务将创建它),然后进行选择以检查任务是否已成功执行。如果没有,它会插入"标记"完成任务说完了。

如果您已经有一个"标记表"您可以通过复制其代码并自定义您需要或不需要的部分来子类化此任务类或创建类似的任务类。