explicit public apis

Author: sungwy

pyiceberg/avro/__init__.py

@@ -21,3 +21,8 @@
 STRUCT_DOUBLE = struct.Struct("<d")  # little-endian double
 STRUCT_INT32 = struct.Struct("<i")
 STRUCT_INT64 = struct.Struct("<q")
+
+# Imported after struct constants to avoid circular imports in encoder/decoder.
+from pyiceberg.avro.file import AvroFile, AvroFileHeader, AvroOutputFile  # noqa: E402
+
+__all__ = ["AvroFile", "AvroFileHeader", "AvroOutputFile"]

pyiceberg/avro/codecs/__init__.py

@@ -50,3 +50,10 @@
 
 # Map to convert the naming from Iceberg to Avro
 CODEC_MAPPING_ICEBERG_TO_AVRO: dict[str, str] = {"gzip": "deflate", "zstd": "zstandard"}
+
+__all__ = [
+    "AvroCompressionCodec",
+    "AVRO_CODEC_KEY",
+    "CODEC_MAPPING_ICEBERG_TO_AVRO",
+    "KNOWN_CODECS",
+]

pyiceberg/avro/decoder.py

@@ -17,14 +17,14 @@
 import io
 from abc import ABC, abstractmethod
 from io import SEEK_CUR
-from typing import (
-    cast,
-)
+from typing import cast
 
 from pyiceberg.avro import STRUCT_DOUBLE, STRUCT_FLOAT
 from pyiceberg.io import InputStream
 from pyiceberg.typedef import UTF8
 
+__all__: list[str] = []
+
 
 class BinaryDecoder(ABC):
     """Decodes bytes into Python physical primitives."""

pyiceberg/avro/encoder.py

@@ -21,6 +21,8 @@
 from pyiceberg.io import OutputStream
 from pyiceberg.typedef import UTF8
 
+__all__: list[str] = []
+
 
 class BinaryEncoder:
     """Encodes Python physical types into bytes."""

pyiceberg/avro/file.py

@@ -50,6 +50,8 @@
 )
 from pyiceberg.utils.schema_conversion import AvroSchemaConversion
 
+__all__: list[str] = []
+
 VERSION = 1
 MAGIC = bytes(b"Obj" + bytearray([VERSION]))
 MAGIC_SIZE = len(MAGIC)

pyiceberg/avro/reader.py

@@ -44,6 +44,8 @@
 from pyiceberg.utils.lazydict import LazyDict
 from pyiceberg.utils.singleton import Singleton
 
+__all__: list[str] = []
+
 
 def _skip_map_array(decoder: BinaryDecoder, skip_entry: Callable[[], None]) -> None:
     """Skips over an array or map.

pyiceberg/avro/resolver.py

@@ -107,6 +107,8 @@
     UUIDType,
 )
 
+__all__: list[str] = []
+
 STRUCT_ROOT = -1
 
 

pyiceberg/avro/writer.py

@@ -36,6 +36,8 @@
 from pyiceberg.utils.decimal import decimal_required_bytes, decimal_to_bytes
 from pyiceberg.utils.singleton import Singleton
 
+__all__: list[str] = []
+
 
 @dataclass(frozen=True)
 class Writer(Singleton):