4e8d143774
Con un handler HTTP que también gestione WebSockets podemos gestionar los WebSockets en el mismo puerto HTTP, así no es necesario usar / abrir dos puertos. Para eso he usado el handler creado por SevenW en: https://github.com/SevenW/httpwebsockethandler y que extiende el SimpleHTTPServer. Aunque ha sido ligeralmente modificado: - He creado un wrapper para las peticiones HTTP GET, el antiguo do_GET ahora es do_GET_HTTP - He desactivado una parte de autenticación que tenía (ni he mirado en intentar mantelerlo, aunque si mantenemos el del do_GET_HTTP, pero supongo que también debería protegerse el de las querys WS y seguramente sería muy muy sencillo reactivarle esa funcionaliad y quitarle el comentario) En el server mantenemos el antiguo do_GET pero renombrado a do_GET_HTTP, y se han movido los métodos del WebSocketServer a este handler. He quitado las cosas que configuran un puerto WebSocket ya que ahora es el mismo que HTTP. Había pensado en enviarlo en otra carpetita (mediaserver-alt) o algo así, por si acaso usar lo otro tiene algún setido (como que no se usen ciertas clases por compatibilidad con pythons antiguos), pero bueno, por ahora lo dejo aquí, siempre se puede hacer rollback + crear una nueva carpeta.
104 lines
4.4 KiB
Python
104 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# ------------------------------------------------------------
|
|
# Controlador para acceso a archivos locales
|
|
# ------------------------------------------------------------
|
|
import os
|
|
import re
|
|
import time
|
|
|
|
from controller import Controller
|
|
from platformcode import config, logger
|
|
|
|
|
|
class fileserver(Controller):
|
|
pattern = re.compile("^/(?:media/.*?)?(?:local/.*?)?$")
|
|
|
|
def run(self, path):
|
|
if path == "/":
|
|
f = open(os.path.join(config.get_runtime_path(), "platformcode", "template", "page.html"), "rb")
|
|
self.handler.send_response(200)
|
|
self.handler.send_header('Content-type', 'text/html')
|
|
self.handler.end_headers()
|
|
respuesta = f.read()
|
|
self.handler.wfile.write(respuesta)
|
|
f.close()
|
|
|
|
elif path.startswith("/local/"):
|
|
import base64
|
|
import urllib
|
|
Path = path.replace("/local/", "").split("/")[0]
|
|
Path = base64.b64decode(urllib.unquote_plus(Path))
|
|
Size = int(os.path.getsize(Path.decode("utf8")))
|
|
f = open(Path.decode("utf8"), "rb")
|
|
if not self.handler.headers.get("range") == None:
|
|
if "=" in str(self.handler.headers.get("range")) and "-" in str(self.handler.headers.get("range")):
|
|
Inicio = int(self.handler.headers.get("range").split("=")[1].split("-")[0])
|
|
if self.handler.headers.get("range").split("=")[1].split("-")[1] <> "":
|
|
Fin = int(self.handler.headers.get("range").split("=")[1].split("-")[1])
|
|
else:
|
|
Fin = Size - 1
|
|
|
|
else:
|
|
Inicio = 0
|
|
Fin = Size - 1
|
|
|
|
if not Fin > Inicio: Fin = Size - 1
|
|
|
|
if self.handler.headers.get("range") == None:
|
|
logger.info("-------------------------------------------------------")
|
|
logger.info("Solicitando archivo local: " + Path)
|
|
logger.info("-------------------------------------------------------")
|
|
|
|
self.handler.send_response(200)
|
|
self.handler.send_header("Content-Disposition", "attachment; filename=video.mp4")
|
|
self.handler.send_header('Accept-Ranges', 'bytes')
|
|
self.handler.send_header('Content-Length', str(Size))
|
|
self.handler.send_header("Connection", "close")
|
|
self.handler.end_headers()
|
|
while True:
|
|
time.sleep(0.2)
|
|
buffer = f.read(1024 * 250)
|
|
if not buffer:
|
|
break
|
|
self.handler.wfile.write(buffer)
|
|
self.handler.wfile.close()
|
|
f.close()
|
|
else:
|
|
logger.info("-------------------------------------------------------")
|
|
logger.info("Solicitando archivo local: " + Path)
|
|
logger.info("Rango: " + str(Inicio) + "-" + str(Fin) + "/" + str(Size))
|
|
logger.info("-------------------------------------------------------")
|
|
f.seek(Inicio)
|
|
|
|
self.handler.send_response(206)
|
|
self.handler.send_header("Content-Disposition", "attachment; filename=video.mp4")
|
|
self.handler.send_header('Accept-Ranges', 'bytes')
|
|
self.handler.send_header('Content-Length', str(Fin - Inicio))
|
|
self.handler.send_header('Content-Range', str(Inicio) + "-" + str(Fin) + "/" + str(Size))
|
|
self.handler.send_header("Connection", "close")
|
|
|
|
self.handler.end_headers()
|
|
while True:
|
|
time.sleep(0.2)
|
|
buffer = f.read(1024 * 250)
|
|
if not buffer:
|
|
break
|
|
self.handler.wfile.write(buffer)
|
|
self.handler.wfile.close()
|
|
f.close()
|
|
elif path.startswith("/media/"):
|
|
file = os.path.join(config.get_runtime_path(), "platformcode", "template", path[7:])
|
|
from mimetypes import MimeTypes
|
|
mime = MimeTypes()
|
|
mime_type = mime.guess_type(file)
|
|
try:
|
|
mim = mime_type[0]
|
|
except:
|
|
mim = ""
|
|
f = open(file, "rb")
|
|
self.handler.send_response(200)
|
|
self.handler.send_header('Content-type', mim)
|
|
self.handler.end_headers()
|
|
self.handler.wfile.write(f.read())
|
|
f.close()
|