added the --export-template and --export-dir_template

to export the default template

1 files changed, 43 insertions, 3 deletions

diff --git a/becommands/html.py b/becommands/html.py
index f0998e2..31b97a1 100644
--- a/becommands/html.py
+++ b/becommands/html.py
@@ -50,12 +50,16 @@ def execute(args, manipulate_encodings=True):
         raise cmdutil.UsageError, 'Too many arguments.'
 
     template = options.template
+
     bd = bugdir.BugDir(from_disk=True,
                        manipulate_encodings=manipulate_encodings)
     bd.load_all_bugs()
 
     html_gen = HTMLGen(bd, template=options.template, verbose=options.verbose,
                        title=options.title, index_header=options.index_header )
+    if options.exp_template == True:
+        html_gen.write_default_template(options.exp_dir_template)
+        return
     html_gen.run(options.out_dir)
 
 def get_parser():
@@ -71,7 +75,11 @@ def get_parser():
         help='Set the index page headers (%default)',
         default='BugsEverywhere Bug List')
     parser.add_option('-v', '--verbose',  action='store_true', metavar='verbose', dest='verbose',
-        help='Verbose output, default is no', default=False)
+        help='Verbose output, default is %default', default=False)
+    parser.add_option('-e', '--export-template',  action='store_true', metavar='STRING', dest='exp_template',
+        help='Export the default template into a directory (%default)', default='./templates/default')
+    parser.add_option('-d', '--export-dir_template', metavar='STRING', dest='exp_dir_template',
+        help='Export the default template into a directory (%default)', default='./templates/default')
     return parser
 
 longhelp="""
@@ -110,11 +118,13 @@ class HTMLGen (object):
         if template != None:
             self._load_user_templates()
 
+
+            
     def run(self, out_dir):
         if self.verbose == True:
             print "Creating the html output in %s using templates in %s" \
                 % (out_dir, self.template)
-
+        
         bugs_active = []
         bugs_inactive = []
         bugs = [b for b in self.bd]
@@ -135,6 +145,31 @@ class HTMLGen (object):
         self._write_index_file(bugs_inactive, title=self.title,
                                index_header=self.index_header, bug_type="inactive")
 
+    def write_default_template(self, out_dir):
+        if self.verbose:
+            print "Creating output directories"
+        self.out_dir = os.path.abspath(os.path.expanduser(out_dir))
+        if not os.path.exists(self.out_dir):
+            try:
+                os.mkdir(self.out_dir)
+            except:
+                raise cmdutil.UsageError, "Cannot create output directory '%s'." % self.out_dir
+        if self.verbose:
+            print "Creating css file"            
+        self._write_css_file()
+        if self.verbose:
+            print "Creating index_file.tpl file"        
+        self._write_file(self.index_file, "index_file.tpl")
+        if self.verbose:
+            print "Creating index_bug_entry.tpl file"        
+        self._write_file(self.index_bug_entry, "index_bug_entry.tpl")
+        if self.verbose:
+            print "Creating bug_file.tpl file"
+        self._write_file(self.bug_file, "bug_file.tpl")
+        if self.verbose:
+            print "Creating bug_comment_entry.tpl file"
+        self._write_file(self.bug_comment_entry, "bug_comment_entry.tpl")
+
     def _create_output_directories(self, out_dir):
         if self.verbose:
             print "Creating output directories"
@@ -297,7 +332,12 @@ class HTMLGen (object):
                 f = codecs.open(fullpath, "r", self.encoding)
                 setattr(self, attr, f.read())
                 f.close()
-
+                
+    def _write_file(self, tpl, filename ):
+        f = codecs.open(os.path.join(self.out_dir, filename), "w", self.encoding)
+        f.write(tpl)
+        f.close()
+        
     def _load_default_templates(self):
         self.css_file = """
             body {