From 501f27cc4f55f2d4f96513068ed1c6423935a7ad Mon Sep 17 00:00:00 2001 From: LongHairedHacker Date: Thu, 21 Jan 2016 22:24:32 +0100 Subject: [PATCH] Added a serve parameter for easier testing --- verdandi/constants.py | 2 ++ verdandi/verdandi.py | 17 ++++++++++++++--- 2 files changed, 16 insertions(+), 3 deletions(-) diff --git a/verdandi/constants.py b/verdandi/constants.py index 68cc2a7..ff92f5d 100644 --- a/verdandi/constants.py +++ b/verdandi/constants.py @@ -3,3 +3,5 @@ OUTPUT_DIRECTORY = "rendered_root" TEMPLATE_DIRECTORY = "templates" CONTENT_DIRECTORY = "content" + +SERVE_PORT = 8000 diff --git a/verdandi/verdandi.py b/verdandi/verdandi.py index 6ad7b7e..b2db4f0 100644 --- a/verdandi/verdandi.py +++ b/verdandi/verdandi.py @@ -1,9 +1,15 @@ #!/usr/bin/env python2 import os -from jinja2 import Environment, FileSystemLoader +import sys +import SocketServer -from constants import OUTPUT_DIRECTORY, TEMPLATE_DIRECTORY +from jinja2 import Environment, FileSystemLoader +from SimpleHTTPServer import SimpleHTTPRequestHandler + + + +from constants import OUTPUT_DIRECTORY, TEMPLATE_DIRECTORY, SERVE_PORT class Verdandi(object): @@ -33,7 +39,7 @@ class Verdandi(object): messages = messages + self.send_message(message) - def collect_assets(self): + def collect_assets(self): for module in self.modules: module.collect_assets(self.output_directory) @@ -49,3 +55,8 @@ class Verdandi(object): self.pass_messages() self.collect_assets() self.render() + + if len(sys.argv) > 1 and sys.argv[1] == 'serve': + os.chdir(self.output_directory) + httpd = SocketServer.TCPServer(("", SERVE_PORT), SimpleHTTPRequestHandler) + httpd.serve_forever()