Python 2.7 - 遍历 CSV

Question

下午，

我在下面有以下 Python 2.7 代码。我希望能够做到以下几点：

1. 读取包含“ofile”变量值的 CSV 文件。
2. 运行以下脚本并对 CSV 中的每一行重复。
3. csv 文件不包含标题，并且具有“ofile”值 - A 列中每行一个。

我不想运行这个脚本，然后手动更新 ofile 变量，我想自动化这个过程。

提前致谢。

import time
import pysftp 
import sys
import os
from datetime import datetime
import calendar
import zipfile
import re

ofile = 'abc_'   
oupload = pysftp.Connection(host="xxxx", username="sxx", password="xNxxxx")

d = datetime.utcnow()
unixtime=calendar.timegm(d.utctimetuple())

import datetime
month = datetime.datetime.now().strftime("%m")

string = ofile+month+".*\.txt$"

possibleFiles = oupload.listdir("/")
for filename in possibleFiles:
        filedate = re.search(string, filename)
        if filedate:
            newfile = filename


timestamp  = oupload.stat(newfile).st_atime  

if timestamp != unixtime: 

        newtime=unixtime + 1800  
        zipname = ofile+str(newtime)+'.sync.zip'

        create_zip = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED) 

        oupload.get(newfile, newfile)
        oupload.close()

        newfilename = ofile+str(newtime)+'.sync' 
        os.rename(newfile, newfilename)


        create_zip.write(newfilename)

        create_zip.close()


else: 

        print "No file found"

CSV 文件没有标题，只有 A 列中的文件名。这是一个 sn-p：

filenamec
filename_b
filename_erf

Answer 1

您需要先将脚本转换为函数，然后遍历 csv 文件的每一行，提取所需的值并将其分配给 ofile 变量。最后，运行传递 ofile 的函数作为论据。我不能更准确，因为您没有发布任何实际 csv 的示例，但实现应该如下所示：

import time
import pysftp 
import sys
import os
from datetime import datetime
import calendar
import zipfile
import re


def myscript(ofile):  
    oupload = pysftp.Connection(host="xxxx", username="sxx", password="xNxxxx")
    d = datetime.utcnow()
    unixtime=calendar.timegm(d.utctimetuple())
    month = datetime.now().strftime("%m")
    string = ofile+month+".*\.txt$"

    possibleFiles = oupload.listdir("/")
    for filename in possibleFiles:
            filedate = re.search(string, filename)
            if filedate:
                newfile = filename

    timestamp  = oupload.stat(newfile).st_atime  
    if timestamp != unixtime:
        newtime=unixtime + 1800
        zipname = ofile+str(newtime)+'.sync.zip'
        create_zip = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)
        oupload.get(newfile, newfile)
        oupload.close()
        newfilename = ofile+str(newtime)+'.sync' 
        os.rename(newfile, newfilename)
        create_zip.write(newfilename)
        create_zip.close()

    else: 
        print "No file found"


mycsv = 'mydocument.csv'

with open(mycsv, 'r') as f:
    for line in f:
        if not line.startswith('\n'): #skip empty lines
            ofile = line.strip() #assuming you have only one column
            myscript(ofile)