g2h: Add utility script

Also move the sources into the more conventional src/g2h dir

1 files changed, 30 insertions, 0 deletions

diff --git a/src/g2h/page_factory.py b/src/g2h/page_factory.py
new file mode 100644
index 0000000..4a46f14
--- /dev/null
+++ b/src/g2h/page_factory.py
@@ -0,0 +1,30 @@
+import os
+
+from .commit_page import CommitPage
+from .file_page import FilePage
+from .overview_page import OverviewPage
+from .log_page import LogPage
+from .tree_page import TreePage
+
+class PageFactory:
+    def __init__(self, out_dir, name, content_subdir):
+        self.out_dir = out_dir
+        self.name = name
+        self.content_subdir = content_subdir
+
+    def new_commit(self, commit):
+        return CommitPage(self.out_dir, self.name, commit, self.content_subdir)
+
+    def new_file(self, commit, path):
+        filename = os.path.basename(path)
+        dir = os.path.dirname(path)
+        return FilePage(self.out_dir, self.name, commit, filename, dir, self.content_subdir)
+
+    def new_overview(self, branches, tags, head):
+        return OverviewPage(self.out_dir, self.name, branches, tags, self.content_subdir, head)
+
+    def new_log(self, commits, branches, tags, head):
+        return LogPage(self.out_dir, self.name, commits, branches, tags, self.content_subdir, head)
+
+    def new_tree(self, commit, tree_segment, repo):
+        return TreePage(self.out_dir, self.name, commit, tree_segment, repo, self.content_subdir)