mkimg.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -------------------------
# (c) kelu124
# cc-by-sa/4.0/
#
# Library for creating images from the 
# files created by pulser module.
# Used in `20180216a`
# Definitely not optimized
#
# -------------------------

'''Description: Autotagging images for the `20180216a` experiment.'''

__author__      = "kelu124"
__copyright__   = "Copyright 2018, Kelu124"
__license__ 	= "cc-by-sa/4.0/" 


import sys
import pyexiv2
import os

class bcolors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

Imgs = []
for dirpath, dirnames, filenames in os.walk("."):
    for filename in [f for f in filenames if ( f.endswith(".jpg") or f.endswith(".png") )]:
        Imgs.append( os.path.join(dirpath, filename) )

print Imgs

for FileName in Imgs:
	edit = 0

	metadata = pyexiv2.ImageMetadata(FileName)
	try:
		metadata.read()
	except IOError:
		print "Not an image"
	else:
		# Modules
		metadata['Exif.Image.Software'] = "elmo, tomtom, alt.tbo, doj, goblin, cletus"
		# Experiment
		metadata['Exif.Image.Make'] = "20180216a"

		# Type of picture
		if "DAT_rawsignal" in FileName:
			metadata['Exif.Photo.MakerNote'] = "graph"
			# Description
			metadata['Exif.Image.ImageDescription'] = "Map of the DATA file from the 20180216a experiment"
		elif "DAT_spectrum" in FileName:
			metadata['Exif.Photo.MakerNote'] = "graph"
			# Description
			metadata['Exif.Image.ImageDescription'] = "Spectrum of the DATA file from the 20180216a experiment"
		elif "_line" in FileName and "_clean" not in FileName:
			metadata['Exif.Photo.MakerNote'] = "BSC"
			# Description
			metadata['Exif.Image.ImageDescription'] = "Raw line from the 20180216a experiment"
		elif "_line" in FileName and "_clean" in FileName:
			metadata['Exif.Photo.MakerNote'] = "BSC"
			# Description
			metadata['Exif.Image.ImageDescription'] = "Cleaned line from the 20180216a experiment"
		elif "IMG_" in FileName:
			metadata['Exif.Photo.MakerNote'] = "setup"
			# Description
			metadata['Exif.Image.ImageDescription'] = " Setting up the 20180216a experiment"

		# Saving the image

		metadata.write()
		print "Image "+FileName+" saved"