使用 Python 重新提取多行字符串的某些部分 [复制]

Question

我一直试图在多行文本中提取SQL查询，但总是得到错误的输出。

如何在一个或三个引号之间获得文本？

注意：在第一个完整引号"，"，""""""，''''''之前和之后都可以有任何东西，我只对在引号之间找到第一个文本感兴趣。

import re

cell_text = """\
#%%sql
q = \"\"\"
select 
name, breed, sum(weight) over (partition by breed order by name) as running_total_weight
from cats 
order by breed, name
\"\"\"

f(q)
"""
print(cell_text)

我的尝试：

pat = """.*select(.*)['"].*"""
out = re.findall(pat,cell_text,flags=re.M)[0]
sql = 'select ' + out
print(sql)

# I am getting empty outputs for re.findall instead of text there.

所需输出：

input
----

#%%sql
q = """
select 
name, breed, sum(weight) over (partition by breed order by name) as running_total_weight
from cats 
order by breed, name
"""

f(q)

output
------

select 
name, breed, sum(weight) over (partition by breed order by name) as running_total_weight
from cats 
order by breed, name


input
-----
#%%sql
q = "select * from cats;"

f(q)

output
-------
select * from cats;

input
-----
q = 'select * from cats limit 2'

output
------
select * from cats limit 2

Answer 1

您需要使用DOTALL或（？ s）模式，如下所示：

>>> print (re.findall(r'(?s)"""(.*?)"""', cell_text)[0])

select
name, breed, sum(weight) over (partition by breed order by name) as running_total_weight
from cats
order by breed, name

您还可以在re.findall中使用标志参数：

re.findall(r'"""(.*?)"""', cell_text, flags=re.DOTALL)

编辑：请注意，要匹配所有单引号或三引号文本，您可以交替使用此正则表达式：

r"""\"\"\"(.*?)\"\"\"|'''(.*?)'''|"(.*?)"|'(.*?)'"""

正则表达式演示