diff --git a/web.py b/web.py new file mode 100644 index 0000000..bf15963 --- /dev/null +++ b/web.py @@ -0,0 +1,80 @@ +import os +from http.server import HTTPServer, SimpleHTTPRequestHandler +from io import BytesIO + +class PDFServer: + def __init__(self, directory, port=8000): + """Initialize the server with a directory to serve PDF files and a specified port.""" + self.directory = directory + self.port = port + + def start_server(self): + """Start the HTTP server and serve PDF files.""" + os.chdir(self.directory) # Change to the specified directory + server_address = ('', self.port) + httpd = HTTPServer(server_address, self.PDFRequestHandler) + print(f"Serving PDF files on http://localhost:{self.port}") + httpd.serve_forever() + + class PDFRequestHandler(SimpleHTTPRequestHandler): + """Custom request handler to display a list of PDF files with dark material styling.""" + + def list_directory(self, path): + """Override the directory listing method to only show PDF files with custom styling.""" + try: + file_list = os.listdir(path) + except os.error: + self.send_error(404, "No permission to list directory") + return None + file_list = [f for f in file_list if f.endswith('.pdf')] # Filter only PDF files + file_list.sort(key=lambda a: a.lower()) + + # HTML content with dark and blue-themed Material Design styling + html_content = self.generate_html(file_list) + return self.send_html_response(html_content) + + def generate_html(self, file_list): + """Generate HTML content for displaying the PDF files.""" + html = ''' + + PDF Files + + + +
+

Available PDF Files for Download

+ +
Serving files locally
+
+ + ''' + return html + + def send_html_response(self, html_content): + """Send an HTML response to the client.""" + f = BytesIO() + f.write(html_content.encode('utf-8')) + f.seek(0) + self.send_response(200) + self.send_header("Content-type", "text/html; charset=utf-8") + self.send_header("Content-Length", str(len(f.getvalue()))) + self.end_headers() + return f + +# Usage +if __name__ == "__main__": + directory = '/path/to/your/directory' + server = PDFServer(directory, port=8000) + server.start_server() \ No newline at end of file