rewrite parsing code in file.py to be more class-based
This commit is contained in:
parent
d408a990f5
commit
b5aec5940e
|
@ -0,0 +1,26 @@
|
||||||
|
cat test:
|
||||||
|
foo: bar
|
||||||
|
|
||||||
|
rec f0000:
|
||||||
|
date: 01-23-1998
|
||||||
|
inherit: test
|
||||||
|
inherit_order: 1
|
||||||
|
|
||||||
|
rec f0001:
|
||||||
|
date: 09-22-1998
|
||||||
|
inherit: test
|
||||||
|
inherit_order: 2
|
||||||
|
|
||||||
|
cat g0002:
|
||||||
|
inherit: test
|
||||||
|
inherit_order: 3
|
||||||
|
|
||||||
|
rec f0003:
|
||||||
|
date: 09-19-1998
|
||||||
|
inherit: g0002
|
||||||
|
inherit_order: 1
|
||||||
|
|
||||||
|
rec f0004:
|
||||||
|
date: 09-21-1998
|
||||||
|
inherit: g0002
|
||||||
|
inherit_order: 2
|
|
@ -0,0 +1,69 @@
|
||||||
|
class FileParsingError(SyntaxError):
|
||||||
|
pass
|
||||||
|
|
||||||
|
class Record:
|
||||||
|
def __init__(self):
|
||||||
|
self.name = None
|
||||||
|
self.data = {}
|
||||||
|
|
||||||
|
self.children = {}
|
||||||
|
self.parents = {}
|
||||||
|
|
||||||
|
def __getitem__(self, item):
|
||||||
|
return self.data[item]
|
||||||
|
|
||||||
|
def __setitem__(self, item, value):
|
||||||
|
self.data[item] = value
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"[{self.data}, {len(self.children)}, {len(self.parents)}]"
|
||||||
|
|
||||||
|
class RecordCollection():
|
||||||
|
def __init__(self, file):
|
||||||
|
if type(file) is str:
|
||||||
|
fd = open(file, "r")
|
||||||
|
elif hasattr(file, "read"):
|
||||||
|
fd = file
|
||||||
|
|
||||||
|
self._fromFile(fd)
|
||||||
|
self._parent()
|
||||||
|
|
||||||
|
def _fromFile(self, fd):
|
||||||
|
lines = [i.rstrip() for i in fd.readlines()]
|
||||||
|
self.objects = {}
|
||||||
|
current = Record()
|
||||||
|
|
||||||
|
for i in lines:
|
||||||
|
ind = i[0:2] == ' '
|
||||||
|
|
||||||
|
if ind:
|
||||||
|
spl = i[2:].split(': ')
|
||||||
|
try:
|
||||||
|
current[spl[0]] = spl[1]
|
||||||
|
except IndexError:
|
||||||
|
raise FileParsingError(f"error parsing '{i}'")
|
||||||
|
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
name = i.split(' ')[1][0:-1]
|
||||||
|
except IndexError:
|
||||||
|
current = Record()
|
||||||
|
continue
|
||||||
|
|
||||||
|
current.name = name
|
||||||
|
self.objects[name] = current
|
||||||
|
|
||||||
|
def _parent(self):
|
||||||
|
for i in self.objects:
|
||||||
|
current = self.objects[i]
|
||||||
|
try:
|
||||||
|
for j, k in zip(
|
||||||
|
current['inherit'].split(' '),
|
||||||
|
[int(i) for i in current['inherit_order'].split(' ')]
|
||||||
|
):
|
||||||
|
|
||||||
|
current.parents[j] = self.objects[j]
|
||||||
|
self.objects[j].children[k] = current
|
||||||
|
|
||||||
|
except KeyError:
|
||||||
|
pass
|
|
@ -0,0 +1,5 @@
|
||||||
|
import record
|
||||||
|
|
||||||
|
records = record.RecordCollection('datafile')
|
||||||
|
|
||||||
|
print(records.objects)
|
Loading…
Reference in New Issue