要使用directory.getfiles
仅过滤.sql
文件,您可以使用以下代码示例:
import os
def get_sql_files(directory):
sql_files = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.sql'):
sql_files.append(os.path.join(root, file))
return sql_files
directory = 'path/to/your/directory'
sql_files = get_sql_files(directory)
print(sql_files)
这段代码首先导入os
模块,然后定义一个名为get_sql_files
的函数,该函数接受一个目录参数。在函数中,我们使用os.walk()
方法遍历目录中的所有文件和子目录。然后,我们检查每个文件是否以.sql
结尾,如果是,则将其添加到sql_files
列表中。最后,我们返回sql_files
列表。
在主代码中,我们将目录路径传递给get_sql_files()
函数,并将返回的.sql
文件列表打印出来。
请注意,这个答案不涉及任何云计算品牌商,而是提供了一个纯Python实现。
领取专属 10元无门槛券
手把手带您无忧上云