From 6025db45a779ddc1dd2ac9559fa50f28cadbb193 Mon Sep 17 00:00:00 2001 From: Eric Holscher Date: Thu, 25 Aug 2016 10:58:06 -0700 Subject: [PATCH] Clean up how we handle file suffix. --- autoapi/extension.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/autoapi/extension.py b/autoapi/extension.py index 4eb4999..011b35d 100644 --- a/autoapi/extension.py +++ b/autoapi/extension.py @@ -66,6 +66,16 @@ def run_autoapi(app): else: ignore_patterns = default_ignore_patterns.get(app.config.autoapi_type, []) + if '.rst' in app.config.source_suffix: + out_suffix = '.rst' + elif '.txt' in app.config.source_suffix: + out_suffix = '.txt' + else: + # Fallback to first suffix listed + out_suffix = app.config.source_suffix[0] + + # Actual meat of the run. + app.info(bold('[AutoAPI] ') + darkgreen('Loading Data')) domain_obj.load( patterns=file_patterns, @@ -79,8 +89,7 @@ def run_autoapi(app): app.info(bold('[AutoAPI] ') + darkgreen('Rendering Data')) domain_obj.output_rst( root=normalized_root, - # TODO: Better way to determine suffix? - source_suffix=app.config.source_suffix[0], + source_suffix=out_suffix, )