从CSV文件读取，然后替换脚本中的值

Question

我正在使用一个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, g, o) VALUES ('John', 'sales-north', 'Joanna Smith');")
c.execute("INSERT INTO t (u, g, o) VALUES ('Mary', 'sales-south', 'Joanna Smith');")
c.execute("INSERT INTO t (u, g, o) VALUES ('Paul', 'sales-north', 'Joanna Smith');")
c.execute("INSERT INTO t (u, g, o) VALUES ('Stacy', 'it-north', 'Marcus Chen');")
c.execute("INSERT INTO t (u, g, o) VALUES ('Lisa', 'marketing-north', 'Casley Williams');")
c.execute("INSERT INTO t (u, g, o) VALUES ('David', 'sales-north', 'Joanna Smith');")
c.execute("INSERT INTO t (u, g, 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%';