Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2025-01-31 11:41:35 +01:00
Code Issues Releases Wiki Activity

update server code

Browse Source 
- favicon (minimal 1x1 pixel GIF)
- multithreading (one new thread per connection; improvable)
- accepts '/api/json/N' paths
This commit is contained in:
Mike Fährmann 2019-01-15 16:10:57 +01:00
parent 8662e72bdd
commit e80be1bd5c
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 26 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
gallery_dl/server.py
View File

@ -6,20 +6,33 @@
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
from http.server import HTTPServer, BaseHTTPRequestHandler
from . import extractor, config, text
import socketserver
import http.server
import json
class RequestHandler(BaseHTTPRequestHandler):
FAVICON = (b"GIF87a\x01\x00\x01\x00\x80\x00\x00\xff\xff\xff\xff\xff\xff,"
b"\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;")
class HttpServer(socketserver.ThreadingMixIn, http.server.HTTPServer):
daemon_threads = True
class RequestHandler(http.server.BaseHTTPRequestHandler):
def do_GET(self):
"""Handle GET requests"""
path, _, query = self.path.partition("?")
args = text.parse_query(query)
if path == "/api/json":
if path == "/api/json" or path.startswith("/api/json/"):
self._send_json(self.api_json(args))
elif path == "/favicon.ico":
self._send_data(FAVICON, "image/gif", 200)
else:
self._send_json({"error": "Not Found"}, 404)
@ -45,18 +58,23 @@ class RequestHandler(BaseHTTPRequestHandler):
return url, data, None
def _send_data(self, data, mime, status=200):
self.send_response(status)
self.send_header("Content-Type", mime)
self.send_header("Content-Length", str(len(data)))
self.end_headers()
self.wfile.write(data)
def _send_json(self, data, status=200):
"""Send 'data' as JSON response"""
self.send_response(status)
self.send_header("Content-Type", "application/json")
self.end_headers()
self.wfile.write(json.dumps(data, separators=(',', ':')).encode())
data = json.dumps(data, separators=(',', ':')).encode()
self._send_data(data, "application/json", status)
def run():
bind = config.get(("server", "bind"), "127.0.0.1")
port = config.get(("server", "port"), 6412)
httpd = HTTPServer((bind, port), RequestHandler)
httpd = HttpServer((bind, port), RequestHandler)
try:
httpd.serve_forever()