mirror of
https://github.com/jayofelony/pwnagotchi.git
synced 2025-07-01 18:37:27 -04:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import _thread
|
|
import secrets
|
|
import logging
|
|
import os
|
|
|
|
# https://stackoverflow.com/questions/14888799/disable-console-messages-in-flask-server
|
|
logging.getLogger('werkzeug').setLevel(logging.ERROR)
|
|
os.environ['WERKZEUG_RUN_MAIN'] = 'true'
|
|
|
|
from flask import Flask
|
|
from flask_cors import CORS
|
|
from flask_wtf.csrf import CSRFProtect
|
|
|
|
from pwnagotchi.ui.web.handler import Handler
|
|
|
|
|
|
class Server:
|
|
def __init__(self, agent, config):
|
|
self._enabled = config['video']['enabled']
|
|
self._port = config['video']['port']
|
|
self._address = config['video']['address']
|
|
self._origin = None
|
|
self._agent = agent
|
|
if 'origin' in config['video']:
|
|
self._origin = config['video']['origin']
|
|
|
|
if self._enabled:
|
|
_thread.start_new_thread(self._http_serve, ())
|
|
|
|
def _http_serve(self):
|
|
if self._address is not None:
|
|
app = Flask(__name__)
|
|
app.secret_key = secrets.token_urlsafe(256)
|
|
|
|
if self._origin:
|
|
CORS(app, resources={r"*": {"origins": self._origin}})
|
|
|
|
CSRFProtect(app)
|
|
Handler(self._agent, app)
|
|
|
|
app.run(host=self._address, port=self._port, debug=False)
|
|
else:
|
|
logging.info("could not get ip of usb0, video server not starting")
|