[v2.0-alpha8] Multiple fixes + complete re-writing of script initial configuration + exceptions support enhancement + code cleanup

locales/en_us.toml

@@ -1,4 +1,4 @@
-# Localization file for video2geoframes.py script
+# Localization file for video2geoframes.py
 # English (US / World)
 #
 # Last edition : 2024-06-23

locales/fr_fr.toml

@@ -1,4 +1,4 @@
-# Localization file for video2geoframes.py script
+# Localization file for video2geoframes.py
 # French (France)
 #
 # Last edition : 2024-06-23

video2geoframes.py

@@ -8,7 +8,7 @@ Designed for contribution to street-level imagery projects like Mapillary or Pan
 
 __author__ = "Lucas MATHIEU (@campanu)"
 __license__ = "AGPL-3.0-or-later"
-__version__ = "2.0-alpha7"
+__version__ = "2.0-alpha8"
 __maintainer__ = "Lucas MATHIEU (@campanu)"
 __email__ = "campanu@luc-geo.fr"
 
@@ -19,9 +19,9 @@ from datetime import datetime, timedelta
 
 import cv2
 import piexif
-from tomlkit import dumps, loads
+from tomlkit import loads
 from tqdm import tqdm
-from exif import Image, GpsAltitudeRef
+
 
 # Functions
 def unix_path(path):
@@ -45,19 +45,56 @@ def byte_multiple(size):
         size = size / 1024
 
     multiple = multiples[i]
-
     return size, multiple
 
 
+def existing_items(expected_items: list, items: list):
+    presents_items = []
+    duplicated_items = []
+    missing_items = []
+
+    for eit in expected_items:
+        i = 0
+        for it in items:
+            if it == eit:
+                i += 1
+
+        if i == 0:
+            missing_items.append(eit)
+        elif i == 1:
+            presents_items.append(eit)
+        else:
+            duplicated_items.append(eit)
+
+    return {'presents': presents_items, 'duplicated': duplicated_items, 'missing': missing_items}
+
+
+def list_enumerator(item_list: list, intermediate_separator: str, last_separator: str):
+    i = 1
+
+    for it in item_list:
+        if i == 1:
+            enumerated_list = it
+        elif 1 < i < len(item_list):
+            enumerated_list = '{}{}{}'.format(enumerated_list, intermediate_separator, it)
+        else:
+            enumerated_list = '{}{}{}'.format(enumerated_list, last_separator, it)
+
+        i += 1
+
+    return enumerated_list
+
+
 # Start
-print('# video2geoframes.py ({})'.format(__version__))
+print("# video2geoframes.py (v{})\n".format(__version__))
 
 # Configuration settings
 base_path = unix_path(os.path.dirname(__file__))
-ini_file_path = '{}/video2geoframes.ini'.format(base_path)
+conf_file_path = '{}/video2geoframes_conf.toml'.format(base_path)
-ini_file_err = False
+conf_file_err = False
 
-## Default values
+# Default values
+mandatory_parameters = ['locale', 'exiftool_path']
 locale = 'en_us'
 min_frame_samp = 0.5
 max_frame_samp = 60.0
@@ -68,33 +105,46 @@ max_frame_height = 9000
 min_time_offset = -10.0
 max_time_offset = 10.0
 
-## Platform-dependent commands
+# Platform-dependent default paths
 if platform.system() == 'Windows':
-    ffmpeg_path = '{}/dependencies/ffmpeg-essentials/bin/ffmpeg.exe'.format(base_path)
     exiftool_path = '{}/dependencies/exiftool.exe'.format(base_path)
 else:
-    ffmpeg_path = 'ffmpeg'
     exiftool_path = 'exiftool'
 
-## ini file reading
-if os.path.exists(ini_file_path):
-    configuration = {}
-
 try:
-        with open(ini_file_path, 'r') as file:
+    # Configuration file reading
-            for line in file:
+    try:
-                if line[0] == '#':
+        if os.path.exists(conf_file_path):
-                    continue
+            with codecs.open(conf_file_path, mode='r', encoding='utf-8') as f:
+                conf_toml = loads(f.read())
+                f.close()
         else:
-                    (key, value) = line.split()
+            raise FileNotFoundError
-                    configuration[key] = value.replace('"', '')
 
-        locale = configuration.get('ui_language')
+        # Configuration check
-        max_frame_samp = float(configuration.get('max_frame_sample'))
+        reading_parameters = conf_toml['system'].keys()
-        ffmpeg_path = configuration.get('ffmpeg_path').replace('./', '{}/'.format(base_path))
+        check_result = existing_items(mandatory_parameters, reading_parameters)
-        exiftool_path = configuration.get('exiftool_path').replace('./', '{}/'.format(base_path))
+
-    except:
+        if len(check_result['missing']) != 0:
-        print('\nError... not readable or incomplete ini file. Default configuration will be used.')
+            missing_parameters_list = list_enumerator(check_result['missing'], ', ', ' and ')
+
+            if len(missing_parameters_list) > 1:
+                verb = 'is'
+            else:
+                verb = 'are'
+
+            print("(!) {} {} missing in configuration file.".format(missing_parameters_list, verb))
+            raise ValueError
+
+        # Configuration assignment
+        locale = conf_toml['system']['locale']
+        exiftool_path = unix_path(conf_toml['system']['exiftool_path'])
+    except (FileNotFoundError, ValueError):
+        print("\nError... configuration file doesn't exists or invalid.")
+        default_conf = str(input("Use default configuration instead (Y/N) ? ").upper())
+
+        if default_conf != 'Y':
+            raise InterruptedError
 
     # Localization
     locale_file_path = '{}/locales/{}.toml'.format(base_path, locale)
@@ -104,8 +154,8 @@ if os.path.exists(locale_file_path):
             locale_toml = loads(f.read())
             f.close()
     else:
-    print("Error.... file for locale \"{}\" doesn't exists or invalid.".format(locale))
+        print("\nError.... file for locale \"{}\" doesn't exists or invalid.".format(locale))
-    ValueError
+        raise InterruptedError
 
     user_agree = locale_toml['user']['agree'][0].upper()
     user_disagree = locale_toml['user']['disagree'][0].upper()
@@ -115,7 +165,7 @@ path_error = locale_toml['ui']['paths']['path_err']
     print(locale_toml['ui']['info']['intro'])
 
     # User input
-## TOML setting file
+    # TOML setting file
     toml_setting = input('\n{}'.format(locale_toml['ui']['parameters']['toml_setting'].format(user_agree, user_disagree)))
     i = 0
 
@@ -126,26 +176,24 @@ if toml_setting.upper() == 'O':
                 toml_file_path = unix_path(input('{}'.format(locale_toml['ui']['paths']['toml_file']))).strip()
 
                 if os.path.exists(toml_file_path):
-                break
-            else:
-                print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
-                True
-        except:
-            print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
-
                     with codecs.open(toml_file_path, mode='r', encoding='utf-8') as f:
                         setting_toml = loads(f.read())
                         f.close()
+                    break
+                else:
+                    raise FileNotFoundError
+            except (FileNotFoundError, ValueError):
+                print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
+                True
 
         # <--coding in progress-->
-    video_path = ''
+        raise NotImplementedError
-    gps_track_path = ''
 
-## Paths
+    # Paths
     else:
         print('\n{}'.format(locale_toml['ui']['info']['paths_title']))
 
-    ### Video file
+        # Video file
         while True:
             try:
                 video_path = unix_path(input('{}'.format(locale_toml['ui']['paths']['video_file']))).strip()
@@ -158,14 +206,14 @@ else:
                 print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
                 True
 
-    ### Video metadatas extraction
+        # Video metadatas extraction
         video = cv2.VideoCapture(video_path)
         video_fps = video.get(cv2.CAP_PROP_FPS)
         video_width = video.get(cv2.CAP_PROP_FRAME_WIDTH)
         video_height = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
         video_total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
 
-    ### GPS track file
+        # GPS track file
         while True:
             try:
                 gps_track_path = unix_path(input('{}'.format(locale_toml['ui']['paths']['gps_track']))).strip()
@@ -178,34 +226,34 @@ else:
                 print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
                 True
 
-    ### Output folder
+        # Output folder
         output_folder = unix_path(input(locale_toml['ui']['paths']['output_folder']))
 
-    ## Parameters
+        # Parameters
         print('\n{}'.format(locale_toml['ui']['info']['parameters_title']))
 
-    ### Timelapse video
+        # Timelapse video
-    timelapse = input(locale_toml['ui']['parameters']['timelapse'].format(user_agree, user_disagree))
+        timelapse = input(locale_toml['ui']['parameters']['timelapse'].format(user_agree, user_disagree)).upper()
 
-    if timelapse.upper() == user_agree:
+        if timelapse == user_agree:
-        ### Timelapse framerate parameter
+            # Timelapse framerate parameter
             while True:
                 try:
                     timelapse_fps = int(input(locale_toml['ui']['parameters']['timelapse_fps'].format(min_timelapse_fps,
                                                                                                       max_timelapse_fps)))
                     if max_timelapse_fps >= timelapse_fps >= min_timelapse_fps:
-                    frame_sampling = 1 / timelapse_fps
+                        frame_sampling = float(1 / timelapse_fps)
                         break
                     else:
-                    print(locale_toml['ui']['parameters']['timelapse_fps_err'].format(min_timelapse_fps,
+                        print('\n{}'.format(locale_toml['ui']['parameters']['timelapse_fps_err'].format(min_timelapse_fps,
-                                                                                      max_timelapse_fps))
+                                                                                          max_timelapse_fps)))
                         True
                 except ValueError:
-                print(locale_toml['ui']['parameters']['timelapse_fps_err'].format(min_timelapse_fps, max_timelapse_fps))
+                    print('\n{}'.format(locale_toml['ui']['parameters']['timelapse_fps_err'].format(min_timelapse_fps, max_timelapse_fps)))
                     True
 
         else:
-        ### Frame sampling parameter
+            # Frame sampling parameter
             while True:
                 try:
                     frame_sampling = float(input(locale_toml['ui']['parameters']['frame_samp'].format(min_frame_samp,
@@ -214,13 +262,13 @@ else:
                     if max_frame_samp >= frame_sampling >= min_frame_samp:
                         break
                     else:
-                    print(locale_toml['ui']['parameters']['frame_samp_err'].format(min_frame_samp, max_frame_samp))
+                        print('\n{}'.format(locale_toml['ui']['parameters']['frame_samp_err'].format(min_frame_samp, max_frame_samp)))
                         True
                 except ValueError:
-                print(locale_toml['ui']['parameters']['frame_samp_err'].format(min_frame_samp, max_frame_samp))
+                    print('\n{}'.format(locale_toml['ui']['parameters']['frame_samp_err'].format(min_frame_samp, max_frame_samp)))
                     True
 
-    ## Frame height parameter
+        # Frame height parameter
         if video_height <= max_frame_height:
             max_frame_height = int(round(video_height, 0))
 
@@ -234,26 +282,26 @@ else:
                 elif frame_height == 0:
                     break
                 else:
-                print(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height))
+                    print('\n{}'.format(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height)))
                     True
             except ValueError:
-            print(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height))
+                print('\n{}'.format(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height)))
                 True
 
-    ### Video start datetime parameter
+        # Video start datetime parameter
         while True:
             try:
                 video_start_datetime = input(locale_toml['ui']['parameters']['video_start_datetime'])
                 video_start_datetime_obj = datetime.strptime(video_start_datetime, '%Y-%m-%dT%H:%M:%S.%f')
                 break
             except ValueError:
-            print(locale_toml['ui']['parameters']['video_start_datetime_err'])
+                print('\n{}'.format(locale_toml['ui']['parameters']['video_start_datetime_err']))
                 True
 
-    ### Video recording timezone
+        # Video recording timezone
         video_rec_timezone = input(locale_toml['ui']['parameters']['rec_timezone'])
 
-    ### Time offset parameter
+        # Time offset parameter
         while True:
             try:
                 time_offset = float(input(locale_toml['ui']['parameters']['time_offset'].format(min_time_offset,
@@ -262,13 +310,13 @@ else:
                 if max_time_offset >= frame_sampling >= min_time_offset:
                     break
                 else:
-                print(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset))
+                    print('\n{}'.format(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset)))
                     True
             except ValueError:
-            print(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset))
+                print('\n{}'.format(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset)))
                 True
 
-    ## User-defined metadata
+        # User-defined metadata
         print('\n{}'.format(locale_toml['ui']['info']['tags_title']))
 
         make = input(locale_toml['ui']['metadatas']['make'])
@@ -284,9 +332,9 @@ video_duration = video_total_frames / video_fps
 
     video_start_datetime_obj = video_start_datetime_obj + timedelta(seconds=time_offset)
     video_start_datetime = video_start_datetime_obj.strftime('%Y-%m-%d %H:%M:%S')
-video_start_subsectime = int(video_start_datetime_obj.strftime('%f') / 1000)
+    video_start_subsectime = int(int(video_start_datetime_obj.strftime('%f')) / 1000)
 
-# Metadata recap
+    # Metadatas recap
     print('\n{}'.format(locale_toml['ui']['info']['metadatas'].format(video_file_name,
                                                                       round(video_file_size[0], 3), video_file_size[1],
                                                                       video_duration, video_start_datetime,
@@ -298,7 +346,7 @@ output_folder = '{}/{}'.format(output_folder, video_file_name)
     existing_path(output_folder)
 
     # Processes
-## Frame sampling + tagging (OpenCV + piexif)
+    # Frame sampling + tagging (OpenCV + piexif)
     print('\n{}'.format(locale_toml['processing']['sampling']))
 
     i = 0
@@ -316,11 +364,11 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
         video.set(cv2.CAP_PROP_POS_MSEC, t)
         ret, frame = video.read()
 
-    ### Image resizing
+        # Image resizing
         if frame_height != 0:
             resize_factor = video_height / frame_height
             image_height = frame_height
-        image_width = int(round(video_height * resize_factor), 0)
+            image_width = int(round(video_height * resize_factor, 0))
 
             frame = cv2.resize(frame, (image_width, image_height), interpolation=cv2.INTER_LANCZOS4)
 
@@ -330,28 +378,12 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
 
         cv2.imwrite(image_path, frame, [cv2.IMWRITE_JPEG_QUALITY, 88, cv2.IMWRITE_JPEG_PROGRESSIVE, 1, cv2.IMWRITE_JPEG_SAMPLING_FACTOR, 0x411111])
 
-    ## Time tags formatting
+        # Time tags formatting
         time_shift = i * frame_sampling
         current_datetime_obj = video_start_datetime_obj + timedelta(seconds=time_shift)
         current_datetime = current_datetime_obj.strftime('%Y:%m:%d %H:%M:%S')
         current_subsec_time = int(int(current_datetime_obj.strftime('%f')) / 1000)
 
-    # exif code
-    # with open(image_path, 'rb') as image_file:
-    #     image = Image(image_file)
-    #     image.make = make
-    #     image.model = model
-    #     image.author = author
-    #     image.copyright = "{}, {}".format(author, video_start_datetime_obj.strftime('%Y'))
-    #     image.datetime_original = current_datetime
-    #     #image.offset_time_original = video_rec_timezone
-    #
-    #     if current_subsec_time > 0:
-    #         image.subsec_time_original = str(current_subsec_time)
-    #
-    # with open(image_path, 'wb') as tagged_image_file:
-    #     tagged_image_file.write(image.get_file())
-
         # piexif code
         image_exif = piexif.load(image_path)
 
@@ -388,3 +420,8 @@ geotagging = os.system(geotagging_cmd)
 
     # End
     input('\n{}'.format(locale_toml['ui']['info']['end']))
+except NotImplementedError:
+    print("\nSorry, this function is not implemented, work in progress ;)")
+except InterruptedError:
+    input("\nEnd of program, press Enter to quit.")
+

video2geoframes_conf.toml

@@ -0,0 +1,19 @@
+# Default configuration file for video2geoframes.py
+# See documentation for more information
+#
+# Last edition : 2024-06-23
+
+# Mandatory section
+[system]
+# See documentation for supported locales
+locale = "en_us"
+exiftool_path = "./dependencies/exiftool.exe"
+
+[default]
+
+# Optional section to avoid manual input
+# Tags can be overwrited if presents in TOML setting file
+[default.tags]
+author = "Campanu"
+make = "Camera maker"
+camera = "Camera model"