CursosPythonMexico/Introducción-a-Python-creando-scripts-con-Click/script_05_desmembrando_xml/catxml-0.2.0.py at master · PythonMexico/CursosPythonMexico · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import click
from xml.dom import minidom

def filtra_attr(nom, val, attr, l):
    """ Imprime sólo los atributos que son igual a attr """
    if attr:
        if nom == attr:
            print("   "*l + "|-", nom, "=", val)
        else:
            pass
    else:
        print("   "*l + "|-", nom, "=", val)

def imprime_atributos(nodo, attr, l):
    """
    Imprime los atributos de un nodo
    """
    for nom, val in nodo.attributes.items():
        filtra_attr(nom, val, attr, l)

def filtra_nodo_raiz(actual, nodo, attr):
    """ Imprime sólo los nodos que son igual a node """
    if nodo:
        if actual.nodeName == nodo:
            print(actual.nodeName)
        else:
            pass
    else:
        print(actual.nodeName)

def filtra_hijos(hijo, nodo, l):
    """ Imprime sólo los hijos que son igual a nodo """
    if nodo:
        if hijo.nodeName == nodo:
            print("   "*l + "|-", hijo.nodeName)
        else:
            pass
    else:
        print("   "*l + "|-", hijo.nodeName)

def filtra_hijos_data(raiz, data, nodo, l):
    """ Imprime sólo las data de los hijos que son igual a nodo """
    if nodo:
        if raiz.nodeName == nodo:
            print("   "*l + "|-", data)
        else:
            pass
    else:
        print("   "*l + "|-", data)

def imprime_hijos(nodo, node, attr, l=0):
    """
    Imprime el contenido de un nodo en la salida estándar
    """
    if nodo.attributes:
        if node:
            if nodo.nodeName == node:
                imprime_atributos(nodo, attr, l)
            else:
                pass
        else:
            imprime_atributos(nodo, attr, l)

    for hijo in nodo.childNodes:
        if hijo.nodeName == "#text":
            if hijo.data.strip() != "":
                filtra_hijos_data(nodo, hijo.data, node, l)
                # print("   "*l, "|-", hijo.data)
        else:
            filtra_hijos(hijo, node, l)
            imprime_hijos(hijo, node, attr, l+1)

def imprime_atributos_html(nodo, l):
    """
    Imprime los atributos de un nodo en formato html
    """
    for nom, val in nodo.attributes.items():
        print("<li class='attr'>", nom, "=", val, "</li>")

def imprime_hijos_html(nodo, l=0):
    """
    Imprime el contenido de un nodo en la salida estándar en formato html
    """
    print("<ul class='nodo'>")
    if nodo.attributes:
        imprime_atributos_html(nodo, l)

    for hijo in nodo.childNodes:
        if hijo.nodeName == "#text":
            if hijo.data.strip() != "":
                print("<li class='hijo'>", hijo.data, "</li>")
        else:
            print("<li class='hijo'>", hijo.nodeName, "</li>")
            imprime_hijos_html(hijo, l+1)
    print("</ul>")


@click.command()
@click.option("--raw", is_flag=True,
    help="Imprime el contenido del archivo sin modificación alguna")
@click.option("--html", is_flag=True,
    help="Imprime el contenido del archivo en formato html")
@click.option("--node", type=str,
    help="Muestra solamente los nodos cuyo nombre esta dado por TEXT")
@click.option("--attr", type=str,
    help="Muestra solamente los atributos cuyo nombre esta dado por TEXT")
@click.argument("pathfile", type=click.Path(exists=True), required=True)
def catxml(raw, html, node, attr, pathfile):
    """
    Imprime en la salida estándar el contenido del archivo xml
    indicado por PATHFILE
    """
    if raw:
        with open(pathfile) as f:
            for linea in f:
                print(linea, end="")
    elif html:
        dom = minidom.parse(pathfile)
        raiz = dom.childNodes[0]
        #filtra_nodo_raiz(raiz, node, attr)
        print("<div class='raiz'>", raiz.nodeName, "</div>")
        # imprime_hijos_html(raiz, node, attr)
        imprime_hijos_html(raiz)
    else:
        dom = minidom.parse(pathfile)
        raiz = dom.childNodes[0]
        filtra_nodo_raiz(raiz, node, attr)
        imprime_hijos(raiz, node, attr)


if __name__ == "__main__":
    catxml()