geo_extract.py

import os
import shutil

from functools import partial
from geopy.geocoders import Nominatim

from ImageMetaData import ImageMetaData


def get_file_list(path):
    """Get the list of files in a specifc path"""
    files_list = [f for f in os.listdir(path) if (
        os.path.isfile(os.path.join(path, f)) and f != '.DS_Store')]
    return files_list


def get_jpg_file_list(img_list):
    """Return a list of accepted jpg files from the pictures list"""
    jpg_ext = ['.jpg', '.jpeg']
    jpg_list = [img for img in img_list if (
        os.path.splitext(img)[-1].lower() in jpg_ext)]
    return jpg_list


def img_metadata_to_dict():
    """
    Extracts the metadata of the images in the pictures folders and
    convert it into an useful, readable dictionary.
    """
    path_name = os.path.join(os.getcwd(), 'pictures')
    jpg_img_list = get_jpg_file_list(get_file_list(path_name))
    jpg_img_data_list = []
    geolocator = Nominatim(user_agent="geo_pic_finder")
    reverse = partial(geolocator.reverse, language="en")

    for img in jpg_img_list:
        img_path = os.path.join(path_name, img)

        try:
            meta_data = ImageMetaData(img_path)
            lat, lon = meta_data.get_lat_lon()
            location = reverse([lat, lon])
            country = location.raw['address']['country']
            jpg_img_data_list.append({'filename': img,
                                      'filepath': img_path,
                                      'metadata': meta_data,
                                      'latitude': lat,
                                      'longitude': lon,
                                      'country': country})
        except Exception:
            pass

    return jpg_img_data_list


def create_folders_for_pictures(img_list):
    """
    Create a folder for each distinct country and move the specific
    picture files into the respective folders.
    """
    path_name = os.path.join(os.getcwd(), 'pictures')

    for img in img_list:
        img_path = os.path.join(path_name, img['country'])
        if not os.path.exists(img_path):
            os.makedirs(img_path)

        dest = os.path.join(img_path, img['filename'])
        shutil.move(img['filepath'], dest)


def main():
    img_list = img_metadata_to_dict()
    create_folders_for_pictures(img_list)


if __name__ == '__main__':
    main()