mirror of
https://github.com/SoPat712/BeReal-Export-Manager.git
synced 2025-08-22 02:38:45 -04:00
add verbose function
This commit is contained in:
19
main.py
19
main.py
@@ -14,6 +14,7 @@ def init_parser():
|
|||||||
"Wildcards can be used: 'DD.MM.YYYY-*'")
|
"Wildcards can be used: 'DD.MM.YYYY-*'")
|
||||||
parser.add_argument('-y', '--year', type=int, help="Exports the given year")
|
parser.add_argument('-y', '--year', type=int, help="Exports the given year")
|
||||||
parser.add_argument('-p', '--path', type=str, help="Set a custom output path (default ./out)")
|
parser.add_argument('-p', '--path', type=str, help="Set a custom output path (default ./out)")
|
||||||
|
parser.add_argument('-v', '--verbose', action=argparse.BooleanOptionalAction, help="Explain what is being done")
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
if args.year and args.timespan:
|
if args.year and args.timespan:
|
||||||
@@ -25,6 +26,7 @@ def init_parser():
|
|||||||
def init_global_var(args: argparse.Namespace):
|
def init_global_var(args: argparse.Namespace):
|
||||||
global time_span
|
global time_span
|
||||||
global out_path
|
global out_path
|
||||||
|
global verbose
|
||||||
|
|
||||||
if args.timespan:
|
if args.timespan:
|
||||||
temp_times = args.timespan.strip().split("-")
|
temp_times = args.timespan.strip().split("-")
|
||||||
@@ -40,8 +42,12 @@ def init_global_var(args: argparse.Namespace):
|
|||||||
else:
|
else:
|
||||||
out_path = "./out"
|
out_path = "./out"
|
||||||
|
|
||||||
|
verbose = args.verbose
|
||||||
|
|
||||||
|
|
||||||
|
def verbose_msg(msg: str):
|
||||||
|
if verbose: print(msg)
|
||||||
|
|
||||||
def get_img_filename(image: json):
|
def get_img_filename(image: json):
|
||||||
"""
|
"""
|
||||||
Returns the image filename from a image object (frontImage, backImage, primary, secondary)
|
Returns the image filename from a image object (frontImage, backImage, primary, secondary)
|
||||||
@@ -61,19 +67,24 @@ def apply_memory_on_imgs(memory: json, memory_dt: datetime):
|
|||||||
"""
|
"""
|
||||||
Makes a copy of the front and back images and adds information from the memory object as exif tags to the image
|
Makes a copy of the front and back images and adds information from the memory object as exif tags to the image
|
||||||
"""
|
"""
|
||||||
|
|
||||||
img_names = ["%s/%s_%s.webp" % (out_path, memory_dt.strftime('%Y-%m-%d_%H-%M-%S'), temp_times)
|
img_names = ["%s/%s_%s.webp" % (out_path, memory_dt.strftime('%Y-%m-%d_%H-%M-%S'), temp_times)
|
||||||
for temp_times in ['front', 'back']]
|
for temp_times in ['front', 'back']]
|
||||||
|
|
||||||
for img_type, img_name in zip(['frontImage', 'backImage'], img_names):
|
for img_type, img_name in zip(['frontImage', 'backImage'], img_names):
|
||||||
cp("./Photos/post/%s" % get_img_filename(memory[img_type]), img_name)
|
img_filename = get_img_filename(memory[img_type])
|
||||||
|
verbose_msg("Export %s image to %s" % (img_filename, img_name))
|
||||||
|
cp("./Photos/post/%s" % img_filename, img_name)
|
||||||
|
|
||||||
if 'location' in memory:
|
if 'location' in memory:
|
||||||
|
verbose_msg("Add metadata to image:\n - DateTimeOriginal=%s\n - GPS=(%s, %s)" % (memory_dt, memory['location']['latitude'], memory['location']['longitude']))
|
||||||
et().set_tags(img_names,
|
et().set_tags(img_names,
|
||||||
tags={"DateTimeOriginal": memory_dt.strftime("%Y:%m:%d %H:%M:%S"),
|
tags={"DateTimeOriginal": memory_dt.strftime("%Y:%m:%d %H:%M:%S"),
|
||||||
"GPSLatitude*": memory['location']['latitude'],
|
"GPSLatitude*": memory['location']['latitude'],
|
||||||
"GPSLongitude*": memory['location']['longitude']},
|
"GPSLongitude*": memory['location']['longitude']},
|
||||||
params=["-P", "-overwrite_original"])
|
params=["-P", "-overwrite_original"])
|
||||||
else:
|
else:
|
||||||
|
verbose_msg("Add metadata to image:\n - DateTimeOriginal=%s" % memory_dt)
|
||||||
et().set_tags(img_names,
|
et().set_tags(img_names,
|
||||||
tags={"DateTimeOriginal": memory_dt.strftime("%Y:%m:%d %H:%M:%S")},
|
tags={"DateTimeOriginal": memory_dt.strftime("%Y:%m:%d %H:%M:%S")},
|
||||||
params=["-P", "-overwrite_original"])
|
params=["-P", "-overwrite_original"])
|
||||||
@@ -81,9 +92,10 @@ def apply_memory_on_imgs(memory: json, memory_dt: datetime):
|
|||||||
|
|
||||||
def export_images(memories: json):
|
def export_images(memories: json):
|
||||||
|
|
||||||
for i in memories:
|
for i, n in zip(memories, range(len(memories))):
|
||||||
memory_dt = get_datetime_from_str(i['takenTime'])
|
memory_dt = get_datetime_from_str(i['takenTime'])
|
||||||
if time_span[0] <= memory_dt <= time_span[1]:
|
if time_span[0] <= memory_dt <= time_span[1]:
|
||||||
|
verbose_msg("\nExport BeReal nr %s:" % n)
|
||||||
apply_memory_on_imgs(i, memory_dt)
|
apply_memory_on_imgs(i, memory_dt)
|
||||||
|
|
||||||
|
|
||||||
@@ -93,11 +105,14 @@ if __name__ == '__main__':
|
|||||||
init_global_var(args)
|
init_global_var(args)
|
||||||
|
|
||||||
if not os.path.exists(out_path):
|
if not os.path.exists(out_path):
|
||||||
|
verbose_msg("Create %s folder for output" % out_path)
|
||||||
os.makedirs(out_path)
|
os.makedirs(out_path)
|
||||||
|
|
||||||
|
verbose_msg("Open memories.json file")
|
||||||
f = open('memories.json')
|
f = open('memories.json')
|
||||||
|
|
||||||
|
|
||||||
|
verbose_msg("Start exporting images")
|
||||||
export_images(json.load(f))
|
export_images(json.load(f))
|
||||||
|
|
||||||
f.close()
|
f.close()
|
||||||
|
Reference in New Issue
Block a user