Browse Source

startup: add function `join_pgp_packets`

Fabian Peter Hammerle 2 years ago
parent
commit
98af05958f
1 changed files with 8 additions and 0 deletions
  1. 8 0
      profile_default/startup/init.py

+ 8 - 0
profile_default/startup/init.py

@@ -17,6 +17,14 @@ pandas.options.display.max_rows = 200
 sympy.init_printing(pretty_print=True)
 
 
+def join_pgp_packets(
+    packets: typing.Iterator[typing.Union[bytearray, pgpdump.packet.Packet]],
+) -> bytes:
+    return b"".join(
+        p.data if isinstance(p, pgpdump.packet.Packet) else p for p in packets
+    )
+
+
 def numpy_array_from_file(
     path: typing.Union[str, pathlib.Path], dtype
 ) -> numpy.ndarray: