th3r00t/pyShelf
1
0
Fork 0
mirror of https://github.com/th3r00t/pyShelf.git synced 2026-04-28 01:59:35 -04:00
Code Issues Packages Projects Releases Wiki Activity

Further work on the web server and frontend

Browse Source
This commit is contained in:
Mike
2019-11-06 18:27:16 -05:00
parent 1bf1640156
commit 1e41483c29
5 changed files with 70 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
lib/pyShelf.py
View File

@@ -34,18 +34,35 @@ class RequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
# TODO determine how to include stylesheets
self.send_response(200)
mimetype = mimetypes.guess_type(self.path)
self.send_header('Content-type', mimetype)
if self.path == '/':
self.path = '../static/index.html'
mimetype = 'text/html'
serve_file = open(self.path[1:]).read()
self.send_header('Content-type', mimetype)
self.end_headers()
self.wfile.write(bytes(serve_file, 'utf-8'))
elif self.path.split('.', 1)[1] == 'css':
self.path = '../static' + self.path
mimetype = 'text/css'
serve_file = open(self.path[1:]).read()
self.send_header('Content-type', mimetype)
self.end_headers()
self.wfile.write(bytes(serve_file, 'utf-8'))
elif self.path.endswith('.png'):
self.path = '../static' + self.path
mimetype = 'image/png'
serve_file = open(self.path[1:], 'rb') # Important to rb read binary for images
self.send_header('Content-type', mimetype)
self.end_headers()
self.wfile.write(serve_file.read())
else:
self.send_response(404)
serve_file = "File Not Found"
serve_file = open(self.path[1:]).read()
self.end_headers()
self.wfile.write(bytes(serve_file, 'utf-8'))
mimetype = 'text/html'
self.send_header('Content-type', mimetype)
self.end_headers()
try: serve_file.close()
except Exception: pass
class BookServer:
"""HTTP Frontend"""