如何解决从CSV文件读取,然后替换脚本中的值
我正在使用一个csv文件来查找变量以在python中构造SQL查询。我想在执行查询之前替换其中一个变量。
如果以后更改此变量,我只想在1行(在query.py中)而不是多行(在orgs.csv中)进行更改。
代码
initdb.py
#!/usr/bin/python3
import csv
import sqlite3
conn = sqlite3.connect('demo.db')
c = conn.cursor()
c.execute("CREATE TABLE t (u,g,o);")
c.execute("INSERT INTO t (u,o) VALUES ('John','sales-north','Joanna Smith');")
c.execute("INSERT INTO t (u,o) VALUES ('Mary','sales-south',o) VALUES ('Paul',o) VALUES ('Stacy','it-north','Marcus Chen');")
c.execute("INSERT INTO t (u,o) VALUES ('Lisa','marketing-north','Casley Williams');")
c.execute("INSERT INTO t (u,o) VALUES ('David',o) VALUES ('Craig','marketing-south','Casley Williams');")
conn.commit()
orgs.csv
# headers #
#group,org
#sales-north,"Joanna Smith"
#sales-south,"Joanna Smith"
#marketing-north,"Casley Williams"
#marketing-south,"Casley Williams"
#it-north,"Marcus Chen"
#it-south,"Marcus Chen"
sales-north,sales
sales-south,sales
marketing-north,marketing
marketing-south,marketing
it-north,it
it-south,it
query.py
#!/usr/bin/python3
import csv
import sqlite3
conn = sqlite3.connect('demo.db')
c = conn.cursor()
def decomment(csvfile):
for row in csvfile:
raw = row.split('#')[0].strip()
if raw: yield raw
sales = "Joanna Smith"
marketing = "Casley Williams"
it = "Marcus Chen"
with open('orgs.csv') as csvfile:
reader = csv.reader(decomment(csvfile))
for row in reader:
query = ('SELECT * FROM t WHERE g LIKE \'%' + row[0] + '%\' AND o LIKE \'%' + row[1] + '%\';')
group = row[0]
print(query)
c.execute(query)
rows = c.fetchall()
#print(rows)
必需的输出
SELECT * FROM t WHERE g LIKE '%sales-north%' AND o LIKE '%Joanna Smith%';
SELECT * FROM t WHERE g LIKE '%sales-south%' AND o LIKE '%Joanna Smith%';
SELECT * FROM t WHERE g LIKE '%marketing-north%' AND o LIKE '%Casley Williams%';
SELECT * FROM t WHERE g LIKE '%marketing-south%' AND o LIKE '%Casley Williams%';
SELECT * FROM t WHERE g LIKE '%it-north%' AND o LIKE '%Marcus Chen%';
SELECT * FROM t WHERE g LIKE '%it-south%' AND o LIKE '%Marcus Chen%';
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com(将#修改为@)