Esempio n. 1
0
def write_tags_to_image_via_et(image, tags):

    options = [
        '-codedcharacterset=UTF8', '-charset', 'iptc=cp1251', '-charset',
        'filename=cp1251', '{0}'.format(image)
    ]

    for tag in tags:
        if tag.lower() == "keywords":
            for i in tags[tag]:
                options.append('-iptc:{0}={1}'.format(tag, i))
        else:
            options.append('-iptc:{0}={1}'.format(tag, tags[tag]))

    cmd = [exiftool]
    cmd.extend(options)

    trace.debug('Cmd: {0}'.format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE)
    out, err = proc.communicate()

    trace.debug('stdout: {0}; stderr: {1}'.format(out, err))

    if proc.returncode:
        trace.error('Unexpected return code: {0}; cmd: {1}'.format(
            proc.returncode, cmd))

    return proc.returncode
def get_keywords(image):
    with open(cascades_json, encoding='utf-8') as json_file:
        cascades = json_load(json_file)

    keywords = []
    for cascade in sorted(cascades):
        if not cascades[cascade]['use']:
            continue

        trace.debug('Check with cascade {0}'.format(cascade))

        img = cv2.imread(image)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        faces = cv2_get_objects(gray,
                                cascades[cascade]['cascade_file'],
                                cascades[cascade]['scale_factor'],
                                cascades[cascade]['min_neighbors'],
                                cascades[cascade]['flags'],
                                cascades[cascade]['min_x'],
                                cascades[cascade]['min_y'],
                                cascades[cascade]['max_x'],
                                cascades[cascade]['max_y'])

        if faces.__len__():
            keywords.extend(cascades[cascade]['keywords'])
            if 'nested_object' in cascades[cascade]:
                for (x1, y1, x2, y2) in faces:
                    #show face
                    #img = cv2.rectangle(img,(x1,y1),(x1+x2,y1+y2),(255,0,0),2)
                    #roi_color = img[y1:y1+y2, x1:x1+x2]

                    roi_gray = gray[y1:y1+y2, x1:x1+x2]
                    smiles = cv2_get_objects(roi_gray,
                                             cascades[cascade]['nested_object']['cascade_file'],
                                             cascades[cascade]['nested_object']['scale_factor'],
                                             cascades[cascade]['nested_object']['min_neighbors'],
                                             cascades[cascade]['nested_object']['flags'],
                                             cascades[cascade]['nested_object']['min_x'],
                                             cascades[cascade]['nested_object']['min_y'],
                                             cascades[cascade]['nested_object']['max_x'],
                                             cascades[cascade]['nested_object']['max_y'])

                    if smiles.__len__():
                        #show smile
                        #for (ex,ey,ew,eh) in smiles:
                        #    cv2.rectangle(roi_color,(ex,ey),(ex+ew,ey+eh),(0,255,0),2)

                        keywords.extend(cascades[cascade]['nested_object']['keywords'])

            #resize
            #height, width = img.shape[:2]
            #img = cv2.resize(img, (int(height/1.5), int(width/1.2)))

            #show image
            #cv2.imshow('img',img)
            #cv2.waitKey(0)
            #cv2.destroyAllWindows()

    return list(set(keywords))
Esempio n. 3
0
def get_geo_tags_in_iptc_format(image):
    exif_gps = exif.get_exif_field(image, "GPSInfo")
    try:
        gps_info = exif.parse_gps_info(exif_gps, "dd") if exif_gps else None
    except KeyError:
        gps_info = None
    if not gps_info:
        return {}

    latitude = float(gps_info["latitude"][0])
    longitude = float(gps_info["longitude"][0])

    trace.debug("GPS-coordinates of {0}: lat {1}; lon {2}".format(
        image, latitude, longitude))

    y_d = get_info_yandex(latitude, longitude)
    iptc_d = dict(keywords=[])

    for tag in y_d:
        iptc_key = ADDR_TO_IPTC_FORMAT[tag]
        if iptc_key:
            iptc_d[iptc_key] = y_d[tag]
        else:
            iptc_d["keywords"].append(y_d[tag])

    iptc_d["keywords"].extend(get_info_osm(latitude, longitude))
    return iptc_d
Esempio n. 4
0
def write_tags_to_image_via_et(image, tags):

    options = ['-codedcharacterset=UTF8',
               '-charset', 'iptc=cp1251',
               '-charset', 'filename=cp1251',
               '{0}'.format(image)]

    for tag in tags:
        if tag.lower() == "keywords":
            for i in tags[tag]:
                options.append('-iptc:{0}={1}'.format(tag, i))
        else:
            options.append('-iptc:{0}={1}'.format(tag, tags[tag]))

    cmd = [exiftool]
    cmd.extend(options)

    trace.debug('Cmd: {0}'.format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE)
    out, err = proc.communicate()

    trace.debug('stdout: {0}; stderr: {1}'.format(out, err))

    if proc.returncode:
        trace.error('Unexpected return code: {0}; cmd: {1}'.format(proc.returncode, cmd))

    return proc.returncode
Esempio n. 5
0
def get_geo_tags_in_iptc_format(image):
    exif_gps = exif.get_exif_field(image, "GPSInfo")
    try:
        gps_info = exif.parse_gps_info(exif_gps, "dd") if exif_gps else None
    except KeyError:
        gps_info = None
    if not gps_info:
        return {}

    latitude = float(gps_info["latitude"][0])
    longitude = float(gps_info["longitude"][0])

    trace.debug("GPS-coordinates of {0}: lat {1}; lon {2}".format(image, latitude, longitude))

    y_d = get_info_yandex(latitude, longitude)
    iptc_d = dict(keywords=[])

    for tag in y_d:
        iptc_key = ADDR_TO_IPTC_FORMAT[tag]
        if iptc_key:
            iptc_d[iptc_key] = y_d[tag]
        else:
            iptc_d["keywords"].append(y_d[tag])

    iptc_d["keywords"].extend(get_info_osm(latitude, longitude))
    return iptc_d
Esempio n. 6
0
def get_keywords(image):
    with open(cascades_json, encoding='utf-8') as json_file:
        cascades = json_load(json_file)

    keywords = []
    for cascade in sorted(cascades):
        if not cascades[cascade]['use']:
            continue

        trace.debug('Check with cascade {0}'.format(cascade))

        img = cv2.imread(image)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        faces = cv2_get_objects(
            gray, cascades[cascade]['cascade_file'],
            cascades[cascade]['scale_factor'],
            cascades[cascade]['min_neighbors'], cascades[cascade]['flags'],
            cascades[cascade]['min_x'], cascades[cascade]['min_y'],
            cascades[cascade]['max_x'], cascades[cascade]['max_y'])

        if faces.__len__():
            keywords.extend(cascades[cascade]['keywords'])
            if 'nested_object' in cascades[cascade]:
                for (x1, y1, x2, y2) in faces:
                    #show face
                    #img = cv2.rectangle(img,(x1,y1),(x1+x2,y1+y2),(255,0,0),2)
                    #roi_color = img[y1:y1+y2, x1:x1+x2]

                    roi_gray = gray[y1:y1 + y2, x1:x1 + x2]
                    smiles = cv2_get_objects(
                        roi_gray,
                        cascades[cascade]['nested_object']['cascade_file'],
                        cascades[cascade]['nested_object']['scale_factor'],
                        cascades[cascade]['nested_object']['min_neighbors'],
                        cascades[cascade]['nested_object']['flags'],
                        cascades[cascade]['nested_object']['min_x'],
                        cascades[cascade]['nested_object']['min_y'],
                        cascades[cascade]['nested_object']['max_x'],
                        cascades[cascade]['nested_object']['max_y'])

                    if smiles.__len__():
                        #show smile
                        #for (ex,ey,ew,eh) in smiles:
                        #    cv2.rectangle(roi_color,(ex,ey),(ex+ew,ey+eh),(0,255,0),2)

                        keywords.extend(
                            cascades[cascade]['nested_object']['keywords'])

            #resize
            #height, width = img.shape[:2]
            #img = cv2.resize(img, (int(height/1.5), int(width/1.2)))

            #show image
            #cv2.imshow('img',img)
            #cv2.waitKey(0)
            #cv2.destroyAllWindows()

    return list(set(keywords))
Esempio n. 7
0
def get_data_from_image(image, option):
    cmd = [exiftool, '-j', '-g', '-charset', 'exiftool=Russian', str(option), str(image)]

    trace.debug('Cmd: {0}'.format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True)
    out, err = proc.communicate()
    trace.debug('stdout: {0}; stderr: {1}'.format(out, err))

    return json_loads(out[1:-2])
Esempio n. 8
0
def get_data_from_image(image, option):
    cmd = [exiftool, "-j", "-g", "-charset", "exiftool=Russian", str(option), str(image)]

    trace.debug("Cmd: {0}".format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True)
    out, err = proc.communicate()
    trace.debug("stdout: {0}; stderr: {1}".format(out, err))

    return json_loads(out[1:-2])
def get_keywords(image):

    with open(cascades_json, encoding='utf-8') as f:
        cascades = json_load(f)

    keywords = []
    for cascade in sorted(cascades):
        if not cascades[cascade]['use']:
            continue

        trace.debug('Check with cascade {0}'.format(cascade))

        is_nested_object = 'nested_object' in cascades[cascade]

        cmd = [object_detector, image,
               str(cascades[cascade]['cascade_file']),
               str(cascades[cascade]['scale_factor']),
               str(cascades[cascade]['min_neighbors']),
               str(cascades[cascade]['flags']),
               str(cascades[cascade]['min_x']),
               str(cascades[cascade]['min_y']),
               str(cascades[cascade]['max_x']),
               str(cascades[cascade]['max_y']),
               str(int(is_nested_object))]

        if is_nested_object:
            cmd.extend([str(cascades[cascade]['nested_object']['cascade_file']),
                        str(cascades[cascade]['nested_object']['scale_factor']),
                        str(cascades[cascade]['nested_object']['min_neighbors']),
                        str(cascades[cascade]['nested_object']['flags']),
                        str(cascades[cascade]['nested_object']['min_x']),
                        str(cascades[cascade]['nested_object']['min_y']),
                        str(cascades[cascade]['nested_object']['max_x']),
                        str(cascades[cascade]['nested_object']['max_y'])])

        proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True)
        out, _ = proc.communicate()

        trace.debug('Output: {0}'.format(out))

        if int(out.split()[1]) > 0:
            keywords.extend(cascades[cascade]['keywords'])
            if int(out.split()[4]) > 0:
                keywords.extend(cascades[cascade]['nested_object']['keywords'])

    return list(set(keywords))
Esempio n. 10
0
def write_iptc_tags_to_image(image, iptc_dict):
    cmd = [exiftool, "-codedcharacterset=UTF8", "-charset", "iptc=Russian", "-charset", "filename=Russian"]

    for item in iptc_dict:
        if item == "keywords":
            for kw in iptc_dict[item]:
                cmd.append("-iptc:{0}-={1}".format(item, kw))
                cmd.append("-iptc:{0}+={1}".format(item, kw))
        else:
            cmd.append("-iptc:{0}={1}".format(item, iptc_dict[item]))

    cmd.append(str(image))

    trace.debug("Cmd: {0}".format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True)
    out, err = proc.communicate()
    trace.debug("stdout: {0}; stderr: {1}".format(out, err))
Esempio n. 11
0
def write_iptc_tags_to_image(image, iptc_dict):
    cmd = [exiftool,
           '-codedcharacterset=UTF8',
           '-charset',
           'iptc=Russian',
           '-charset', 'filename=Russian']

    for item in iptc_dict:
        if item == "keywords":
            for kw in iptc_dict[item]:
                cmd.append("-iptc:{0}-={1}".format(item, kw))
                cmd.append("-iptc:{0}+={1}".format(item, kw))
        else:
            cmd.append("-iptc:{0}={1}".format(item, iptc_dict[item]))

    cmd.append(str(image))

    trace.debug('Cmd: {0}'.format(cmd))

    proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True)
    out, err = proc.communicate()
    trace.debug('stdout: {0}; stderr: {1}'.format(out, err))
    def cmd_save_photo(self):
        dir_with_photo = filedialog.askdirectory(parent=self.master, title=get_name("dia_save_photo"), initialdir='/')
        if not dir_with_photo:
            return
        if settings["save_photo"]["save_originals"] == "True":
            file_operation = shutil_copy2
        else:
            file_operation = shutil_move

        # Get list of files to save
        pho_for_saving_without_date = []
        ph_for_saving_with_date = []

        if settings["save_photo"]["check_unsorted"] == "True":
            # Check unsorted files
            files = next(os_walk(os_path.join(settings["projects_dir"], dir_unsorted)))[2]
            for file in files:
                if os_path.splitext(file)[-1].lower() in supported_image_ext:
                        found_matching = dt_in_fn_regex.match(file)
                        if found_matching:
                            try:
                                # Convert collected numeric parts into datetime object
                                ph_for_saving_with_date.append([os_path.join(settings["projects_dir"],
                                                                             dir_unsorted,
                                                                             file),
                                                                datetime.strptime(str(found_matching.group(1)),
                                                                                  '%Y-%m-%d_%H-%M-%S'),
                                                                None])
                            except ValueError:
                                continue

        for root, _, files in os_walk(dir_with_photo):
            for file in files:
                if os_path.splitext(file)[-1].lower() in supported_image_ext:
                    try:
                        # Try to find date/time in metadata
                        possible_dt = et.get_data_from_image(os_path.join(root, file),
                                                             "-EXIF:DateTimeOriginal")["EXIF"]["DateTimeOriginal"]
                        # Convert collected numeric parts into datetime object
                        ph_for_saving_with_date.append([os_path.join(root,
                                                                     file),
                                                        datetime.strptime(possible_dt,
                                                                          '%Y:%m:%d %H:%M:%S'),
                                                        None])
                    # If date/time were not found in metadata too
                    except ValueError:
                        pho_for_saving_without_date.append(os_path.join(root, file))
                        continue

        # Connect photo and project basing on date/time
        for project in next(os_walk(settings["projects_dir"]))[1]:
            if not os_path.isfile(os_path.join(settings["projects_dir"], project, project_file)):
                continue

            with open(os_path.join(os_path.join(settings["projects_dir"]),
                                   project,
                                   project_file),
                      encoding='utf-8') as _f:
                pd = json_load(_f)

            # Parse project timeslot
            prj_start = '{0} {1}'.format(pd["timeslot"]["start"]["date"], pd["timeslot"]["start"]["time"])
            prj_start = datetime.strptime(prj_start, "%d.%m.%Y %H:%M")
            prj_finish = '{0} {1}'.format(pd["timeslot"]["finish"]["date"], pd["timeslot"]["finish"]["time"])
            prj_finish = datetime.strptime(prj_finish, "%d.%m.%Y %H:%M")

            for ph in ph_for_saving_with_date:
                if ph[2] is not None:
                    continue

                if prj_start <= ph[1] <= prj_finish:  # If photo date/time in project timeslot
                    ph[2] = os_path.join(settings["projects_dir"], project, dir_source)

        for ph in ph_for_saving_with_date:
            dest_dir = os_path.normpath(ph[2]) if ph[2] is not None else os_path.join(settings["projects_dir"],
                                                                                      dir_unsorted)
            # TODO: file renaming according to template YYYY-MM-DD_HH-MM-SS.ext
            if os_path.split(ph[0])[0] == dest_dir:
                trace.debug("Try to move photo to the same location: {0}".format(ph[0]))
            else:
                trace.debug("Save photo: {0} -> {1}".format(os_path.normpath(ph[0]), dest_dir))
                try:
                    # Copy/move image
                    file_operation(os_path.normpath(ph[0]), dest_dir)
                    # Copy/move XMP file too if it exists
                    if os_path.isfile(os_path.splitext(ph[0])[0] + xmp_ext):
                        file_operation(os_path.splitext(ph[0])[0] + xmp_ext, dest_dir)
                except shutil_Error as e:  # For example, if file already exists in destination directory
                    trace.warning(e)
Esempio n. 13
0
    def cmd_save_photo(self):
        dir_with_photo = filedialog.askdirectory(
            parent=self.master,
            title=get_name("dia_save_photo"),
            initialdir='/')
        if not dir_with_photo:
            return
        if settings["save_photo"]["save_originals"] == "True":
            file_operation = shutil_copy2
        else:
            file_operation = shutil_move

        # Get list of files to save
        pho_for_saving_without_date = []
        ph_for_saving_with_date = []

        if settings["save_photo"]["check_unsorted"] == "True":
            # Check unsorted files
            files = next(
                os_walk(os_path.join(settings["projects_dir"],
                                     dir_unsorted)))[2]
            for file in files:
                if os_path.splitext(file)[-1].lower() in supported_image_ext:
                    found_matching = dt_in_fn_regex.match(file)
                    if found_matching:
                        try:
                            # Convert collected numeric parts into datetime object
                            ph_for_saving_with_date.append([
                                os_path.join(settings["projects_dir"],
                                             dir_unsorted, file),
                                datetime.strptime(str(found_matching.group(1)),
                                                  '%Y-%m-%d_%H-%M-%S'), None
                            ])
                        except ValueError:
                            continue

        for root, _, files in os_walk(dir_with_photo):
            for file in files:
                if os_path.splitext(file)[-1].lower() in supported_image_ext:
                    try:
                        # Try to find date/time in metadata
                        possible_dt = et.get_data_from_image(
                            os_path.join(root, file), "-EXIF:DateTimeOriginal"
                        )["EXIF"]["DateTimeOriginal"]
                        # Convert collected numeric parts into datetime object
                        ph_for_saving_with_date.append([
                            os_path.join(root, file),
                            datetime.strptime(possible_dt,
                                              '%Y:%m:%d %H:%M:%S'), None
                        ])
                    # If date/time were not found in metadata too
                    except ValueError:
                        pho_for_saving_without_date.append(
                            os_path.join(root, file))
                        continue

        # Connect photo and project basing on date/time
        for project in next(os_walk(settings["projects_dir"]))[1]:
            if not os_path.isfile(
                    os_path.join(settings["projects_dir"], project,
                                 project_file)):
                continue

            with open(os_path.join(os_path.join(settings["projects_dir"]),
                                   project, project_file),
                      encoding='utf-8') as _f:
                pd = json_load(_f)

            # Parse project timeslot
            prj_start = '{0} {1}'.format(pd["timeslot"]["start"]["date"],
                                         pd["timeslot"]["start"]["time"])
            prj_start = datetime.strptime(prj_start, "%d.%m.%Y %H:%M")
            prj_finish = '{0} {1}'.format(pd["timeslot"]["finish"]["date"],
                                          pd["timeslot"]["finish"]["time"])
            prj_finish = datetime.strptime(prj_finish, "%d.%m.%Y %H:%M")

            for ph in ph_for_saving_with_date:
                if ph[2] is not None:
                    continue

                if prj_start <= ph[
                        1] <= prj_finish:  # If photo date/time in project timeslot
                    ph[2] = os_path.join(settings["projects_dir"], project,
                                         dir_source)

        for ph in ph_for_saving_with_date:
            dest_dir = os_path.normpath(
                ph[2]) if ph[2] is not None else os_path.join(
                    settings["projects_dir"], dir_unsorted)
            # TODO: file renaming according to template YYYY-MM-DD_HH-MM-SS.ext
            if os_path.split(ph[0])[0] == dest_dir:
                trace.debug(
                    "Try to move photo to the same location: {0}".format(
                        ph[0]))
            else:
                trace.debug("Save photo: {0} -> {1}".format(
                    os_path.normpath(ph[0]), dest_dir))
                try:
                    # Copy/move image
                    file_operation(os_path.normpath(ph[0]), dest_dir)
                    # Copy/move XMP file too if it exists
                    if os_path.isfile(os_path.splitext(ph[0])[0] + xmp_ext):
                        file_operation(
                            os_path.splitext(ph[0])[0] + xmp_ext, dest_dir)
                except shutil_Error as e:  # For example, if file already exists in destination directory
                    trace.warning(e)