Add nbtlib.contrib package? about nbtlib HOT 1 OPEN

import zlib, gzip


def GetAnvilChunkData(region, x: int, z: int) -> dict:
    """Returns info stored in anvil header about a chunk at chunk coordinates
    Arguments:
    region -- a file object created with open('/path/')
    x, z -- chunk position (regardless if its region-local or global)
    """
    region.seek(4 * ((x % 32) + (z % 32) * 32), 0)

    offset = int.from_bytes(region.read(3), byteorder ='big')
    sectors = int.from_bytes(region.read(1), byteorder ='big')
    
    region.seek(4 * ((x % 32) + (z % 32) * 32) + 0x1000, 0)
    
    timestamp = int.from_bytes(region.read(4), byteorder ='big')
    
    return dict({'offset':    offset,
                'sectors':   sectors,
                 'timestamp': timestamp})

def GetChunkNbtData(region, x: int, z: int):
    """Finds chunk data sectors and returns uncompressed buffer with NBT data"""
    anvil_data = GetAnvilChunkData(region, x, z)

    if anvil_data['offset'] < 2:
        return bytes(0)
    
    region.seek(anvil_data['offset'] * 0x1000)

    length = int.from_bytes(region.read(4), byteorder ='big')
    compression_type = int.from_bytes(region.read(1), byteorder ='big')
    
    if compression_type == 3:
        return region.read(length)

    if compression_type == 2:
        return zlib.decompress(region.read(length))
    
    if compression_type == 1:
        return gzip.decompress(region.read(length))
    

#!/bin/python
import nbtlib, io
from mvc.mca import GetChunkNbtData


def main():
    file = open('data/region.mca', 'rb')
    nbt_file = nbtlib.File.from_fileobj(io.BytesIO(GetChunkNbtData(file, 3, 3)))
    print(nbt_file.root.keys())

main()