chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Exception handlers in preview script

Browse Source
This commit is contained in:
Panagiotis Vasilopoulos
2019-10-01 16:41:43 +00:00
parent 19c4b42072
commit 33f9917b90
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/preview.py
View File

@ -43,7 +43,10 @@ class CustomVideoHandler(VideoHandler):
@staticmethod @staticmethod
def render(img): def render(img):
with CustomVideoHandler._lock: with CustomVideoHandler._lock:
img.save("/tmp/pwnagotchi-{rand}.png".format(rand=id(CustomVideoHandler)), format='PNG') try:
img.save("/tmp/pwnagotchi-{rand}.png".format(rand=id(CustomVideoHandler)), format='PNG')
except BaseException:
core.log("could not write preview")
def do_GET(self): def do_GET(self):
if self.path == '/': if self.path == '/':
@ -67,7 +70,7 @@ class CustomVideoHandler(VideoHandler):
with open("/tmp/pwnagotchi-{rand}.png".format(rand=id(CustomVideoHandler)), 'rb') as fp: with open("/tmp/pwnagotchi-{rand}.png".format(rand=id(CustomVideoHandler)), 'rb') as fp:
shutil.copyfileobj(fp, self.wfile) shutil.copyfileobj(fp, self.wfile)
except BaseException: except BaseException:
pass core.log("could not open preview")
else: else:
self.send_response(404) self.send_response(404)