From 332b5d7c80f7af68c2f581dae5d70b8c78ec3ddd Mon Sep 17 00:00:00 2001
From: volpino <fox91@anche.no>
Date: Sun, 2 Sep 2012 18:39:04 +0200
Subject: [PATCH] euscan: Added XML output

Signed-off-by: volpino <fox91@anche.no>
---
 TODO                  |  2 +-
 pym/euscan/helpers.py | 31 +++++++++++++++++++++++++++++++
 pym/euscan/out.py     |  4 ++++
 3 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/TODO b/TODO
index eaa0d72..19126b4 100644
--- a/TODO
+++ b/TODO
@@ -13,7 +13,7 @@ euscan
 
 ### Command line interface
 
-- html and xml output
+- html output
 
 ### Misc
 
diff --git a/pym/euscan/helpers.py b/pym/euscan/helpers.py
index f748c2e..5e3e6ea 100644
--- a/pym/euscan/helpers.py
+++ b/pym/euscan/helpers.py
@@ -2,6 +2,7 @@ import os
 import re
 import errno
 import urllib2
+from xml.dom.minidom import Document
 
 import portage
 from portage import dep
@@ -451,3 +452,33 @@ def parse_mirror(uri):
         return None
 
     return uri
+
+
+def dict_to_xml(data, indent):
+    doc = Document()
+    root = doc.createElement("euscan")
+    doc.appendChild(root)
+
+    def _set_value(parent, value):
+        if isinstance(value, dict):
+            for k, v in value.iteritems():
+                node = doc.createElement(k)
+                _set_value(node, v)
+                parent.appendChild(node)
+        elif isinstance(value, list):
+            for item in value:
+                node = doc.createElement("value")
+                text = doc.createTextNode(item)
+                node.appendChild(text)
+                parent.appendChild(node)
+        else:
+            text = doc.createTextNode(unicode(value))
+            parent.appendChild(text)
+
+    for key, value in data.iteritems():
+        node = doc.createElement("package")
+        node.setAttribute("name", key)
+        _set_value(node, value)
+        root.appendChild(node)
+
+    return doc.toprettyxml(indent=" " * indent)
diff --git a/pym/euscan/out.py b/pym/euscan/out.py
index dcdd5f0..e6e6ebf 100644
--- a/pym/euscan/out.py
+++ b/pym/euscan/out.py
@@ -10,6 +10,8 @@ import portage
 from portage.output import EOutput, TermProgressBar
 from gentoolkit import pprinter as pp
 
+from euscan.helpers import dict_to_xml
+
 mirrors_ = None
 
 
@@ -200,6 +202,8 @@ class EuscanOutput(object):
         format_ = format_ or self.config["format"]
         if format_.lower() == "json":
             return json.dumps(data, indent=self.config["indent"])
+        elif format_.lower() == "xml":
+            return dict_to_xml(data, indent=self.config["indent"])
         elif format_.lower() == "dict":
             return data
         else: