Python: создать CSV из XML с различными вложенными элементами - PullRequest
Новая
       58

Python: создать CSV из XML с различными вложенными элементами

0 голосов
/ 26 июня 2018

Это мой xml файл: 

    <?xml version="1.0" encoding="UTF-8"?>
    <Invoice xmlns="urn:oasis:names:specification:ubl:schema:xsd:Invoice-2" xmlns:cac="urn:oasis:names:specification:ubl:schema:xsd:CommonAggregateComponents-2" xmlns:cbc="urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2" xmlns:ccts="urn:oasis:names:specification:ubl:schema:xsd:CoreComponentParameters-2" xmlns:sdt="urn:oasis:names:specification:ubl:schema:xsd:SpecializedDatatypes-2" xmlns:udt="urn:un:unece:uncefact:data:specification:UnqualifiedDataTypesSchemaModule:2" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="urn:oasis:names:specification:ubl:schema:xsd:Invoice-2 UBL-Invoice-2.0.xsd">
       <cbc:ID>102165444</cbc:ID>
       <cac:InvoiceLine>
          <cbc:ID>1.0000</cbc:ID>
          <cbc:Note />
          <cbc:InvoicedQuantity unitCode="CT">1.0000</cbc:InvoicedQuantity>
          <cbc:LineExtensionAmount currencyID="DKK">142.3900</cbc:LineExtensionAmount>
          <cac:TaxTotal>
             <cbc:TaxAmount currencyID="DKK">138.24</cbc:TaxAmount>
             <cac:TaxSubtotal>
                <cbc:TaxableAmount currencyID="DKK">142.39</cbc:TaxableAmount>
                <cbc:TaxAmount currencyID="DKK">7.20</cbc:TaxAmount>
                <cac:TaxCategory>
                   <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxcategoryid-1.1">3645</cbc:ID>
                   <cac:TaxScheme>
                      <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxschemeid-1.1">140</cbc:ID>
                      <cbc:Name>Afgift</cbc:Name>
                      <cbc:TaxTypeCode listAgencyID="320" listID="urn:oioubl:codelist:taxtypecode-1.1">StandardRated</cbc:TaxTypeCode>
                   </cac:TaxScheme>
                </cac:TaxCategory>
             </cac:TaxSubtotal>
          </cac:TaxTotal>
       </cac:InvoiceLine>
          <cbc:ID>2.0000</cbc:ID>
          <cbc:Note />
          <cbc:InvoicedQuantity unitCode="CT">1.0000</cbc:InvoicedQuantity>
          <cbc:LineExtensionAmount currencyID="DKK">142.3900</cbc:LineExtensionAmount>
          <cac:TaxTotal>
             <cbc:TaxAmount currencyID="DKK">138.24</cbc:TaxAmount>
             <cac:TaxSubtotal>
                <cbc:TaxableAmount currencyID="DKK">142.39</cbc:TaxableAmount>
                <cbc:TaxAmount currencyID="DKK">7.20</cbc:TaxAmount>
                <cac:TaxCategory>
                   <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxcategoryid-1.1">3645</cbc:ID>
                   <cac:TaxScheme>
                      <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxschemeid-1.1">140</cbc:ID>
                      <cbc:Name>Afgift</cbc:Name>
                      <cbc:TaxTypeCode listAgencyID="320" listID="urn:oioubl:codelist:taxtypecode-1.1">StandardRated</cbc:TaxTypeCode>
                   </cac:TaxScheme>
                </cac:TaxCategory>
             </cac:TaxSubtotal>
          </cac:TaxTotal>
          <cac:TaxTotal>
             <cbc:TaxAmount currencyID="DKK">35.60</cbc:TaxAmount>
             <cac:TaxSubtotal>
                <cbc:TaxableAmount currencyID="DKK">142.39</cbc:TaxableAmount>
                <cbc:TaxAmount currencyID="DKK">35.60</cbc:TaxAmount>
                <cac:TaxCategory>
                   <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxcategoryid-1.1">StandardRated</cbc:ID>
                   <cbc:Percent>25</cbc:Percent>
                   <cac:TaxScheme>
                      <cbc:ID schemeAgencyID="320" schemeID="urn:oioubl:id:taxschemeid-1.1">63</cbc:ID>
                      <cbc:Name>Moms</cbc:Name>
                   </cac:TaxScheme>
                </cac:TaxCategory>
             </cac:TaxSubtotal>
          </cac:TaxTotal>
       </cac:InvoiceLine>
    </Invoice>

Как видите, файл имеет идентификатор, несколько «строк счета», каждый из которых имеет свой идентификатор вместе с другими дочерними элементами.

Я хочу создать CSV-файл со строкой для каждой строки счета-фактуры с информацией о конкретных вложенных элементах. Проблема в том, что для каждой строки может быть несколько дочерних элементов TaxTotal. В этом случае я бы хотел другую строку с такой информацией, как эта: 

ID;/InvoiceLine/ID;InvoiceLine/InvoicedQuantity;/InvoiceLine/LineExtensionAmount;/InvoiceLine/TaxTotal/TaxAmount;/InvoiceLine/TaxTotal/TaxSubtotal/TaxableAmount    /InvoiceLine/TaxTotal/TaxSubtotal/TaxAmount ;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/ID;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/Percent;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/ID;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/Name
102165444;1;1;142,39;138,24;142,39;7,20;3645,00;;140;Afgift
102165444;2;1;142,39;138,24;142,39;7,20;3646,00;;140;Afgift
102165444;2;1;142,39;35,60;142,39;35,60;StandardRated;25,00;63;Moms

Как мне это сделать?

1 Ответ

0 голосов
/ 30 июня 2018

Поскольку всегда будет хотя бы один элемент TaxTotal, я бы создал новую строку csv для каждого и вернулся бы к дереву для предыдущих значений.

Вот пример использования lxml. Я добавил функцию, чтобы упростить обработку пустых значений, но любое дополнительное форматирование значений я оставлю на ваше усмотрение.

Python 3,6 

from lxml import etree
import csv


def get_value(target_tree, xpath, namespaces):
    try:
        return target_tree.xpath(xpath, namespaces=namespaces)[0].text
    except IndexError:
        return ""


tree = etree.parse("input.xml")

ns = {"cac": "urn:oasis:names:specification:ubl:schema:xsd:CommonAggregateComponents-2",
      "cbc": "urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2",
      "i2": "urn:oasis:names:specification:ubl:schema:xsd:Invoice-2"}

with open("output.csv", "w") as csvfile:
    csvwriter = csv.writer(csvfile, delimiter=";", lineterminator="\n", quoting=csv.QUOTE_MINIMAL)
    # Header
    csvwriter.writerow(["ID", "/InvoiceLine/ID", "/InvoiceLine/InvoicedQuantity", "/InvoiceLine/LineExtensionAmount",
                        "/InvoiceLine/TaxTotal/TaxAmount", "/InvoiceLine/TaxTotal/TaxSubtotal/TaxableAmount",
                        "/InvoiceLine/TaxTotal/TaxSubtotal/TaxAmount",
                        "/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/ID",
                        "/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/Percent",
                        "/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/ID",
                        "/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/Name"])
    for tax_total in tree.xpath("//cac:TaxTotal", namespaces=ns):
        csvwriter.writerow([get_value(tax_total, "/i2:Invoice/cbc:ID", ns),
                            get_value(tax_total, "../cbc:ID", ns),
                            get_value(tax_total, "../cbc:InvoicedQuantity", ns),
                            get_value(tax_total, "../cbc:LineExtensionAmount", ns),
                            get_value(tax_total, "cbc:TaxAmount", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cbc:TaxableAmount", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cbc:TaxAmount", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cac:TaxCategory/cbc:ID", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cac:TaxCategory/cbc:Percent", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cac:TaxCategory/cac:TaxScheme/cbc:ID", ns),
                            get_value(tax_total, "cac:TaxSubtotal/cac:TaxCategory/cac:TaxScheme/cbc:Name", ns)])

Выход (output.csv) 

ID;/InvoiceLine/ID;/InvoiceLine/InvoicedQuantity;/InvoiceLine/LineExtensionAmount;/InvoiceLine/TaxTotal/TaxAmount;/InvoiceLine/TaxTotal/TaxSubtotal/TaxableAmount;/InvoiceLine/TaxTotal/TaxSubtotal/TaxAmount;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/ID;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/Percent;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/ID;/InvoiceLine/TaxTotal/TaxSubtotal/TaxCategory/TaxScheme/Name
102165444;1.0000;1.0000;142.3900;138.24;142.39;7.20;3645;;140;Afgift
102165444;2.0000;1.0000;142.3900;138.24;142.39;7.20;3645;;140;Afgift
102165444;2.0000;1.0000;142.3900;35.60;142.39;35.60;StandardRated;25;63;Moms
...