iajoiner commented on a change in pull request #9702:
URL: https://github.com/apache/arrow/pull/9702#discussion_r787405908
##########
File path: python/pyarrow/_orc.pyx
##########
@@ -36,7 +36,233 @@ from pyarrow.lib cimport (check_status, _Weakrefable,
pyarrow_unwrap_table,
get_reader,
get_writer)
-from pyarrow.lib import tobytes
+from pyarrow.lib import frombytes, tobytes
+
+
+cdef compression_type_from_enum(CCompressionType compression_type_):
+ return {
+ CCompressionType_UNCOMPRESSED: 'UNCOMPRESSED',
+ CCompressionType_GZIP: 'ZLIB',
+ CCompressionType_SNAPPY: 'SNAPPY',
+ CCompressionType_LZ4: 'LZ4',
+ CCompressionType_ZSTD: 'ZSTD',
+ }.get(compression_type_, 'UNKNOWN')
+
+
+cdef CCompressionType compression_type_from_name(name) except *:
+ if not isinstance(name, str):
+ raise TypeError('compression must be a string')
+ name = name.upper()
+ if name == 'ZLIB':
+ return CCompressionType_GZIP
+ elif name == 'SNAPPY':
+ return CCompressionType_SNAPPY
+ elif name == 'LZ4':
+ return CCompressionType_LZ4
+ elif name == 'ZSTD':
+ return CCompressionType_ZSTD
+ elif name == 'UNCOMPRESSED':
+ return CCompressionType_UNCOMPRESSED
+ raise ValueError('Unknown CompressionKind: {0}'.format(name))
Review comment:
Yup dropping 3.5 support does make things easier. :)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
