tester and person entity

1 files changed, 21 insertions, 0 deletions

diff --git a/testers/factorytester.py b/testers/factorytester.py
new file mode 100644
index 0000000..cc95bab
--- /dev/null
+++ b/testers/factorytester.py
@@ -0,0 +1,21 @@
+from __future__ import absolute_import
+import parsers
+import factories
+import xml.etree.cElementTree as ET
+
+xmlpath = 'C:/Users/Ilsar/Documents/datahack/xml_example.xml'
+whitelist = ['100', '374', '400', '151', '451', '550', '551', '678']
+
+
+xmltree = ET.parse(xmlpath)
+entities = list()
+inl_factory = factories.INLFactory()
+
+for record in xmltree.getroot():
+    inl_parser = parsers.INLXmlParser(record, whitelist)
+    clean_record = inl_parser.clearxml()
+    entities.append(inl_factory.get_entity(clean_record))
+
+for entity in entities:
+    entity.print_entity()
+