根据xml个节点复制并重命名图片

Question

我正在尝试将所有图片从一个目录（也包括子目录）复制到另一个目标目录。 每当在 xml 文件之一中找到确切的图片名称时，该工具应抓取所有信息（父节点和子节点中的属性）并根据这些节点信息创建子目录，它还应该重命名图片文件。 从节点中提取所有信息的部分已经完成。

from bs4 import BeautifulSoup as bs

path_xml = r"path\file.xml"

content = []

with open(res, "r") as file:
    content = file.readlines()

content = "".join(content)

def get_filename(_content):
    bs_content = bs(_content, "html.parser")

    # some code
    
    picture_path = f'{pm_1}{pm_2}\{pm_3}\{pm_4}\{pm_5}_{pm_6}_{pm_7}\{pm_8}\{pm_9}.jpg'

get_filename(content)

所以最后我得到了一个字符串值，其中包含我想要的目录路径和文件名。

现在我很难打开一个目录中的所有 xml 个文件，而不是只打开一个文件。 我试过这个：

import os
dir_xml = r"path"

res = []

for path in os.listdir(dir_xml):
    if os.path.isfile(os.path.join(dir_xml, path)):
        res.append(path)

with open(res, "r") as file:
    content = file.readlines()

但它给了我这个错误： TypeError: expected str, bytes or os.PathLike object, not list

我怎样才能通读所有 xml 个文件而不是一个？ 我有数百个 xml 文件，所以需要花点时间：D

还有一个问题：如何根据字符串创建目录？

假设picture_path的值为AB\C\D\E_F_G\H\I.jpg

我需要另一个目录路径作为创建文件夹的目的地，以及一个 function 以某种方式基于该字符串创建文件夹。 我怎样才能做到这一点？

Answer 1

您可以使用os.walk()递归搜索文件：

import os

dir_xml = r"path"

for root, dirs, files in os.walk(dir_xml):  #topdown=False
    for names in files:
        if ".xml" in names:
            print(f"file path: {root}\n XML-Files: {names}")
            with open(names, 'r') as file:
                content = file.readlines()

Answer 2

要读取一个目录下的所有XML个文件，可以修改你的代码如下：

import os
dir_xml = r"path"

for path in os.listdir(dir_xml):
    if path.endswith(".xml"):
        with open(os.path.join(dir_xml, path), "r") as file:
            content = file.readlines()
        content = "".join(content)
        get_filename(content)

此代码使用 os.listdir() function 获取 dir_xml 指定目录中所有文件的列表。 然后它使用 for 循环遍历文件列表，检查每个文件是否以 .xml 扩展名结尾。 如果是，它会打开文件，读取其内容，并将其传递给 get_filename function。

要基于字符串创建目录，可以使用 os.makedirs function。例如：

import os

picture_path = r'AB\C\D\E_F_G\H\I.jpg'
 dest_path = r'path_to_destination'

os.makedirs(os.path.join(dest_path, os.path.dirname(picture_path)), exist_ok=True)

在这段代码中，os.path.join 用于将 dest_path 和 picture_path 的目录部分合并为一个完整路径。 os.path.dirname 用于提取 picture_path 的目录部分。 然后使用 os.makedirs function 创建路径指定的目录，并将 exist_ok 参数设置为 True 以允许 function 成功，即使目录已经存在。

最后可以使用shutil库将图片文件复制到目的地并重命名，像这样：

import shutil

src_file = os.path.join(src_path,  picture_path)
dst_file = os.path.join(dest_path, picture_path)

shutil.copy(src_file, dst_file)

这里，src_file 是源图片文件的完整路径，dst_file 是目标文件的完整路径。 然后使用 shutil.copy function 将文件从源复制到目标。