black format writer

Signed-off-by: Zen <[email protected]>

Author: desultory

src/pycpio/writer/__init__.py

@@ -1,3 +1,3 @@
 from .writer import CPIOWriter
 
-__all__ = ['CPIOWriter']
+__all__ = ["CPIOWriter"]

src/pycpio/writer/writer.py

@@ -1,9 +1,9 @@
-from zenlib.logging import loggify
-from pycpio.header import CPIOHeader, HEADER_NEW
-
-from pathlib import Path
-from os import fsync
 from lzma import CHECK_CRC32
+from os import fsync
+from pathlib import Path
+
+from pycpio.header import HEADER_NEW, CPIOHeader
+from zenlib.logging import loggify
 
 
 @loggify
@@ -12,7 +12,17 @@ class CPIOWriter:
     Takes a list of CPIOData objects, gets their bytes representation, then appends a trailer before writing them to a file.
     Compresses the data if compression is specified.
     """
-    def __init__(self, cpio_entries: list, output_file: Path, structure=None, compression=False, xz_crc=CHECK_CRC32, *args, **kwargs):
+
+    def __init__(
+        self,
+        cpio_entries: list,
+        output_file: Path,
+        structure=None,
+        compression=False,
+        xz_crc=CHECK_CRC32,
+        *args,
+        **kwargs,
+    ):
         self.cpio_entries = cpio_entries
         self.output_file = Path(output_file)
 
@@ -21,34 +31,35 @@ def __init__(self, cpio_entries: list, output_file: Path, structure=None, compre
         self.compression = compression or False
         if isinstance(compression, str):
             compression = compression.lower()
-            if compression == 'true':
+            if compression == "true":
                 compression = True
-            elif compression == 'false':
+            elif compression == "false":
                 compression = False
             self.compression = compression
 
         self.xz_crc = xz_crc
 
     def __bytes__(self):
-        """ Creates a bytes representation of the CPIOData objects. """
+        """Creates a bytes representation of the CPIOData objects."""
         cpio_bytes = bytes(self.cpio_entries)
         trailer = CPIOHeader(structure=self.structure, name="TRAILER!!!", logger=self.logger)
         self.logger.debug("Building trailer: %s" % trailer)
         cpio_bytes += bytes(trailer)
         return cpio_bytes
 
     def compress(self, data):
-        """ Attempts to compress the data using the specified compression type. """
-        if self.compression == 'xz' or self.compression is True:
+        """Attempts to compress the data using the specified compression type."""
+        if self.compression == "xz" or self.compression is True:
             import lzma
-            self.logger.info("XZ compressing the CPIO data, original size: %.2f MiB" % (len(data) / (2 ** 20)))
+
+            self.logger.info("XZ compressing the CPIO data, original size: %.2f MiB" % (len(data) / (2**20)))
             data = lzma.compress(data, check=self.xz_crc)
         elif self.compression is not False:
             raise NotImplementedError("Compression type not supported: %s" % self.compression)
         return data
 
     def write(self, safe_write=True):
-        """ Writes the CPIOData objects to the output file. """
+        """Writes the CPIOData objects to the output file."""
         self.logger.debug("Writing to: %s" % self.output_file)
         data = self.compress(bytes(self))
         with open(self.output_file, "wb") as f:
@@ -60,5 +71,4 @@ def write(self, safe_write=True):
             else:
                 self.logger.warning("File not fsynced, data may not be written to disk: %s" % self.output_file)
 
-        self.logger.info("Wrote %.2f MiB to: %s" % (len(data) / (2 ** 20), self.output_file))
-
+        self.logger.info("Wrote %.2f MiB to: %s" % (len(data) / (2**20), self.output_file))