Python如何解析SQL语句并转换为Python对象，SQLParse类库的使用

Posted 2023-03-16

SQLParse是一个Python类库，可以用于解析SQL语句并将其转换为Python对象。该库支持多种类型的SQL语句，包括SELECT，INSERT，UPDATE和DELETE等。

本教程将介绍SQLParse的基础知识，包括如何安装、如何使用、如何解析和转换SQL语句等。让我们开始学习！

安装

要使用SQLParse库，首先需要安装它。你可以使用以下命令来安装它：

pip install sqlparse

使用

一旦你安装了SQLParse库，你就可以开始使用它来解析和转换SQL语句了。

以下是一个简单的例子，演示了如何使用SQLParse来解析一个SQL查询语句并打印出其中的所有列：

import sqlparse

query = SELECT column1, column2 FROM table1
parsed = sqlparse.parse(query)[0]

for token in parsed.tokens:
    if isinstance(token, sqlparse.sql.IdentifierList):
        for identifier in token.get_identifiers():
            print(identifier)

这个例子中，我们首先导入了sqlparse类库，然后定义了一个包含SQL查询语句的字符串。接着，我们使用sqlparse.parse()方法来解析该查询语句并将其转换为Python对象。这个方法返回一个对象列表，因为在某些情况下可能会有多个查询语句。

我们使用列表中的第一个元素，因为我们只有一个查询语句。然后我们遍历了该查询语句的所有标记，并打印出了标记中的所有列。

解析和转换SQL语句

SQLParse库不仅可以解析SQL语句，还可以将其转换为Python对象，以便于进一步处理。以下是一些常用的方法：

• sqlparse.parse(sql)：将SQL语句解析为Python对象列表。
• str(parsed)：将Python对象转换为SQL语句字符串。
• parsed.tokens：返回一个包含查询语句中所有标记的列表。
• token.ttype：返回标记的类型，例如标识符、关键字或字符串。
• token.value：返回标记的值，例如列名或表名。
• token.parent：返回标记的父节点，例如SELECT语句中的列。
• token.tokens：返回标记的子节点，例如SELECT语句中的列名。

以下是一个更详细的例子，演示了如何使用SQLParse库解析和转换一个SQL查询语句：

import sqlparse

# Define a SQL query
query = 
SELECT column1, column2 
FROM table1
WHERE column1 = value


# Parse the query
parsed = sqlparse.parse(query)[0]

# Print the parsed object
print(parsed)

# Print the string representation of the parsed object
print(str(parsed))

# Loop through the tokens and print each one
for token in parsed.tokens:
    print(token.ttype, token.value)

# Loop through the WHERE clause tokens and print each one
where_clause = None
for token in parsed.tokens:
  if isinstance(token, sqlparse.sql.Where):
    where_clause = token

if where_clause:
  for token in where_clause.tokens:
    print(token.ttype, token.value)

在这个例子中，我们定义了一个包含SQL查询语句的多行字符串。然后，我们使用`sqlparse.parse()`方法将该字符串解析为Python对象。我们打印出这个Python对象，以便看到解析的结果。

接着，我们使用`str(parsed)`方法将Python对象转换为SQL查询语句字符串，并打印出来。这个查询语句字符串应该与我们定义的多行字符串是一样的。

然后，我们遍历了解析后的对象中的所有标记，并打印出了每个标记的类型和值。最后，我们找到了WHERE子句，并打印出了其中的所有标记。

总结

SQLParse是一个非常有用的Python类库，可以用于解析和转换SQL语句。在本教程中，我们介绍了如何安装SQLParse，如何使用SQLParse来解析和转换SQL语句，以及一些常用的方法。希望这个教程对你有所帮助，让你更好地了解SQLParse的用法。

python操作excel转换为sql语句

python操作excel转换为sql语句

使用pandas 库。

打开excel取出对应的列。

然后转换为多维list

然后利用format格式转换sql语句。

最后写入文件即可。

import pandas as pd


def excel_one_line_to_list():
    df = pd.read_excel("402序列.xlsx", usecols=[1, 2],
                       names=None)  # 读取项目名称列,不要列名
    df_li = df.values.tolist()
    # print(df_li)
    # update team set room = xxx where teamid = yyyy;
    ans_lst = []
    for i in df_li:
        sql = "update team set room=\\'1\\' where teamid=0;".format(i[0][4:], i[1])
        ans_lst.append(sql)
    print(ans_lst)
    f = open("test.sql", "w")
    for i in ans_lst:
        f.write(i + '\\n')
    df = pd.read_excel("zw_301.xlsx", usecols=[1, 3],
                       names=None)  # 读取项目名称列,不要列名
    df_li = df.values.tolist()
    ans_lst = []
    for i in df_li:
        sql = "update team set room =  \\'1\\' where teamid = 0;".format(i[0][4:], i[1])
        ans_lst.append(sql)
    for i in ans_lst:
        f.write(i + '\\n')
    f.close()


if __name__ == '__main__':
    excel_one_line_to_list()

另外：linux 执行sql脚本的命令是：

source /xxx/xxx/xxx.sql