1 files changed, 41 insertions, 0 deletions
diff --git a/bcdd/PBNFile.py b/bcdd/PBNFile.py
new file mode 100644
index 0000000..16f4d6d
--- /dev/null
+++ b/bcdd/PBNFile.py
@@ -0,0 +1,41 @@
+import shutil
+import tempfile
+
+from .PBNBoard import PBNBoard
+
+class PBNFile(object):
+
+    def __init__(self, filename):
+        self._filename = filename
+        self.output_file = None
+        self.boards = []
+        lines = []
+        with open(self._filename) as pbn_file:
+            contents = pbn_file.readlines()
+        for line in contents:
+            line = line.strip()
+            if not line:
+                if len(lines) > 0:
+                    self.boards.append(PBNBoard(lines))
+                    lines = []
+            else:
+                lines.append(line)
+        if len(lines) > 0:
+            self.boards.append(PBNBoard(lines))
+        if not self.boards[0].has_field('Event'):
+            self.boards[0].write_event('')
+
+    def write_board(self, board):
+        if self.output_file is None:
+            self.output_file = tempfile.NamedTemporaryFile(
+                mode='w', encoding='utf-8', delete=False)
+        for field in board.fields:
+            self.output_file.write(field.raw_field + '\n')
+        self.output_file.write('\n')
+
+    def save(self):
+        if self.output_file is None:
+            raise IOError('No boards written to PBN file, unable to save it.')
+        tmp_path = self.output_file.name
+        self.output_file.close()
+        shutil.move(tmp_path, self._filename)