122 lines
2.8 KiB
Python
122 lines
2.8 KiB
Python
#!/usr/bin/env python2
|
|
|
|
import os
|
|
import sys
|
|
import SocketServer
|
|
import traceback
|
|
|
|
from time import sleep
|
|
from jinja2 import Environment, FileSystemLoader
|
|
from SimpleHTTPServer import SimpleHTTPRequestHandler
|
|
from multiprocessing import Process
|
|
from watchdog.observers import Observer
|
|
from watchdog.events import FileSystemEventHandler
|
|
|
|
from constants import CONTENT_DIRECTORY, OUTPUT_DIRECTORY, TEMPLATE_DIRECTORY, SERVE_PORT, SERVE_BIND_ADDRESS
|
|
|
|
|
|
def serve(output_directory):
|
|
os.chdir(output_directory)
|
|
httpd = SocketServer.TCPServer((SERVE_BIND_ADDRESS, SERVE_PORT), SimpleHTTPRequestHandler)
|
|
print "Serving under %s:%d" % (SERVE_BIND_ADDRESS, SERVE_PORT)
|
|
httpd.serve_forever()
|
|
|
|
|
|
class DirectoryObserver(FileSystemEventHandler):
|
|
def __init__(self, verdandi):
|
|
self._verdandi = verdandi
|
|
|
|
def on_any_event(self, event):
|
|
print "File changed: %s" % event.src_path
|
|
self._verdandi.generate_output(False)
|
|
|
|
|
|
|
|
class Verdandi(object):
|
|
|
|
output_directory = OUTPUT_DIRECTORY
|
|
modules = []
|
|
base_url = ""
|
|
|
|
|
|
def __init__(self):
|
|
self.jinja_env = Environment(loader=FileSystemLoader(TEMPLATE_DIRECTORY))
|
|
self.jinja_env.globals['base_url'] = self.base_url
|
|
|
|
|
|
def send_message(self, message):
|
|
results = []
|
|
|
|
for module in self.modules:
|
|
results += module.process_message(message)
|
|
|
|
return results
|
|
|
|
|
|
def pass_messages(self):
|
|
messages = self.send_message(None)
|
|
|
|
while len(messages) > 0:
|
|
message = messages.pop()
|
|
messages = messages + self.send_message(message)
|
|
|
|
|
|
def collect_assets(self):
|
|
if not os.path.exists(self.output_directory):
|
|
os.mkdir(self.output_directory)
|
|
if os.listdir(self.output_directory) != []:
|
|
print "[Warn] Ouput directory is not empty"
|
|
for module in self.modules:
|
|
module.collect_assets(self.output_directory)
|
|
|
|
|
|
def render(self):
|
|
if not os.path.exists(self.output_directory):
|
|
os.makedirs(self.output_directory)
|
|
for module in self.modules:
|
|
module.render(self.output_directory, self.jinja_env)
|
|
|
|
|
|
def run(self):
|
|
if len(sys.argv) > 1 and sys.argv[1] == 'serve':
|
|
self.generate_output(False)
|
|
self.serve()
|
|
else:
|
|
self.generate_output()
|
|
sys.exit(0)
|
|
|
|
|
|
def generate_output(self, fail_on_exception = True):
|
|
try:
|
|
self.pass_messages()
|
|
self.collect_assets()
|
|
self.render()
|
|
except Exception:
|
|
print "[Error] I have a bad feeling about this ..."
|
|
traceback.print_exc()
|
|
if fail_on_exception:
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
def serve(self):
|
|
event_handler = DirectoryObserver(self)
|
|
observer = Observer()
|
|
observer.schedule(event_handler, CONTENT_DIRECTORY, recursive=True)
|
|
observer.schedule(event_handler, TEMPLATE_DIRECTORY, recursive=True)
|
|
observer.start()
|
|
|
|
server_process = Process(target=serve, args=(self.output_directory,))
|
|
server_process.start()
|
|
|
|
try:
|
|
server_process.join()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
observer.stop()
|
|
observer.join()
|
|
|
|
print "Be vigilant!"
|
|
sys.exit(0)
|