diff options
Diffstat (limited to 'host/gifserver.py')
-rwxr-xr-x | host/gifserver.py | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/host/gifserver.py b/host/gifserver.py new file mode 100755 index 0000000..7fbbfe3 --- /dev/null +++ b/host/gifserver.py @@ -0,0 +1,80 @@ +#!/usr/bin/env python + +import argparse, time, argparse, textwrap, threading, io +from urllib.parse import urlparse +from PIL import Image, GifImagePlugin, ImageSequence, ImageOps +from flask import Flask, request + +import crap + +import config + + +gif = None +gif_cond = threading.Condition() + +def putgif(filelike): + global gif + with gif_cond: + gif = filelike + gif_cond.notify() + + +def sendgifs(target): + thost, tport = target.split(':') + crapthing = crap.CRAPClient(thost, int(tport)) + last_gif = None + + while True: + with gif_cond: + gif_cond.wait_for(lambda: last_gif != gif) + + last_gif = gif + img = Image.open(last_gif) + palette = img.getpalette() + last_frame = Image.new("RGBA", img.size) + sleepiness = min(img.info.get('duration', 100) / 1000.0, 10.0) + + for frame in ImageSequence.Iterator(img): + #This works around a known bug in Pillow + #See also: http://stackoverflow.com/questions/4904940/python-converting-gif-frames-to-png + frame.putpalette(palette) + c = frame.convert("RGBA") + + if 'background' in img.info and 'transparency' in img.info: + last_frame.paste(c, c) + else: + last_frame = c + + im = last_frame.copy() + im = ImageOps.fit(im, (config.display_width, config.display_height), Image.NEAREST, centering=(0.5, 0.5)) + crapthing.sendframe(im.tobytes()) + time.sleep(sleepiness) + if last_gif != gif: + break + + +app = Flask(__name__) +#app.debug = True + +@app.route('/', methods=('POST', 'GET')) +def index(): + if request.method == 'GET': + return textwrap.dedent(''' + <form action="#" method="POST" enctype="multipart/form-data"> + <input name="gif" type="file" accept="image/gif"> + <input type="submit" value="upload"> + </form>''') + else: #POST + putgif(io.BytesIO(request.files['gif'].read())) + return 'Done, thanks!' + +if __name__ == '__main__': + parser = argparse.ArgumentParser(description='Present a web interface to stream .gif files via CRAP to a matelight') + parser.add_argument('target', type=str, default='localhost:1337', nargs='?', help='Target matelight, host:port notation') + args = parser.parse_args() + + gif_sender = threading.Thread(target=sendgifs, args=(args.target,), daemon=True) + gif_sender.start() + app.run() + |