minor refactor

Author: PokkaKiyo

litestar/middleware/compression/zstd_facade.py

@@ -15,6 +15,7 @@
     except ImportError as e:
         raise MissingDependencyException("backports.zstd", extra="zstd") from e
 
+
 if TYPE_CHECKING:
     from io import BytesIO
 
@@ -41,11 +42,9 @@ def write(
         body: bytes | bytearray,
         final: bool = False,
     ) -> None:
-        if not final:
-            self.buffer.write(self.compressor.compress(body, mode=zstd.ZstdCompressor.FLUSH_BLOCK))
-        else:
-            self.buffer.write(self.compressor.compress(body, mode=zstd.ZstdCompressor.FLUSH_FRAME))
+        mode = zstd.ZstdCompressor.FLUSH_FRAME if final else zstd.ZstdCompressor.FLUSH_BLOCK
+        self.buffer.write(self.compressor.compress(body, mode=mode))
 
     def close(self) -> None:
-        if self.compressor.last_mode is not zstd.ZstdCompressor.FLUSH_FRAME:
+        if self.compressor.last_mode != zstd.ZstdCompressor.FLUSH_FRAME:
             self.buffer.write(self.compressor.flush())