Add hugo-new-photo.py to create a photo post from a list of photos

hugo-new-photo.py

@@ -0,0 +1,76 @@
+#!env/bin/python3
+# Eryn Wells <eryn@erynwells.me>
+
+'''
+New script.
+'''
+
+import argparse
+import datetime
+import os.path
+import shutil
+import subprocess
+from PIL import Image
+from PIL.ExifTags import TAGS
+
+PHOTOS_CONTENT_DIR = 'content/photos'
+
+def parse_args(argv, *a, **kw):
+    parser = argparse.ArgumentParser(*a, **kw)
+    parser.add_argument('-n', '--dry-run', action='store_true')
+    parser.add_argument('-t', '--title', type=str)
+    parser.add_argument('photos', nargs='+')
+    args = parser.parse_args(argv)
+    return args
+
+def main(argv):
+    args = parse_args(argv[1:], prog=argv[0])
+
+    earliest_exif_date = None
+
+    for photo in args.photos:
+        try:
+            image = Image.open(photo)
+        except IOError:
+            pass
+
+        raw_exif = image._getexif()
+        friendly_exif = {TAGS[k]: v for k, v in raw_exif.items() if k in TAGS}
+
+        date_string = f'{friendly_exif["DateTime"]} {friendly_exif["OffsetTime"]}'
+        exif_date = datetime.datetime.strptime(date_string, '%Y:%m:%d %H:%M:%S %z')
+        print(f'{photo} -> {exif_date.isoformat()}')
+
+        if not earliest_exif_date or exif_date < earliest_exif_date:
+            earliest_exif_date = exif_date
+
+    year = earliest_exif_date.year
+    month = earliest_exif_date.month
+
+    name = args.title if args.title else os.path.splitext(os.path.basename(photo))[0]
+    post_path = os.path.join(PHOTOS_CONTENT_DIR, str(year), str(month), name)
+
+    try:
+        hugo_command = ['hugo', 'new', '--clock', earliest_exif_date.isoformat(), post_path]
+        if not args.dry_run:
+            result = subprocess.run(hugo_command)
+            result.check_returncode()
+        else:
+            print(' '.join(hugo_command))
+    except subprocess.CalledProcessError:
+        print(f'Failed to create new Hugo post', file=sys.stderr)
+        return -1
+
+    for photo in args.photos:
+        print(f'Copy {photo} -> {post_path}')
+        try:
+            if not args.dry_run:
+                shutil.copy(photo, post_path)
+        except IOError:
+            print(f'Failed to copy {photo}', file=sys.stderr)
+            return -2
+
+if __name__ == '__main__':
+    import sys
+    result = main(sys.argv)
+    sys.exit(0 if not result else result)