From f4117ad096a38eaa9d8d01859d71437cb3970238 Mon Sep 17 00:00:00 2001
From: Blake Blackshear <blakeb@blakeshome.com>
Date: Tue, 24 Nov 2020 20:36:56 -0600
Subject: [PATCH] add endpoint for event thumbnail

---
 frigate/http.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/frigate/http.py b/frigate/http.py
index df04c6148..d49f2f44d 100644
--- a/frigate/http.py
+++ b/frigate/http.py
@@ -1,3 +1,4 @@
+import base64
 import datetime
 import logging
 import os
@@ -8,7 +9,7 @@ import cv2
 import numpy as np
 from flask import (Blueprint, Flask, Response, current_app, jsonify,
                    make_response, request)
-from peewee import SqliteDatabase, operator, fn
+from peewee import SqliteDatabase, operator, fn, DoesNotExist
 from playhouse.shortcuts import model_to_dict
 
 from frigate.models import Event
@@ -65,7 +66,20 @@ def events_summary():
 
 @bp.route('/events/<id>')
 def event(id):
-    return model_to_dict(Event.get(Event.id == id))
+    try:
+        return model_to_dict(Event.get(Event.id == id))
+    except DoesNotExist:
+        return "Event not found", 404
+
+@bp.route('/events/<id>/snapshot.jpg')
+def event_snapshot(id):
+    try:
+        event = Event.get(Event.id == id)
+        response = make_response(base64.b64decode(event.thumbnail))
+        response.headers['Content-Type'] = 'image/jpg'
+        return response
+    except DoesNotExist:
+        return "Event not found", 404
 
 @bp.route('/events')
 def events():