Merge pull request #1 from BravelyPeculiar/master

Added get_bytes() to Writer for convenience
zeldamods · Apr 7, 2019 · 508c590 · 508c590
2 parents e528ec9 + 5f9adee
commit 508c590
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/byml/byml.py b/byml/byml.py
@@ -2,6 +2,7 @@
 # Licensed under GPLv2+
 from enum import IntEnum
 from sortedcontainers import SortedDict # type: ignore
+import io
 import struct
 import typing
 
@@ -235,6 +236,11 @@ def __init__(self, data: typing.Union[dict, list], be=False, version=2) -> None:
         self._sort_string_table(self._hash_key_table)
         self._sort_string_table(self._string_table)
 
+    def get_bytes(self):
+        stream = io.BytesIO()
+        self.write(stream)
+        return stream.getvalue()
+
     def write(self, stream: typing.BinaryIO) -> None:
         # Header
         stream.write(b'BY' if self._be else b'YB')