Added exception for "Documentation" string.

dietmarw · dietmarw · commit 1eb4db8ef430 · 2020-02-21T09:34:49.000+01:00
diff --git a/setup.py b/setup.py
@@ -24,7 +24,7 @@
 META = {
     'name': 'ttws',
     'url': 'https://github.com/dietmarw/trimtrailingwhitespaces',
-    'version': '0.8.1',
+    'version': '0.8.2',
     'description': 'Script to remove trailing whitespaces from textfiles.',
     'classifiers': CLASSIFIERS,
     'license': 'UNLICENSE',
diff --git a/test/tests.mo b/test/tests.mo
@@ -44,6 +44,9 @@ annotation (Diagram(coordinateSystem(extent={{-100,-100},{100,100}},
 
 // Empty Documentation should be removed:
 annotation (Documentation);
+annotation( Documentation );
+annotation( Documentation, );
+annotation( Documentation , );
 
 // Empty Documentation should be removed:
 annotation (Documentation(info="<html>
@@ -53,4 +56,11 @@ with model LossyGear and a proposed solution</a>&quot;
 </p>
 </html>"));
 
+// Simple Documentation string should not be removed:
+annotation (Documentation(info="<html>
+<p>
+Some Documentation can be found here.
+Documentation at the start.</p>
+</html>"));
+
 end LargeModel;
diff --git a/ttws/__init__.py b/ttws/__init__.py
@@ -177,7 +177,7 @@ def cleanAnnotation(filepath, eol):
         # special care of the last one again
         lastDocRef = (Optional(',') + ZeroOrMore(White(' \t'))
                          + (Keyword('Documentation')) + ~FollowedBy('/')
-                         + ~nestedExpr() + ZeroOrMore(White(' \t')))
+                         + ~nestedExpr() + FollowedBy(')'))
         out = Suppress(lastDocRef).transformString(out)
 
 
