如何将exif数据提取到csv文件中?

时间:2016-10-13 01:41:35

标签: python csv metadata exif pyexiv2

import pyexiv2
import os

print "Enter the full path to the directory that your images are conatined in."
print "-------------------------------------------"
newFileObj = open('C:\\users\\wilson\\desktop\\Metadata.csv', 'w')
targ_dir = raw_input('Path: ')

targ_files = os.listdir(targ_dir)

def getEXIFdata (imageFile):
    if imageFile.endswith(".db"):
        f = 1
    else:

        EXIFData = pyexiv2.ImageMetadata(imageFile)
        EXIFData.read()
        CamMake = EXIFData['Exif.Image.Make']
        DateTime = EXIFData['Exif.Image.DateTime']
        CamModel = EXIFData['Exif.Image.Model']
    for image in targ_files:
        getEXIFdata(targ_dir+"\\"+img)
        newFileObj.write(DateTime+' , '+CamMake+' , '+CamModel+'\r\n')
newFileObj.close()

end = raw_input("Press Enter to Finish: ")

这是我到目前为止所做的,但我只是不明白如何将数据实际存入文件。它包装文件,但它只是空白。我试过在底部移动,但我似乎无法让它发挥作用。我是python的新手,所以当你暗示我应该做什么时,请你保持简单。

1 个答案:

答案 0 :(得分:0)

如果您想获取exif数据,请使用.value获取数据值。 这是一个例子。

# -*-coding:utf-8-*-
import sys
import csv
import os
import argparse
import pyexiv2

def main():
    parser = argparse.ArgumentParser(description="Change the txt file to csv.")
    parser.add_argument("-i", action="store", dest="infile")
    parser.add_argument("-o", action="store", dest="outfile")
    parser_argument = parser.parse_args()

    fatherdir = os.getcwd()  # code directory
    inputfile = outputfile = None

    # input exif file
    if parser_argument.infile:
        infilepaths = os.path.split(parser_argument.infile)
        # 'C:\User\lenovo\Desktop\pakistan.txt' ---> ['C:\User\lenovo\Desktop','pakistan.txt']
        if infilepaths[0]:  # full path
            inputfile = parser_argument.infile
            fatherdir = infilepaths[0]
        # 'pakistan.txt' ---> ['','pakistan.txt']
        else:  # only file name
            inputfile = fatherdir + '/' + parser_argument.infile
    # output csv file
    if parser_argument.outfile:
        outfilepaths = os.path.split(parser_argument.outfile)
        if outfilepaths[0]:  # full path
            outputfile = parser_argument.outfile
        else:
            outputfile = fatherdir + '/' + parser_argument.outfile
    else:
        outputfile = fatherdir + '/test_csv.csv'
    parse(inputfile, outputfile)


def parse(inputfile, outputfile):
    csvcontent = file(outputfile, 'wb')
    writer = csv.writer(csvcontent)

    exif_data = getEXIFdata(inputfile)
    writer.writerow([exif_data['Exif.Image.Orientation'].value,
                     exif_data['Exif.Photo.PixelXDimension'].value,
                     exif_data['Exif.Photo.PixelYDimension'].value])
    # for line in open(inputfile).readlines():
    #     writer.writerow([a for a in line.split('\t')])

    csvcontent.close()

def getEXIFdata (imageFile):
    if imageFile.endswith(".db"):
        print 'Skip this file'
    else:
        exif_data = pyexiv2.ImageMetadata(imageFile)
        exif_data.read()
        for s, v in exif_data.items():
            print s, v
        cam_a = exif_data['Exif.Image.Orientation'].value
        cam_b = exif_data['Exif.Photo.PixelXDimension'].value
        cam_c = exif_data['Exif.Photo.PixelYDimension'].value
        # add exif value
        ekey = 'Exif.Photo.UserComment'
        evalue = 'A comment.'
        exif_data[ekey] = pyexiv2.ExifTag(ekey, evalue)
        #metadata.write()
        return exif_data

if __name__ == '__main__':
    main()

像这样运行此代码

python exif2csv.py -i wifi.jpg -o demo_csv.csv

如果你想在目录中循环文件,我想你可以自己解决。