在psql中执行.sql文件，从Excel单元格值传递参数

Question

我试图在psql中执行一个.sql文件（通过.bat脚本），将test.xlsm工作簿的单元格B1和B2中包含的值作为参数传递。 我一直在寻找一种可能的解决方案，但我不明白如何在.sql文件中引用这些值（Excel中的单元格B1和B2）。

到目前为止，这是我一直在努力的工作。

download_from_postgresql.bat文件包含以下脚本，以便建立连接并通过psql.exe运行上述的download_view.sql文件：

@echo off
setlocal
set PGPASSWORD=mypassword
"D:\PostgreSQL\10\bin\psql.exe" -h myserver.com -U myuser -d mydb -f C:/QA_nepal/download_view.sql
endlocal

download_view.sql文件包含以下脚本，以便导出直接在我连接到的PostgreSQL服务器上创建的特定视图的.csv文件：

\copy (SELECT * FROM public_qa_stfc_india."join_tables_date_filter=custom_user_filter=none") to 'C:\\QA_nepal\\test.csv' with CSV HEADER;
\q

我所指的join_tables_date_filter = custom_user_filter = none视图是直接在PostgreSQL服务器上创建的，看起来像这样：

SELECT * FROM public_qa_stfc_india.step_3_joined_tables
WHERE "left"(step_3_joined_tables._0_general_start, 10) >= '2018-05-03'::text
AND "left"(step_3_joined_tables._0_general_start, 10) <= '2018-05-04'::text;

我想要做的是编辑上面包含的视图定义，以便具有以下内容：

SELECT * FROM public_qa_stfc_india.step_3_joined_tables
WHERE "left"(step_3_joined_tables._0_general_start, 10) >= 'date_par_1'::text
AND "left"(step_3_joined_tables._0_general_start, 10) <= 'date_par_2'::text;

并将单元格B1的值传递给date_par_1，并将B2的值传递给date_par_2。 我要引用的Excel工作簿位于C：\\ QA_nepal \\ excel \\ test.xlsm中。

在这一点上，我不确定如何继续。 我应该指示psql从Excel读取这些值吗？ 还要在download_from_postgresql.bat中添加说明吗？ 还是应该以某种方式直接在download_view.sql文件中引用Excel中的值？

在此先感谢您的帮助，Stefano

更新2018/06/19

我设法在VB中编写了一个宏，该宏能够将单元格B1和B2中的值（分别为2018-01-01和2015-02-07）传递到.bat文件：

Sub batch()
Dim wsh As Object
Dim data_1 As String
Dim data_2 As String
Set wsh = VBA.CreateObject("WScript.Shell")
Dim waitOnReturn As Boolean: waitOnReturn = True
Dim windowStyle As Integer: windowStyle = 1

data_1 = ThisWorkbook.Sheets("Foglio1").Range("B1").Value
data_2 = ThisWorkbook.Sheets("Foglio1").Range("B2").Value

wsh.Run "C:\QA_nepal\download_from_postgresql.bat " & data_1 & " " & data_2, windowStyle, waitOnReturn

End Sub

现在，我的download_from_postgresql.bat文件如下所示：

@echo off
SET data_1=%1%
SET data_2=%2%
ECHO %data_1% 'test if the value in B1 is passed to the .bat file correctly
ECHO %data_2% 'test if the value in B2 is passed to the .bat file correctly
setlocal
set PGPASSWORD=mypassword
"D:\PostgreSQL\10\bin\psql.exe" -h myserver.com -U myuser -d mydb -v v1=%data_1% -v v2=%data_2% -f C:/QA_nepal/download_view.sql
endlocal

此时，我不确定如何在download_view.sql中引用变量v1和v2 ：

\copy (SELECT * FROM public_qa_stfc_india."join_tables_date_filter=custom_user_filter=none") to 'C:\\QA_nepal\\test.csv' with CSV HEADER;
\q

考虑到我的join_tables_date_filter = custom_user_filter = none视图看起来像这样：

SELECT * FROM public_qa_stfc_india.step_3_joined_tables
WHERE "left"(step_3_joined_tables._0_general_start, 10) >= '2018-05-03'::text
AND "left"(step_3_joined_tables._0_general_start, 10) <= '2018-05-04'::text;

我知道我必须用2个变量（比如说var1和var2）替换上面视图中的日期（“ 2018-05-03”和“ 2018-05-04”），然后我必须“连接”这些变量（var1和var2）以及.bat文件（v1和v2）中的变量，我以不同的方式尝试过，例如具有以下内容：

\copy (SELECT * FROM public_qa_stfc_india."join_tables_date_filter=custom_user_filter=none" WHERE var1 = (:v1) AND var2 = (:v2)) to 'C:\\QA_nepal\\test.csv' with CSV HEADER;
\q

和：

SELECT * FROM public_qa_stfc_india.step_3_joined_tables
WHERE "left"(step_3_joined_tables._0_general_start, 10) >= '$var1'::text
AND "left"(step_3_joined_tables._0_general_start, 10) <= '$var2'::text;

但没有成功。 一般来说，我实际上对SQL还是很陌生，所以我可能走错了路。 有什么建议么？

在此先感谢，Stefano

Answer 1

考虑使用功能来接收参数并以字符串格式运行COPY命令的功能来替换存储的视图，这些视图的硬编码日期和不接收参数。 然后让psql调用该函数。

CREATE OR REPLACE FUNCTION my_date_filter_func(start_dt TEXT, end_dt TEXT, csv_file TEXT)
    RETURNS void AS 
$func$
  BEGIN
    EXECUTE format('COPY (SELECT * FROM public_qa_stfc_india.step_3_joined_tables s
                          WHERE LEFT(s._0_general_start, 10) >= %L
                            AND LEFT(s._0_general_start, 10) <= %L) TO %L with CSV HEADER;', 
                   start_dt, end_dt, csv_file);
  END
$func$ LANGUAGE plpgsql;

通过VBA Shell调用psql命令

...
path = "C:/QA_nepal/test.csv"
cmd_str = "psql ... -c ""SELECT my_date_filter_func('" & data_1 & "', '" & data_2 & "', '" & path & "')"""

wsh.Run cmd_str, windowStyle, waitOnReturn

最后，目前还不清楚为什么要使用LEFT()将日期字段视为文本，因为您可以直接在WHERE子句中使用日期文字，并且将其时间戳隐式定义为午夜： 00:00:00 : 00:00:00 : 00:00:00 。

SELECT * 
FROM public_qa_stfc_india.step_3_joined_tables s
WHERE s._0_general_start >= '2018-05-03'
  AND s._0_general_start <= '2018-05-04';