psycopg2 与 sys.stdin.read()

Question

我有如下的小代码：

#!/usr/bin/python
import psycopg2, sys

try:
    conn = psycopg2.connect("dbname='smdr' user='bino'");
except:
    print "I am unable to connect to the database"
cur = conn.cursor()
v_num = '1'  
cur.execute("SELECT * from genctr WHERE code = %(num)s", dict(num=v_num))
rows = cur.fetchall()
print "\nShow me the databases:\n"
ctrnum =0
for row in rows:
    print row[0]+","+row[1]

当我运行它时，我

bino@erp:~/mydoc/openerp/smdr$ ./genctr.py 

Show me the databases:

1,Bahamas
1,Barbados
1,Canada
1,Cayman Islands
1,United States
1,Virgin Islands U.S.

尝试将“v_num = '1'”替换为“v_num = sys.stdin.read()”

#!/usr/bin/python
import psycopg2, sys

try:
    conn = psycopg2.connect("dbname='smdr' user='bino'");
except:
    print "I am unable to connect to the database"
cur = conn.cursor()
#v_num = '1'
v_num = sys.stdin.read()
cur.execute("SELECT * from genctr WHERE code = %(num)s", dict(num=v_num))
rows = cur.fetchall()
print "\nShow me the databases:\n"
ctrnum =0
for row in rows:
    print row[0]+","+row[1]

但是当我运行它时，我只得到这个：

bino@erp:~/mydoc/openerp/smdr$ echo 1 |./genctr.py 

Show me the databases:

请给我你如何解决它的启示

真诚地

-bino-

Answer 1

echo 1 将为您的程序提供“1\n”（即“1”后面带有换行符）。 sys.stdin.read() 将返回确切的字符串，然后 psycopg2 将把 SQL 语句准备为 SELECT * from genctr WHERE code = '1\n'.结果将是没有匹配的结果，因此 for 循环内的代码将永远不会执行，这就是为什么您看不到任何额外的输出。

尝试使用 echo -n 1 来抑制换行符，或者使用 sys.stdin.read().strip() 来删除字符串中的任何前导和尾随空格。如果 code 字段是整数，则将 sys.stdin.read() 的结果也转换为 int 可能是一个好主意，如下所示：

int(sys.stdin.read().strip())