使用Python：如何从sql查询中获取表名并在存在模式的表模式之前添加单词

Question

I have a json file that I am using as a dictionary in Python. The json file is very large. I am trying to write a python code to update each "query" by adding a "source." before the table schema. then use the updated dictionary for other programming purposes.

The SQL scripts could have joins, cartesian joins, subqueries, etc.

Expected output:

 "query": "SELECT a.column1, b.column2
 FROM source.abcd.hist a, source.efgh.present b
 WHERE (select column3, column4 from UPS where a.id = b.id )"

 "query": "SELECT a.column1, b.column2
 FROM source.apple.hist a, source.mango.present b
 WHERE (select column3, column4 from source.my.ORANGE where a.id = b.id

{"result":[{
"query": "SELECT a.column1, b.column2
FROM abcd.hist a, efgh.present b
WHERE (select column3, column4 from UPS where a.id = b.id )"
},
{"query": "SELECT a.column1, b.column2
FROM apple.hist a, mango.present b
WHERE (select column3, column4 from my.ORANGE where a.id = b.id )"}
]}

Answer 1

Your result is a dictionary {}, with the first key 'result' containing a list [] of dictionaries {} where each dictionary has a key 'query' that goes to a value that looks like a SQL query.

Assuming your output object is called op you can get your desired result as follows:

for k in op['result']: # For each dictionary in result
    print(str(k)[1:-1]) # Cast to a string and strip curlies

EDIT:

def addSource(q):
    lines = q.split("\n")
    for n,k in enumerate(lines):
        if(k.startswith("FROM")):
            q[n] = k.replace("FROM ","FROM source.").replace(", ",", source.")
    return("\n".join(q))

and with that,

for n,k in enumerate(op['result']):
    op['result'][n]["query"] =  addSource(k["query"])