Add wandio options parameter (#10)

Add wandio options parameter. Allows wandio options to be passed in when creating an AvroRsdosReader object to allow for accessing multiple swift projects in the same code. --------- Co-authored-by: Nicholas264 <[email protected]>
CAIDA · Jun 26, 2023 · 7ab082d · 7ab082d
1 parent 957a3a0
commit 7ab082d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/pyavro_stardust/baseavro.pxd b/src/pyavro_stardust/baseavro.pxd
@@ -93,6 +93,7 @@ cdef class AvroReader:
     cdef bytes unzipped
     cdef AvroRecord currentrec
     cdef dict schemajson
+    cdef dict options
 
     cpdef void _readAvroFileHeader(self)
     cdef int _parseNextRecord(self, const unsigned char[:] buf,

diff --git a/src/pyavro_stardust/baseavro.pyx b/src/pyavro_stardust/baseavro.pyx
@@ -302,8 +302,9 @@ cdef class AvroRecord:
         self.schemaversion = schemaversion
 
 cdef class AvroReader:
-    def __init__(self, filepath):
+    def __init__(self, filepath, options=None):
         self.filepath = filepath
+        self.options = options
         self.syncmarker = None
         self.fh = None
         self.bufrin = bytearray()
@@ -397,7 +398,7 @@ cdef class AvroReader:
         saved = None
         while mode != 'fail':
             try:
-                self.fh = wandio.open(self.filepath, mode=mode)
+                self.fh = wandio.open(self.filepath, mode=mode, options=self.options)
             except ValueError as e:
                 if mode == 'rb':
                     mode = 'r'

diff --git a/src/pyavro_stardust/rsdos.pyx b/src/pyavro_stardust/rsdos.pyx
@@ -177,8 +177,8 @@ cdef class AvroRsdos(AvroRecord):
 @cython.final
 cdef class AvroRsdosReader(AvroReader):
 
-    def __init__(self, filepath):
-        super().__init__(filepath)
+    def __init__(self, filepath, options=None):
+        super().__init__(filepath, options)
         self.currentrec = AvroRsdos()
 
     cdef int _parseNextRecord(self, const unsigned char[:] buf,