0

crop_rows一番下にある calledの出力をdelete_Apps、すぐ後に calledの入力に送信する方法を見つけようとしていますが、何をフィードすればよいかわかりません。また、削除する必要がありますか:

file_obj.close()

私のスクリプトが両方の関数を実行し続けるように、crop_rows関数から？

import os, csv, sys, Tkinter, tkFileDialog as fd


# stop tinker shell from opening as only needed for file dialog

root = Tkinter.Tk()
root.withdraw()


#crop_rows deletes the extra info automatically generated in the Waders report 

def crop_rows(in_path):
    # read file into memory
    file_obj = open(in_path, 'rb')
    reader = csv.reader(file_obj, delimiter='\t')
    data = []
    for row in reader:
        if not row or not any(row):
            break #stop at empty row
        else:
            data.append(row)
    file_obj.close()

    print 'Found', len(data), 'rows of data without empty lines.'
    conf = raw_input('delete remaining lines? (Y|N): ').upper()[0]

    if conf == 'Y':
        # write data to file
        file_obj = open(in_path, 'wb')
        writer = csv.writer(file_obj)
        writer.writerows(data)
        file_obj.close



#delete_Apps deletes and leads that are currently Applicants in gHire as their Status

def delete_Apps(in_path):
    # read file into memory
    file_obj = open(in_path, 'rb')
    reader = csv.reader(file_obj, delimiter='\t')
    data = []
    for row in reader:
        if 'Applicant' not in row:
            data.append(row)
    file_obj.close()

    print 'Found', len(data), 'Leads with Applicant in gHire as Status.'
    conf = raw_input('delete these leads? (Y|N): ').upper()[0]

    if conf == 'Y':
        # write data to file
        file_obj = open(in_path, 'wb')
        writer = csv.writer(file_obj)
        writer.writerows(data)
        file_obj.close


def main():
    in_path = None
    prog_name = sys.argv[0]

    # check if in_path are inlcuded as cmd line args...
    if len(sys.argv) > 1:
        in_path = sys.argv[1]
        if not os.path.exists(in_path):
            print 'Usage:', prog_name, '[file_path>]'
            print 'cannot find the file provided for file_path:\n', in_path
            sys.exit("Error - invalid excel_file_path arg")
    else:
        try:
            # set current working directory to user's my documents folder
            os.chdir(os.path.join(os.getenv('userprofile'),'documents'))
        except:
            pass

    # ask user for path to file...
    while not in_path:
        print "Please select the file to read data from ..."
        try:
            in_path = fd.askopenfilename()
        except:
            print 'Error selecting file.'
        if not in_path:
            cont = raw_input('Do you want to continue? (Y|N): ').upper()[0]
            if cont == 'N':
                sys.exit("Error - unable to select input file")

    crop_rows(in_path)


    delete_Apps(in_path)




if __name__ == '__main__':
    main()
4

2 に答える 2

1

crop_rows をジェネレーター関数 ( https://wiki.python.org/moin/Generatorsを参照) に変更し、delete_Apps で使用します。

于 2013-10-03T16:55:19.133 に答える
0
  1. .csv ファイルを個別に読み取る
  2. crop_rows行を入力として受け取り、新しいデータを返すように変更します
  3. 次にdelete_Apps、行の入力を受け取り、変更されたデータを返すように変更します
  4. 最後に .csv ファイルを書き込みます

次に、次のように、関数への呼び出しを単純にネストできます。

def read_input_file(in_path):
    # use with not to worry about closing the file
    with open(in_path, 'rb') as file_obj:
        reader = csv.reader(file_obj, delimiter='\t')
        return list(reader)

def crop_rows(input_rows):
    data = []
    for row in input_rows:
        if not row or not any(row):
            break #stop at empty row
        else:
            data.append(row)

    print 'Found', len(data), 'rows of data without empty lines.'
    conf = raw_input('delete remaining lines? (Y|N): ').upper()[0]

    if conf == 'Y':
        return data       
    else:
        # return unmodified data
        return input_rows

def delete_Apps(input_rows):
    data = []
    for row in input_rows:
        if 'Applicant' not in row:
            data.append(row)

    print 'Found', len(data), 'Leads with Applicant in gHire as Status.'
    conf = raw_input('delete these leads? (Y|N): ').upper()[0]

    if conf == 'Y':
        return data
    else:
        return input_rows

def write_output_file(data, out_path):
    wirg open(out_path, 'wb') as file_obj:
    writer = csv.writer(file_obj)
    writer.writerows(data)
    file_obj.close()

最後の呼び出し:

file_path='/tmp/whatever.csv'
write_output_file(delete_Apps(crop_rows(read_input_file(file_path)), file_path)
于 2013-10-03T17:09:57.620 に答える