从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 (将#修改为@）