|
|
|
@ -1,6 +1,6 @@
|
|
|
|
import logging
|
|
|
|
import logging
|
|
|
|
import json
|
|
|
|
import json
|
|
|
|
from typing import Dict, Any, Optional
|
|
|
|
from typing import Dict, Any
|
|
|
|
from pathlib import Path
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
|
|
|
|
from lxml import etree
|
|
|
|
from lxml import etree
|
|
|
|
@ -121,12 +121,9 @@ class XMLDataSetCollection(AbstractDataSet):
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
def __init__(self,
|
|
|
|
def __init__(self,
|
|
|
|
housename: str,
|
|
|
|
housename: str,
|
|
|
|
folderpath: str,
|
|
|
|
folderpath: str) -> None:
|
|
|
|
outputfolderpath: Optional[str]=None) -> None:
|
|
|
|
|
|
|
|
self._housename = housename
|
|
|
|
self._housename = housename
|
|
|
|
self._folderpath = Path(folderpath)
|
|
|
|
self._folderpath = Path(folderpath)
|
|
|
|
if outputfolderpath is not None:
|
|
|
|
|
|
|
|
self.outputfolderpath = Path(outputfolderpath)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _load(self) -> dict[str, EtreeXMLDataSet]:
|
|
|
|
def _load(self) -> dict[str, EtreeXMLDataSet]:
|
|
|
|
"kedro's API loader method"
|
|
|
|
"kedro's API loader method"
|
|
|
|
@ -134,9 +131,7 @@ class XMLDataSetCollection(AbstractDataSet):
|
|
|
|
for filepath in sorted(self._folderpath.glob("*.xml")):
|
|
|
|
for filepath in sorted(self._folderpath.glob("*.xml")):
|
|
|
|
self.datasets[filepath.stem] = EtreeXMLDataSet(
|
|
|
|
self.datasets[filepath.stem] = EtreeXMLDataSet(
|
|
|
|
filepath=str(filepath))
|
|
|
|
filepath=str(filepath))
|
|
|
|
# return self.datasets
|
|
|
|
return self.datasets
|
|
|
|
# we need the object itself during transformation
|
|
|
|
|
|
|
|
return self
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _save(self, data) -> None:
|
|
|
|
def _save(self, data) -> None:
|
|
|
|
"""kedro's API saver method
|
|
|
|
"""kedro's API saver method
|
|
|
|
|