From b54962900da5eac5bbce398e97d1abe997042fed Mon Sep 17 00:00:00 2001 From: "Martin A. Brown" Date: Tue, 1 Mar 2016 11:01:32 -0800 Subject: [PATCH] allow case-insensitive matching for document types The Python names for the document types are upper-cased and probably not predictably named for the end user; allow a case-insensitive match when trying to turn off building or listing for a particular document type. --- tldp/driver.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tldp/driver.py b/tldp/driver.py index d584166..8028bb7 100644 --- a/tldp/driver.py +++ b/tldp/driver.py @@ -115,11 +115,15 @@ def getStatusNames(args): def getDocumentClasses(args): - sought = set() + largs = [x.lower() for x in args] + sought = list() for cls in tldp.typeguesser.knowndoctypes: - if cls.__name__.lower() in args: - sought.add(cls) - remainder = set(args).difference(sought) + if cls.__name__.lower() in largs: + sought.append(cls) + else: + sought.append(None) + remainder = set([y for x, y in zip(sought, args) if x]) + sought = set(sought) return sought, remainder