Fixed #103

Add the --template_location to pass the user custom's template for attribution generation.

Author: chinyeungli

README.rst

@@ -177,6 +177,11 @@ the .ABOUT files in the directory /tmp/thirdparty_about/.
                             0 - Do not print any warning or error messages, just a total count (default)
                             1 - Print error messages
                             2 - Print error and warning messages
+    
+      --template_location=TEMPLATE_LOCATION
+                            Use the custom template for the Attribution Generation
+    
+      --mapping             Configure the mapping key from the MAPPING.CONFIG
 
 Example::
 

USAGE

@@ -192,12 +192,12 @@ Syntax
 ------
 
     Usage: genattrib.py [options] input_path output_path component_list
-
+    
         Input can be a file or directory.
         Output of rendered template must be a file (e.g. .html).
         Component List must be a .csv file which has at least an "about_resource" column.
-
-
+    
+    
     Options:
       -h, --help            Display help
       -v, --version         Display current version, license notice, and copyright notice
@@ -208,6 +208,11 @@ Syntax
                             1 - Print error messages
                             2 - Print error and warning messages
 
+      --template_location=TEMPLATE_LOCATION
+                            Use the custom template for the Attribution Generation
+
+      --mapping             Configure the mapping key from the MAPPING.CONFIG
+
 -------
 Purpose
 -------
@@ -225,4 +230,18 @@ Purpose
     '/home/attribution/attribution.html' is the user's output path
     '/home/project/component_list.csv' is the component list that user want to be generated
 
-        $ python genattrib.py /home/about_files/ /home/attribution/attribution.html /home/project/component_list.csv
+        $ python genattrib.py /home/about_files/ /home/attribution/attribution.html /home/project/component_list.csv
+
+-------
+Options
+-------
+
+    --template_loaction
+
+        This option allows user to use its own template for Attribution Generation.
+        For instance, if the custom template the user want to use is located at:
+        /home/custom_template/template.html
+
+            $ python genattrib.py --template_location=/home/custom_template/template.html input_path output_path component_list
+
+

about_code_tool/about.py

@@ -1086,7 +1086,7 @@ def write_to_csv(self, output_path):
                 row_data = about_object.get_row_data(relative_path)
                 csv_writer.writerow(row_data)
 
-    def generate_attribution(self, template_path='templates/default.html',
+    def generate_attribution(self, template_path=None,
                              limit_to=None):
         """
         Generates an attribution file from the current list of ABOUT objects.
@@ -1096,6 +1096,9 @@ def generate_attribution(self, template_path='templates/default.html',
         if not limit_to:
             limit_to = []
 
+        if not template_path:
+            template_path = 'templates/default.html'
+
         try:
             from jinja2 import Environment, FileSystemLoader, TemplateNotFound
         except ImportError:

about_code_tool/genattrib.py

@@ -89,6 +89,10 @@ def update_path_to_about(input_list):
 2 - Print error and warning messages
 """
 
+TEMPLATE_LOCATION_HELP = """\
+Use the custom template for the Attribution Generation
+"""
+
 MAPPING_HELP = """\
 Configure the mapping key from the MAPPING.CONFIG
 """
@@ -97,7 +101,8 @@ def main(parser, options, args):
     overwrite = options.overwrite
     verbosity = options.verbosity
     mapping_config = options.mapping
-    
+    template_location = options.template_location
+
     if options.version:
         print('ABOUT tool {0}\n{1}'.format(__version__, __copyright__))
         sys.exit(0)
@@ -169,7 +174,7 @@ def main(parser, options, args):
             sublist = component_subset_to_sublist(input_list)
             outlist = update_path_to_about(sublist)
 
-        attrib_str = collector.generate_attribution( limit_to = outlist )
+        attrib_str = collector.generate_attribution(template_path=template_location, limit_to=outlist)
         with open(output_path, "w") as f:
             f.write(attrib_str)
         errors = collector.get_genattrib_errors()
@@ -233,6 +238,7 @@ def format_option(self, option):
     parser.add_option('--overwrite', action='store_true',
                       help='Overwrites the output file if it exists')
     parser.add_option('--verbosity', type=int, help=VERBOSITY_HELP)
+    parser.add_option('--template_location', type='string', help=TEMPLATE_LOCATION_HELP)
     parser.add_option('--mapping', action='store_true', help=MAPPING_HELP)
     return parser