use context managers in RecordCollection.__init__() and RecordCollection.write()

This commit is contained in:
randomuser 2021-11-22 18:43:08 -06:00
parent f66776c9bb
commit a0ebba802e
1 changed files with 19 additions and 23 deletions

View File

@ -28,13 +28,10 @@ class Record:
class RecordCollection():
def __init__(self, file):
fd = filewrapper(file, "r")
with filewrapper(file, "r") as fd:
self._fromFile(fd)
self._parent()
fd.close()
def _fromFile(self, fd):
lines = [i.rstrip() for i in fd.readlines()]
self.objects = {}
@ -85,7 +82,7 @@ class RecordCollection():
return ret
def write(self, file):
fd = filewrapper(file, "w")
with filewrapper(file, "w") as fd:
lines = []
for i in self.objects:
@ -107,4 +104,3 @@ class RecordCollection():
lines[i] += "\n"
fd.writelines(lines)
fd.close()