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
|
#!/usr/bin/python
# sxsw.py - version 0.2
# Requires: python-imaging, qrencode, gphoto2
# Author: Luke Macken <lmacken@redhat.com>
# License: GPLv3
import os
import Image
import urllib
import simplejson
import subprocess
from uuid import uuid4
from os.path import join, basename, expanduser
# Where to spit out our qrcode, watermarked image, and local html
out = expanduser('~/Desktop/sxsw')
# The watermark to apply to all images
watermark_img = expanduser('~/Desktop/fedora.png')
# This assumes ssh-agent is running so we can do password-less scp
ssh_image_repo = 'fedorapeople.org:~/public_html/sxsw/'
# The public HTTP repository for uploaded images
http_image_repo = 'http://lmacken.fedorapeople.org/sxsw/'
# Size of the qrcode pixels
qrcode_size = 10
# Whether or not to delete the photo after uploading it to the remote server
delete_after_upload = True
# The camera configuration
# Use gphoto2 --list-config and --get-config for more information
gphoto_config = {
'/main/imgsettings/imagesize': 3, # small
'/main/imgsettings/imagequality': 0, # normal
}
class PhotoBooth(object):
def capture_photo(self):
""" Capture a photo and download it from the camera """
filename = join(out, '%s.jpg' % str(uuid4()))
cfg = ['--set-config=%s=%s' % (k, v) for k, v in gphoto_config.items()]
subprocess.call('gphoto2 --auto-detect ' +
'--capture-image-and-download ' +
'--filename="%s" ' % filename +
' '.join(cfg), shell=True)
return filename
def process_image(self, filename):
print "Processing %s..." % filename
print "Applying watermark..."
image = self.watermark(filename)
print "Uploading to remote server..."
url = self.upload(image)
print "Generating QRCode..."
qrcode = self.qrencode(url)
print "Generating TinyURL..."
tiny = self.tinyurl(url)
print "Generating HTML..."
html = self.html_output(url, qrcode, tiny)
subprocess.call('firefox "%s"' % html, shell=True)
print "Done!"
def watermark(self, image):
""" Apply a watermark to an image """
mark = Image.open(watermark_img)
im = Image.open(image)
if im.mode != 'RGBA':
im = im.convert('RGBA')
layer = Image.new('RGBA', im.size, (0,0,0,0))
position = (im.size[0] - mark.size[0], im.size[1] - mark.size[1])
layer.paste(mark, position)
outfile = join(out, basename(image))
Image.composite(layer, im, layer).save(outfile)
return outfile
def upload(self, image):
""" Upload this image to a remote server """
os.system('scp "%s" %s' % (image, ssh_image_repo))
if delete_after_upload:
os.unlink(image)
return http_image_repo + basename(image)
def qrencode(self, url):
""" Generate a QRCode for a given URL """
qrcode = join(out, 'qrcode.png')
os.system('qrencode -s %d -o "%s" %s' % (qrcode_size, qrcode, url))
return qrcode
def tinyurl(self, url):
""" Generate a tinyurl for a given URL """
data = urllib.urlopen("http://json-tinyurl.appspot.com/?url=%s" % url).read()
json = simplejson.loads(data)
if not json['ok']:
print 'ERROR: There was a problem generating a tinyurl'
print json
return url
return json['tinyurl']
def html_output(self, image, qrcode, tinyurl):
""" Output HTML with the image, qrcode, and tinyurl """
html = """
<html>
<center>
<table>
<tr>
<td colspan="2">
<b><a href="%(tinyurl)s">%(tinyurl)s</a></b>
</td>
</tr>
<tr>
<td><img src="%(image)s" border="0"/></td>
<td><img src="%(qrcode)s" border="0"/></td>
</tr>
</table>
</center>
</html>
""" % {'image': image, 'qrcode': qrcode, 'tinyurl': tinyurl}
outfile = join(out, basename(image) + '.html')
output = file(outfile, 'w')
output.write(html)
output.close()
return outfile
if __name__ == "__main__":
photobooth = PhotoBooth()
try:
while True:
raw_input("Press enter to capture photo.")
filename = photobooth.capture_photo()
photobooth.process_image(filename)
except KeyboardInterrupt:
print "\nExiting..."
|
