Skip to content

Commit

Permalink
replace ' in server.py
Browse files Browse the repository at this point in the history
  • Loading branch information
@back-to
back-to committed Jun 17, 2022
1 parent 3c59fd8 commit 802852b
Showing 1 changed file with 36 additions and 40 deletions.
76 changes: 36 additions & 40 deletions liveproxy/server.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
from shutil import which
from socketserver import ThreadingMixIn
from time import time
from urllib.parse import parse_qsl, unquote, urlparse

ACCEPTABLE_ERRNO = (
errno.ECONNABORTED,
Expand All @@ -24,10 +23,10 @@
except AttributeError:
pass # Not windows

_re_streamlink = re.compile(r'streamlink', re.IGNORECASE)
_re_youtube_dl = re.compile(r'(?:youtube|yt)[_-]dl(?:p)?', re.IGNORECASE)
_re_streamlink = re.compile(r"streamlink", re.IGNORECASE)
_re_youtube_dl = re.compile(r"(?:youtube|yt)[_-]dl(?:p)?", re.IGNORECASE)

log = logging.getLogger(__name__.replace('liveproxy.', ''))
log = logging.getLogger(__name__.replace("liveproxy.", ""))


class HTTPRequest(BaseHTTPRequestHandler):
Expand All @@ -37,67 +36,67 @@ def log_message(self, format, *args):

def _headers(self, status, content, connection=False):
self.send_response(status)
self.send_header('Server', 'LiveProxy')
self.send_header('Content-type', content)
self.send_header("Server", "LiveProxy")
self.send_header("Content-type", content)
if connection:
self.send_header('Connection', connection)
self.send_header("Connection", connection)
self.end_headers()

def do_HEAD(self):
'''Respond to a HEAD request.'''
self._headers(404, 'text/html', connection='close')
"""Respond to a HEAD request."""
self._headers(404, "text/html", connection="close")

def do_GET(self):
'''Respond to a GET request.'''
"""Respond to a GET request."""
random_id = hex(int(time()))[5:]
log = logging.getLogger('{name}.{random_id}'.format(
name=__name__.replace('liveproxy.', ''),
log = logging.getLogger("{name}.{random_id}".format(
name=__name__.replace("liveproxy.", ""),
random_id=random_id,
))

log.info(f'User-Agent: {self.headers.get("User-Agent", "???")}')
log.info(f'Client: {self.client_address}')
log.info(f'Address: {self.address_string()}')
log.info(f"User-Agent: {self.headers.get('User-Agent', '???')}")
log.info(f"Client: {self.client_address}")
log.info(f"Address: {self.address_string()}")

if self.path.startswith(('/base64/')):
if self.path.startswith(("/base64/")):
# http://127.0.0.1:53422/base64/STREAMLINK-COMMANDS/
# http://127.0.0.1:53422/base64/YOUTUBE-DL-COMMANDS/
# http://127.0.0.1:53422/base64/YT-DLP-COMMANDS/
try:
arglist = shlex.split(base64.urlsafe_b64decode(self.path.split('/')[2]).decode('UTF-8'))
arglist = shlex.split(base64.urlsafe_b64decode(self.path.split("/")[2]).decode("UTF-8"))
except base64.binascii.Error as err:
log.error(f'invalid base64 URL: {err}')
self._headers(404, 'text/html', connection='close')
log.error(f"invalid base64 URL: {err}")
self._headers(404, "text/html", connection="close")
return
else:
self._headers(404, 'text/html', connection='close')
self._headers(404, "text/html", connection="close")
return

prog = which(arglist[0], mode=os.F_OK | os.X_OK)
if not prog:
log.error(f'invalid prog, can not find "{arglist[0]}" on your system')
log.error(f"invalid prog, can not find '{arglist[0]}' on your system")
return

log.debug(f'Video-Software: {prog}')
log.debug(f"Video-Software: {prog}")
if _re_streamlink.search(prog):
arglist.extend(['--stdout', '--loglevel', 'none'])
arglist.extend(["--stdout", "--loglevel", "none"])
elif _re_youtube_dl.search(prog):
arglist.extend(['-o', '-', '--quiet', '--no-playlist', '--no-warnings', '--no-progress'])
arglist.extend(["-o", "-", "--quiet", "--no-playlist", "--no-warnings", "--no-progress"])
else:
log.error('Video-Software is not supported.')
self._headers(404, 'text/html', connection='close')
log.error("Video-Software is not supported.")
self._headers(404, "text/html", connection="close")
return

log.debug(f'{arglist!r}')
self._headers(200, 'video/unknown')
log.debug(f"{arglist!r}")
self._headers(200, "video/unknown")
process = subprocess.Popen(arglist,
stderr=subprocess.PIPE,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=False,
)

log.info(f'Stream started {random_id}')
log.info(f"Stream started {random_id}")
try:
while True:
read = process.stdout.readline()
Expand All @@ -109,26 +108,23 @@ def do_GET(self):
break
except socket.error as e:
if isinstance(e.args, tuple):
if e.errno == errno.EPIPE:
# remote peer disconnected
log.info('Detected remote disconnect')
else:
log.error(f'E1: {e!r}')
if not e.errno in ACCEPTABLE_ERRNO:
log.error(f"E1: {e!r}")
else:
log.error(f'E2: {e!r}')
log.error(f"E2: {e!r}")

log.info(f'Stream ended {random_id}')
log.info(f"Stream ended {random_id}")
process.terminate()
process.wait()
process.kill()


class Server(HTTPServer):
'''HTTPServer class with timeout.'''
"""HTTPServer class with timeout."""
timeout = 5

def finish_request(self, request, client_address):
'''Finish one request by instantiating RequestHandlerClass.'''
"""Finish one request by instantiating RequestHandlerClass."""
try:
self.RequestHandlerClass(request, client_address, self)
except ValueError:
Expand All @@ -139,9 +135,9 @@ def finish_request(self, request, client_address):


class ThreadedHTTPServer(ThreadingMixIn, Server):
'''Handle requests in a separate thread.'''
"""Handle requests in a separate thread."""
allow_reuse_address = True
daemon_threads = True


__all__ = ('HTTPRequest', 'ThreadedHTTPServer')
__all__ = ("HTTPRequest", "ThreadedHTTPServer")

0 comments on commit 802852b

Please sign in to comment.