os.walk() python: xml-Darstellung einer Verzeichnisstruktur, Rekursion

So, ich bin versucht, os.walk() zum generieren einer XML-Darstellung einer Verzeichnisstruktur. Ich scheine immer eine Tonne von Duplikaten. Es richtig stellen Verzeichnisse innerhalb jedes anderen und Dateien am richtigen Platz für den ersten Teil der xml-Datei, aber nachdem Sie es tut richtig geht es dann weiter durchqueren falsch. Ich bin nicht ganz sicher, warum....

Hier ist mein code:

def dirToXML(self,directory):
        curdir = os.getcwd()
        os.chdir(directory)
        xmlOutput=""

        tree = os.walk(directory)
        for root, dirs, files in tree:
            pathName = string.split(directory, os.sep)
            xmlOutput+="<dir><name><![CDATA["+pathName.pop()+"]]></name>"
            if len(files)>0:
                xmlOutput+=self.fileToXML(files)
            for subdir in dirs:
                xmlOutput+=self.dirToXML(os.path.join(root,subdir))
            xmlOutput+="</dir>"

        os.chdir(curdir)
        return xmlOutput

Den fileToXML, einfach analysiert aus der Liste, so dass keine Notwendigkeit zu befürchten, dass.

Die Directory-Struktur ist einfach:

images/
images/testing.xml
images/structure.xml
images/Hellos
images/Goodbyes
images/Goodbyes/foo
images/Goodbyes/bar
images/Goodbyes/square

und die resultierende xml-Datei wurde:

<structure>
<dir>
<name>images</name>
  <files>
    <file>
      <name>structure.xml</name>
    </file>
    <file>
      <name>testing.xml</name>
    </file>
  </files>
  <dir>
    <name>Hellos</name>
  </dir>
  <dir>
    <name>Goodbyes</name>
    <dir>
      <name>foo</name>
    </dir>
    <dir>
      <name>bar</name>
    </dir>
    <dir>
      <name>square</name>
    </dir>
  </dir>
  <dir>
    <name>foo</name>
  </dir>
  <dir>
    <name>bar</name>
  </dir>
  <dir>
      <name>square</name>
    </dir>
  </dir>
  <dir>
    <name>Hellos</name>
  </dir>
  <dir>
    <name>Goodbyes</name>
    <dir>
      <name>foo</name>
    </dir>
    <dir>
      <name>bar</name>
    </dir>
    <dir>
      <name>square</name>
    </dir>
  </dir>
  <dir>
    <name>foo</name>
  </dir>
  <dir>
    <name>bar</name>
  </dir>
  <dir>
    <name>square</name>
  </dir>
</structure>

Jede Hilfe wäre sehr geschätzt werden!

InformationsquelleAutor Parris | 2010-01-20
  1. 8

    Ich würde empfehlen, gegen die Verwendung von os.walk(), da haben Sie so viel zu tun zu massage seinen Ausgang. Stattdessen nutzen Sie einfach eine rekursive Funktion, die verwendet os.listdir(), os.path.join(), os.path.isdir() usw.

    import os
from xml.sax.saxutils import escape as xml_escape

def DirAsXML(path):
    result = '<dir>\n<name>%s</name>\n' % xml_escape(os.path.basename(path))
    dirs = []
    files = []
    for item in os.listdir(path):
        itempath = os.path.join(path, item)
        if os.path.isdir(itempath):
            dirs.append(item)
        elif os.path.isfile(itempath):
            files.append(item)
    if files:
        result += '  <files>\n' \
            + '\n'.join('    <file>\n      <name>%s</name>\n    </file>'
            % xml_escape(f) for f in files) + '\n  </files>\n'
    if dirs:
        for d in dirs:
            x = DirAsXML(os.path.join(path, d))
            result += '\n'.join('  ' + line for line in x.split('\n'))
    result += '</dir>'
    return result

if __name__ == '__main__':
    print '<structure>\n' + DirAsXML(os.getcwd()) + '\n</structure>'

    Persönlich, ich würde empfehlen, eine sehr viel weniger ausführlich, XML-schema, putting Namen, Attribute und loszuwerden, die <files> Gruppe:

    import os
from xml.sax.saxutils import quoteattr as xml_quoteattr

def DirAsLessXML(path):
    result = '<dir name=%s>\n' % xml_quoteattr(os.path.basename(path))
    for item in os.listdir(path):
        itempath = os.path.join(path, item)
        if os.path.isdir(itempath):
            result += '\n'.join('  ' + line for line in 
                DirAsLessXML(os.path.join(path, item)).split('\n'))
        elif os.path.isfile(itempath):
            result += '  <file name=%s />\n' % xml_quoteattr(item)
    result += '</dir>'
    return result

if __name__ == '__main__':
    print '<structure>\n' + DirAsLessXML(os.getcwd()) + '\n</structure>'

    Dieser liefert eine Ausgabe wie:

    <structure>
<dir name="local">
  <dir name=".hg">
    <file name="00changelog.i" />
    <file name="branch" />
    <file name="branch.cache" />
    <file name="dirstate" />
    <file name="hgrc" />
    <file name="requires" />
    <dir name="store">
      <file name="00changelog.i" />

    etc.

    Wenn os.walk() mehr gearbeitet wie expat's Rückrufe, Sie würden eine einfachere Zeit davon.

    • ... und Sie erreicht die gleichen Schluss schon. Nicht sicher, warum ich habe nicht die "1 neue Antwort" - Warnung. -_-;
    • Haha ok... ich werde es dir geben! Danke für die Hilfe :)!
    InformationsquelleAutor Mike DeSimone
  2. 6

    Entfernen Sie die beiden Zeilen:

            for subdir in dirs:
            xmlOutput+=self.dirToXML(os.path.join(root,subdir))

    sind Sie recursing in die Unterverzeichnisse; aber das ist redundant, weil os.walk bezieht sich.

    • Das funktioniert nicht wirklich für mich... gibt es eine Möglichkeit, nur eine Liste der Dateien und nur die Verzeichnisse in python... ich muss die xml-Datei, um mehr Baum.
    InformationsquelleAutor oefe
  3. 0

    War ich versucht, os.entfernt, aber ich sah, dass es nicht funktioniert mit der rekursiven Struktur, die ich wollte zu schaffen, in xml. Ich veränderte meinen code wie folgt, und Sie produzieren das Ergebnis brauche ich: 

    def dirToXML(self,directory):
        curdir = os.getcwd()
        os.chdir(directory)
        xmlOutput=""

        pathName = string.split(directory, os.sep)
        xmlOutput+="<dir><name><![CDATA["+pathName.pop()+"]]></name>"
        for item in os.listdir(directory):
            if os.path.isfile(os.path.join(directory, item)):
                xmlOutput+="<file><name><![CDATA["+item+"]]></name></file>"
            else :
                xmlOutput+=self.dirToXML(os.path.join(directory,item))
        xmlOutput+="</dir>"

        os.chdir(curdir)
        return xmlOutput
    InformationsquelleAutor Parris
Schreibe einen Kommentar