-
Notifications
You must be signed in to change notification settings - Fork 0
/
process_photos.py
178 lines (146 loc) · 5.45 KB
/
process_photos.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
import base64
import datetime
import io
import os
import pathlib
import sys
import exifread
from PIL import Image
from multiprocessing.pool import ThreadPool
class Photo ():
"""Intermediate presentation for generating javascript output"""
def __init__(self,
js_variable_name: str,
src: str,
title: str,
slug: str,
width: int,
height: int,
created_at: datetime.datetime,
blur_data_url: str,
image_artist: str = None,
lat: float = None,
lon: float = None,
previous_photo_slug: str = None,
next_photo_slug: str = None
) -> None:
self.js_variable_name = js_variable_name
self.src = src
self.title = title
self.slug = slug
self.width = width
self.height = height
self.created_at = created_at
self.blur_data_url = blur_data_url
self.image_artist = image_artist
self.lat = lat
self.lon = lon
self.previous_photo_slug = previous_photo_slug
self.next_photo_slug = next_photo_slug
photos_dir = pathlib.Path('./public/photos')
def parse_image(path: pathlib.Path) -> Photo:
# print(path)
relative_file_name = str(path)[len(str(photos_dir)) + 1:]
js_variable_name = relative_file_name[:-4].translate(''.maketrans({
' ': '_',
'.': '_',
'-': '_',
')': '_',
'(': '_',
'/': '__',
'ä': 'ae',
'Ä': 'Ae',
'ü': 'ue',
'Ü': 'Ue',
'ö': 'oe',
'Ö': 'Oe'
}
))
if js_variable_name[0].isdigit():
js_variable_name = '_' + js_variable_name
slug = path.stem.replace(' ', '-').lower()
with path.open('rb') as f:
exif_data = exifread.process_file(f, details=False)
image_artist = exif_data.get('Image Artist')
if exif_date := exif_data.get('EXIF DateTimeOriginal'):
timestamp = datetime.datetime.strptime(
str(exif_date), '%Y:%m:%d %H:%M:%S')
else:
timestamp = datetime.datetime.fromtimestamp(
os.stat(path).st_birthtime)
def parse_exif_gps_value(exif_field):
if exif_gps_value := exif_data.get(exif_field):
degrees, minutes, seconds = exif_gps_value.values
return degrees + minutes/60.0 + seconds/3600.0
else:
return None
lat, lon = parse_exif_gps_value(
'GPS GPSLatitude'), parse_exif_gps_value('GPS GPSLongitude')
with Image.open(path) as image:
# extract diments
width = image.width
height = image.height
# create review image as base64 encoded data url
image.thumbnail(size=(30, 30))
# image.show()
buffered = io.BytesIO()
image.save(buffered, format="PNG")
blur_data_url = 'data:image/jpeg;base64,' + \
base64.b64encode(buffered.getvalue()).decode("utf-8")
title = path.stem
if title.split('_')[-1].isdigit():
title = ''.join(title.split('_')[:-1])
return Photo(js_variable_name=js_variable_name,
src=f'/photos/{relative_file_name}',
title=title,
slug=slug,
width=width,
height=height,
created_at=timestamp,
blur_data_url=blur_data_url,
image_artist=image_artist,
lat=lat,
lon=lon)
photo_paths = list(photos_dir.glob('**/*.[jJ][pP][gG]'))
with ThreadPool(16) as pool:
photos = pool.map(parse_image, photo_paths)
# sort by date
photos.sort(key=lambda p: p.created_at, reverse=True)
# assign previous and next photo
for i, photo in enumerate(photos):
if i > 0:
photo.previous_photo_slug = photos[i-1].slug
if i < len(photos) - 1:
photo.next_photo_slug = photos[i+1].slug
# write javascript file
with open('./app/(photos)/processedPhotos.tsx', 'w') as output:
# output = sys.stdout
print('''
// this file is automatically generated by running `python3 process_images.py`
export class PhotoProps{
constructor(
public src: string,
public title: string,
public slug: string,
public width: number,
public height: number,
public createdAt: Date,
public blurDataUrl: string,
public photo_artist: string,
public lat: number | undefined,
public lon: number | undefined,
public slugPreviousPhoto: string | undefined,
public slugNextPhoto: string | undefined)
{}
}
export const photos = {''', file=output)
for photo in photos:
print(f''' {photo.js_variable_name}: new PhotoProps('{photo.src}', '{photo.title}', '{photo.slug}', {photo.width}, {photo.height}, new Date('{photo.created_at}'), '{photo.blur_data_url}', '{photo.image_artist}', {photo.lat or 'undefined'}, {photo.lon or 'undefined'}, {"'" + photo.previous_photo_slug + "'" if photo.previous_photo_slug else 'undefined'}, {"'" + photo.next_photo_slug + "'" if photo.next_photo_slug else 'undefined'}),''', file=output)
print(f'''}}
export default photos
export const photosBySlug: {{[Key: string]: PhotoProps}} = {{
''', file=output)
for photo in photos:
print(
f''' '{photo.slug}': photos.{photo.js_variable_name},''', file=output)
print(f'''}}''', file=output)