joshpatra/BeReal-Export-Manager
1
0
Fork 0
mirror of https://github.com/SoPat712/BeReal-Export-Manager.git synced 2025-08-21 18:28:46 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add custom path as argument function

Browse Source
This commit is contained in:
lmluk
2024-09-05 11:36:32 +02:00
parent 8b54a0ca83
commit f09f1f849c
1 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
main.py
View File

@@ -1,4 +1,5 @@
import json import json
import os
from exiftool import ExifToolHelper as et from exiftool import ExifToolHelper as et
from shutil import copy2 as cp from shutil import copy2 as cp
import datetime import datetime
@@ -12,6 +13,7 @@ def init_parser():
"Valid format: 'DD.MM.YYYY-DD.MM.YYYY'\n"\ "Valid format: 'DD.MM.YYYY-DD.MM.YYYY'\n"\
"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)")
args = parser.parse_args() args = parser.parse_args()
if args.year and args.timespan: if args.year and args.timespan:
@@ -22,6 +24,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
if args.timespan: if args.timespan:
temp_times = args.timespan.strip().split("-") temp_times = args.timespan.strip().split("-")
@@ -30,6 +33,11 @@ def init_global_var(args: argparse.Namespace):
elif args.year: elif args.year:
time_span = (dt(args.year, 1, 1), dt(args.year, 12, 31)) time_span = (dt(args.year, 1, 1), dt(args.year, 12, 31))
if args.path:
out_path = args.path.strip().removesuffix('/')
else:
out_path = "./out"
def get_img_filename(image: json): def get_img_filename(image: json):
@@ -47,15 +55,15 @@ def get_datetime_from_str(time: str):
return dt.strptime(time, format_string) return dt.strptime(time, format_string)
def apply_memory_on_imgs(memory: json): 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
""" """
memory_dt = get_datetime_from_str(memory['takenTime']) img_names = ["%s/%s_%s.webp" % (out_path, memory_dt.strftime('%Y-%m-%d_%H-%M-%S'), temp_times)
img_names = ["./out/%s_%s.webp" % (memory_dt.strftime('%Y-%m-%d_%H-%M-%S'), temp_times) for temp_times in ['front', 'back']] for temp_times in ['front', 'back']]
cp("./Photos/post/%s" % get_img_filename(memory['frontImage']), img_names[0]) for img_type, img_name in zip(['frontImage', 'backImage'], img_names):
cp("./Photos/post/%s" % get_img_filename(memory['backImage']), img_names[1]) cp("./Photos/post/%s" % get_img_filename(memory[img_type]), img_name)
if 'location' in memory: if 'location' in memory:
et().set_tags(img_names, et().set_tags(img_names,
@@ -71,9 +79,10 @@ def apply_memory_on_imgs(memory: json):
def export_images(memories: json): def export_images(memories: json):
for i in memories:
for temp_times in memories: memory_dt = get_datetime_from_str(i['takenTime'])
apply_memory_on_imgs(temp_times) if time_span[0] <= memory_dt <= time_span[1]:
apply_memory_on_imgs(i, memory_dt)
@@ -81,8 +90,12 @@ if __name__ == '__main__':
args = init_parser() args = init_parser()
init_global_var(args) init_global_var(args)
if not os.path.exists(out_path):
os.makedirs(out_path)
f = open('memories.json') f = open('memories.json')
# export_images(json.load(f))
export_images(json.load(f))
f.close() f.close()