def __init__(self, database, dir_path, debug=False):
self._parser = Parser()
self._wrapper = Wrapper()
self._writer = Writer(database)
self._sitemap_generator = SitemapGenerator(database)
self._feeds_generator = FeedsGenerator(database)
self._debug = debug
self._dir_path = dir_path
self._database = database
self._file_path = ""
class FileMonitor(FileSystemEventHandler):
"""
A monitor for listening the event while file changes.
"""
def __init__(self, database, dir_path, debug=False):
self._parser = Parser()
self._wrapper = Wrapper()
self._writer = Writer(database)
self._sitemap_generator = SitemapGenerator(database)
self._feeds_generator = FeedsGenerator(database)
self._debug = debug
self._dir_path = dir_path
self._database = database
self._file_path = ""
def _parse(self, file_path):
page = self._parser.parse(file_path)
if self._debug:
print "Parse:"
print page
print ""
return page
def _wrap(self, page):
metadata = self._wrapper.wrap(page["metadata"])
if self._debug:
print "Wrap:"
print metadata
print ""
page["metadata"] = metadata
return page
def _write(self, file_path, mode, page=None):
self._writer.write(file_path, mode, page)
if self._debug:
print "Write(%s):" % mode
print_database(self._database)
print ""
def _work(self, file_path, mode):
path = file_path.replace(self._dir_path + "/", "")
self._file_path = path
if mode == "delete":
try:
self._write(path, mode)
except Exception:
self._error("Writing error !")
else:
page = None
try:
page = self._parse(path)
except Exception:
self._error("Parsing error !")
return
try:
page = self._wrap(page)
except Exception:
self._error("Wrapping error !")
return
try:
self._write(path, mode, page)
except Exception:
self._error("Writing error !")
try:
self._sitemap_generator.generate()
except Exception:
self._error("Sitemap generating error !")
try:
self._feeds_generator.generate()
except Exception:
self._error("Feeds generating error !")
def on_created(self, event):
path = event.src_path
if not is_markdown_file(path):
return
logger.info("Create: %s" % path)
self._work(path, "update")
def on_deleted(self, event):
path = event.src_path
if not is_markdown_file(path):
return
logger.info("Delete: %s" % path)
self._work(path, "delete")
def on_modified(self, event):
path = event.src_path
if not is_markdown_file(path):
return
logger.info("Modify: %s" % path)
self._work(path, "update")
def on_moved(self, event):
src_path = event.src_path
dst_path = event.dest_path
if not is_markdown_file(src_path) or not is_markdown_file(dst_path):
return
logger.info("Move: %s, %s" % (src_path, dst_path))
self._work(src_path, "delete")
self._work(dst_path, "update")
def _error(self, message):
line = "%s\n%s\nFile: %s" % (message, traceback.format_exc(), self._file_path)
logger.error(line)
if self._debug:
raise
import os
import sys
sys.path.append(os.path.join(os.path.dirname(__file__), "../"))
from pymongo import MongoClient
from feeds_generator import FeedsGenerator
if __name__ == "__main__":
client = MongoClient()
database = client.get_database("test")
client.close()
fg = FeedsGenerator(database, True)
fg.generate()
