Python 2.7-通过CSV进行迭代

Question

Afternoon,

I have the following Python 2.7 code below. I would like to be able to do the following:

1. Read a CSV file which contains values for the 'ofile' variable.
2. Run the below script and repeat for each row in the CSV.
3. The csv file contains no headers and has the 'ofile' values - one per row in column A.

Instead of me running this script, and then updating the ofile variable manually, i would like to automate this process.

Thanks in advance.

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"

The CSV file has no headers, and only has filenames in column A. Here is a snippet:

filenamec
filename_b
filename_erf

Answer 1

You need to convert your script in a function first, then you iterate over each line of the csv file, you extract the value you need and assign it the to ofile variable.Finally, run the function passing ofile as an argument. I can't be more precise since you haven't posted any example of the actual csv, but the implementation should look something like this:

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)