scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed preview script

Browse Source
This commit is contained in:
Simone Margaritelli 2019-10-01 18:56:27 +02:00
parent bf90fc3031
commit b502912d7f
1 changed files with 7 additions and 9 deletions

10
scripts/preview.py

@ -4,18 +4,18 @@ import sys
import os import os
import time import time
import argparse import argparse
import random
from http.server import HTTPServer from http.server import HTTPServer
import shutil import shutil
import yaml import yaml
sys.path.insert(0, sys.path.insert(0,
os.path.join(os.path.dirname(os.path.realpath(__file__)), os.path.join(os.path.dirname(os.path.realpath(__file__)),
'../sdcard/rootfs/root/pwnagotchi/scripts/')) '../sdcard/rootfs/root/pwnagotchi/scripts/'))
from pwnagotchi.ui.view import View
from pwnagotchi.ui.display import Display, VideoHandler from pwnagotchi.ui.display import Display, VideoHandler
import core import core
class CustomDisplay(Display): class CustomDisplay(Display):
def _http_serve(self): def _http_serve(self):
@ -28,7 +28,6 @@ class CustomDisplay(Display):
else: else:
core.log("could not get ip of usb0, video server not starting") core.log("could not get ip of usb0, video server not starting")
def _on_view_rendered(self, img): def _on_view_rendered(self, img):
CustomVideoHandler.render(img) CustomVideoHandler.render(img)
@ -107,12 +106,11 @@ def main():
type: {display} type: {display}
video: video:
enabled: true enabled: true
address: "127.0.0.1" address: "0.0.0.0"
port: {port} port: {port}
'''.format(display=args.display, '''.format(display=args.display,
port=args.port, port=args.port,
lang=args.lang), lang=args.lang))
Loader=yaml.FullLoader)
DISPLAY = CustomDisplay(config=CONFIG, state={'name': '%s>' % 'preview'}) DISPLAY = CustomDisplay(config=CONFIG, state={'name': '%s>' % 'preview'})