From eab369b5721b933c080516fd1dcdfa67e6e2dfcd Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Thu, 12 Dec 2013 22:19:46 +0000 Subject: Read classes on HTML path elements --- mammoth/html_paths.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'mammoth/html_paths.py') diff --git a/mammoth/html_paths.py b/mammoth/html_paths.py index cb53968..0f2184d 100644 --- a/mammoth/html_paths.py +++ b/mammoth/html_paths.py @@ -5,9 +5,11 @@ def path(elements): return HtmlPath(elements) -def element(names): - return HtmlPathElement(names) +def element(names, class_names=None): + if class_names is None: + class_names = [] + return HtmlPathElement(names, class_names) HtmlPath = collections.namedtuple("HtmlPath", ["elements"]) -HtmlPathElement = collections.namedtuple("HtmlPathElement", ["names"]) +HtmlPathElement = collections.namedtuple("HtmlPathElement", ["names", "class_names"]) -- cgit v1.2.3