| 109 | |
| 110 | |
| 111 | class CompressionSettings: |
| 112 | def __init__(self, compressors: list[str], zlib_compression_level: int): |
| 113 | self.compressors = compressors |
| 114 | self.zlib_compression_level = zlib_compression_level |
| 115 | |
| 116 | def get_compression_context( |
| 117 | self, compressors: Optional[list[str]] |
| 118 | ) -> Union[SnappyContext, ZlibContext, ZstdContext, None]: |
| 119 | if compressors: |
| 120 | chosen = compressors[0] |
| 121 | if chosen == "snappy": |
| 122 | return SnappyContext() |
| 123 | elif chosen == "zlib": |
| 124 | return ZlibContext(self.zlib_compression_level) |
| 125 | elif chosen == "zstd": |
| 126 | return ZstdContext() |
| 127 | return None |
| 128 | return None |
| 129 | |
| 130 | |
| 131 | class SnappyContext: |
no outgoing calls
no test coverage detected