Python - 从 CSV 文件创建多个文件夹答案

【问题标题】：Python - Create multiple folders from CSV filePython - 从 CSV 文件创建多个文件夹
【发布时间】：2016-10-10 05:20:00
【问题描述】：

我想使用 CSV 文件中的信息创建多个文件夹/目录（如果它们不存在）。

我从 csv 得到的信息如下：

    Column0 Column1 Column2 Column3
    51  TestName1   0   https://siteAdress//completed/file.txt
    53  TestName2   0   https://siteAdress//completed/file.txt
    67  TestName1   2   https://siteAdress//uploads/file.txt
    68  TestName1   2   https://siteAdress//uploads/file.txt

我想迭代column3，如果它包含'uploads'，那么它应该使用column1中提到的相应作业名创建一个文件夹，然后创建'input'文件夹并在其中创建相应的file.txt文件，如果column3包含'completed ' 那么它应该创建 'output' 文件夹（在输入文件夹旁边的同一个 jobname 文件夹中），然后在其中创建 'file.txt' 文件。并为第 1 列中提到的所有工作执行此操作。

类似这样的：

TestName1/input/file.txt
TestName1/output/file.txt
TestName1/output2/file.txt

TestName2/input/file.txt
TestName2/output/file.txt

注意：大多数数据将包含每个作业名称的多个输出文件夹。在这种情况下，它应该创建与 csv 文件中提到的一样多的输出文件夹。

到目前为止，我已经这样做了：

import csv, os
#reads from csv file
with open('limitedresult.csv') as csvfile:
    readCSV = csv.reader(csvfile, delimiter = ',')
    for row in readCSV:
        print(row)

非常感谢您的帮助，如果问题仍然令人困惑，请告诉我，我会尝试更详细地解释。

【问题讨论】：

标签： python csv

【解决方案1】：

以下方法应该可以帮助您入门：

打开 CSV 文件并跳过标题行。
读取一行，将其拆分为命名列。
如果file_url包含input，则使用input等子文件夹。
根据output_root 和子文件夹名称创建一个文件夹。
使用 Python Counter 跟踪每个子文件夹的使用次数。
将当前子文件夹计数添加到文件夹名称并创建任何必要的输出文件夹。
使用 Python requests 库从网站下载文本文件。
从 URL 中提取文件名并使用它来写入文件内容。

脚本如下：

from collections import Counter
import requests
import csv
import os

output_root = r'/myroot'
output_counter = Counter()

with open('limitedresult.csv', newline='') as csvfile:
    readCSV = csv.reader(csvfile)
    header = next(readCSV)

    for number, test, col2, file_url in readCSV:
        if 'completed' in file_url:
            sub_folder = 'input'
        elif 'uploads' in file_url:
            sub_folder = 'output' 
        else:
            sub_folder = None
            print('Invalid URL -', file_url)

        if sub_folder:
            output_folder = os.path.join(output_root, test, sub_folder)
            output_counter.update([output_folder])
            output_folder += str(output_counter[output_folder])
            os.makedirs(output_folder, exist_ok=True)
            data = requests.get(file_url)
            file_name = os.path.split(file_url)[1]

            with open(os.path.join(output_folder, file_name), 'w') as f_output:
                f_output.write(data.text)

注意，您可能需要安装requests，这通常可以使用pip install requests 来完成。

【讨论】：