mkimg.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -------------------------
# (c) kelu124
# cc-by-sa/4.0/
#
# Library for tagging images 
# Definitely not optimized
#
# -------------------------

'''Description: Autotagging images for the `20180813a` 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'

import os.path 
 
  
# Path 
path = os.getcwd().split("/")[-1]

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) )

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'] = "matty,mux,myapo"
		# Experiment
		metadata['Exif.Image.Make'] = path

		metadata['Exif.Photo.MakerNote'] = "Testing un0rick with max14866 mux with myapo"
		# Description
		metadata['Exif.Image.ImageDescription'] = "Testing the mux"

		# Saving the image

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