From 743d2f3d81695d82a2bc1ff13ed2d3008a2ccb3f Mon Sep 17 00:00:00 2001
From: thislooksfun <thislooksfun@users.noreply.github.com>
Date: Tue, 20 Aug 2019 13:36:13 -0700
Subject: [PATCH] Add ability to switch output format

---
 src/demo.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/demo.py b/src/demo.py
index da68f7e1..cab7faf6 100755
--- a/src/demo.py
+++ b/src/demo.py
@@ -41,10 +41,22 @@
   filterFunc = getattr(compatibility, filter)
   events = filterFunc(events)
 
-  from feedvalidator.formatter.text_plain import Formatter
+  formatArg = sys.argv[3:] and sys.argv[3] or "--format=xml"
+  format = formatArg[9:]
+
+  if format == 'html':
+    from feedvalidator.formatter.text_html import Formatter
+  elif format == 'xml':
+    from feedvalidator.formatter.text_xml import Formatter
+  else:
+    from feedvalidator.formatter.text_plain import Formatter
+
   output = Formatter(events)
   if output:
-      print "\n".join(output)
+      s = "\n".join(output)
+      if format == 'xml':
+        s = '<?xml version="1.0"?>\n<validationErrors>\n' + s + "</validationErrors>"
+      print s
       sys.exit(1)
   else:
       print "No errors or warnings"