Easily read avro records from bytes

read_avro.py

def read_avro(bytes_buff_or_file):
    import avro.schema
    from avro.datafile import DataFileReader, DataFileWriter
    from avro.io import DatumReader, DatumWriter
    from io import BytesIO
    import json
    if type(bytes_buff_or_file) == bytes:
        from tempfile import TemporaryFile
        # Has to be a file yes or yes because of how avro lib is made
        with TemporaryFile() as tmp:
            tmp.write(bytes_buff_or_file)
            reader = DataFileReader(tmp, DatumReader())
            return [json.loads(o.decode()) for o in reader]
    else:
        raise NotImplementedError