Fix formatting

BudaPortaXmlConvert.py

@@ -11,6 +11,7 @@ import xml.etree.ElementTree as ET
 import csv
 import datetime
 import os
+import xmlformatter
 
 
 # ---------------------------------------------------------------------------- #
@@ -70,11 +71,11 @@ def getXmlData(data):
 
 def checkData(xmlData):
     print()
-    missingData = False
-    for mérő in xmlData:
+    missingData = []
+    for i, mérő in enumerate(xmlData):
         for adat in mérő:
-            if len(mérő[adat]) == 0:
-                missingData = True
+            if len(mérő[adat]) == 0 or mérő[adat] == "diktalos":
+                missingData.append(i)
                 print(f'Adat hiányzik: "{adat}" a következő sorból:')
                 print(mérő)
     return missingData
@@ -147,16 +148,22 @@ for i, table in enumerate(csvArrs):
 xmlReqData = []
 for i, table in enumerate(reqData):
     print(f'Fájl ellenőrzése: {csvFileNames[i]}')
+
+    # unwrap data
     currXmlData = getXmlData(table)
-    if not checkData(currXmlData):
+
+    # check for missing data
+    wrongLines = checkData(currXmlData)
+    if len(wrongLines) == 0:
         print('Fájl hibátlan')
+    else:
+        # remove bad lines
+        print(wrongLines)
+        for ele in sorted(wrongLines, reverse=True):
+            del currXmlData[ele]
     print()
     xmlReqData.append(currXmlData)
 
-# print(xmlReqData)
-# for i in xmlReqData:
-#     print(i)
-
 
 # ------------------------------- Merge arrays ------------------------------- #
 
@@ -182,24 +189,36 @@ while fileNameNotFound:
     if not os.path.exists(fileName):
         fileNameNotFound = False
 
-
 print(f'Fájl mentése ide: {fileName}')
 print()
 
 # --------------------------------- Write xml -------------------------------- #
 
 data = ET.Element('Leolvasasok')
-data.tail = '\r\n'
 for mérő in allXmlData:
     currSub = ET.SubElement(data, 'Leolvasas')
-    currSub.tail = '\r\n'
     for adat in mérő:
         a = ET.SubElement(currSub, adat)
         a.text = mérő[adat]
-        a.tail = '\r\n'
 
-print(ET.dump(data))
-print()
+xmlString = ET.tostring(data, encoding="utf-8", xml_declaration=True)
 
-tree = ET.ElementTree(data)
-tree.write(fileName, encoding="utf-8", xml_declaration=True)
+# -------------------------------- Format xml -------------------------------- #
+
+formatter = xmlformatter.Formatter(indent="2", indent_char=" ")
+formattedXml = formatter.format_string(xmlString)
+
+# print(formattedXml)
+
+# ----------------------------- Fix line endings ----------------------------- #
+
+WINDOWS_LINE_ENDING = b'\r\n'
+UNIX_LINE_ENDING = b'\n'
+
+formattedXml = formattedXml.replace(UNIX_LINE_ENDING, WINDOWS_LINE_ENDING)
+
+# --------------------------------- Save file -------------------------------- #
+
+f = open(fileName, "xb")
+f.write(formattedXml)
+f.close

README.md

@@ -27,3 +27,8 @@ A file nevében ne legyen ékezetes karakter.
 Egy sor sem lehet 50 karakternél hosszabb!
 
 
+## Telepítés
+
+```shell
+pip install -r requirements.txt
+```

requirements.txt

@@ -0,0 +1 @@
+xmlformatter