import json
from exiftool import ExifToolHelper as et
from shutil import copy2 as cp
from datetime import datetime as dt
import time

def get_img_filename(image: json):
  return image['path'].split("/")[-1]


def get_memory_location(memory: json):
  return memory['location'] if 'location' in memory else 'N/A'


def get_datetime_from_str(time: str):
  format_string = "%Y-%m-%dT%H:%M:%S.%fZ"
  return dt.strptime(time, format_string)


def print_memory_info(memory: json):
  """
  Takes a memory object from the ``memories.json`` file and prints out all necessarily information like:
   - names from the images
   - the taken time
   - the location if available
  """
  print(" - BeReal Moment: %s\n - Time Taken: %s\n - Front Image: %s\n - Back Image: %s\n - Location: %s\n - isLate: %s"
         % (memory['berealMoment'],
            memory['takenTime'],
            get_img_filename(memory['frontImage']), get_img_filename(memory['backImage']),
            get_memory_location(memory),
            memory['isLate']))
  

def apply_memory_on_imgs(memory: json):

  memory_dt = get_datetime_from_str(memory['takenTime'])
  memory_dt_str = memory_dt.strftime('%Y-%m-%d_%H-%M-%S')

  cp("./Photos/post/%s" % get_img_filename(memory['frontImage']),
     "./out/%s_front.webp" % memory_dt_str)
  cp("./Photos/post/%s" % get_img_filename(memory['backImage']),
     "./out/%s_back.webp" % memory_dt_str)




if __name__ == '__main__':
  f = open('memories.json')
  
  apply_memory_on_imgs(json.load(f)[0])

  # n = 0
  # for i in json.load(f):
  #   print("\nBeReal Nr. %s: " % n)
  #   print_memory_info(i)
  #   n+=1

  # for d in et().get_metadata("./Photos/post/ZB_-vUE7j2JePeEN.webp"):
  #   for k, v in d.items():
  #     print(f"{k} = {v}")
  

  f.close()