d{IgydZddlmZmZddlmZmZ ddlmZn#e $r ddl mZYnwxYwddl Z ddl Z ddl mZddlZddlmZd Z d Z d Z d Z d Z dZ dZ dZ dZ dZ dZ GddeZGddeZGddeZ GddeZ!GddeZ"Gdde#Z$Gdde%Z&GddeZ'Gd d!eZ(Gd"d#e%Z)Gd$d%e%Z*dS)&z Handle a region file, containing 32x32 chunks. For more information about the region file format: https://minecraft.gamepedia.com/Region_file_format )NBTFileMalformedFileError)packunpack)MappingN)BytesIO)SEEK_ENDic eZdZdZddZdZdS)RegionFileFormatErrorzBase class for all file format errors. Note: InconceivedChunk is not a child class, because it is not considered a format error.c||_dSNmsgselfrs L/var/www/CMSBlueprintStation/venv/lib/python3.11/site-packages/nbt/region.py__init__zRegionFileFormatError.__init__? c|jSrrrs r__str__zRegionFileFormatError.__str__As xrNr)__name__ __module__ __qualname____doc__rr rrrr<sCaarrceZdZdZdS)NoRegionHeaderz=The size of the region file is too small to contain a header.Nr"r#r$r%r&rrr(r(DsGGGGrr(ceZdZdZdS)RegionHeaderErrorz9Error in the header of the region file for a given chunk.Nr)r&rrr+r+GsCCCCrr+ceZdZdZdS)ChunkHeaderErrorzNError in the header of a chunk, included the bytes of length and byte version.Nr)r&rrr-r-JsXXXXrr-ceZdZdZdS)ChunkDataErrorzError in the data of a chunk.Nr)r&rrr/r/Ms''''rr/ceZdZdZddZdS)InconceivedChunkz+Specified chunk has not yet been generated.rc||_dSrrrs rrzInconceivedChunk.__init__RrrNr!)r"r#r$r%rr&rrr1r1Ps.55rr1c0eZdZdZdZdZdZdZdZdS) ChunkMetadataz_ Metadata for a particular chunk found in the 8 kiByte header and 5-byte chunk header. c||_ ||_ d|_ d|_ d|_ d|_ d|_ t|_dSNr) xz blockstart blocklength timestamplength compressionSTATUS_CHUNK_NOT_CREATEDstatusrr7r8s rrzChunkMetadata.__init__[sf33 +B &  B ! /  & &rc d|jj|j|j|j|j|j|j|j|j f zS)Nz]%s(%d, %d, sector=%s, blocklength=%s, timestamp=%s, bytelength=%s, compression=%s, status=%s)) __class__r"r7r8r9r:r;r<r=r?rs rr zChunkMetadata.__str__|sAn ^ $dfdfdotGWY]Yg K)4; 88 8rc:d|jj|j|jfzS)Nz %s(%d,%d))rBr"r7r8rs r__repr__zChunkMetadata.__repr__sdn5tvtvFFFrc6|jdztztzS)N)r< SECTOR_LENGTHrs rrequiredblockszChunkMetadata.requiredblockss a-/MAArc|jdkS)zreturn True if this chunk is created according to the header. This includes chunks which are not readable for other reasons.r)r9rs r is_createdzChunkMetadata.is_createds!##rN) r"r#r$r%rr rDrHrJr&rrr4r4Vsp & & &B888GGGBBB$$$$$rr4c*eZdZdZdZdZdZdZdS)_HeaderWrapperzDWrapper around self.metadata to emulate the old self.header variablec||_dSrmetadatarrOs rrz_HeaderWrapper.__init__   rcP|j|}|j|j|j|jfSr)rOr9r:r;r?rxzms r __getitem__z_HeaderWrapper.__getitem__s% M"  amQ[!(CCrc*t|jSriterrOrs r__iter__z_HeaderWrapper.__iter__DM"""rc*t|jSrlenrOrs r__len__z_HeaderWrapper.__len__4=!!!rNr"r#r$r%rrVrZr_r&rrrLrLsYNN!!!DDD###"""""rrLc*eZdZdZdZdZdZdZdS)_ChunkHeaderWrapperzKWrapper around self.metadata to emulate the old self.chunk_headers variablec||_dSrrNrPs rrz_ChunkHeaderWrapper.__init__rQrc^|j|}|jdkr|jnd|j|jfSr6)rOr<r=r?rSs rrVz_ChunkHeaderWrapper.__getitem__s/ M" HqLLdAM18LLrc*t|jSrrXrs rrZz_ChunkHeaderWrapper.__iter__r[rc*t|jSrr]rs rr_z_ChunkHeaderWrapper.__len__r`rNrar&rrrcrcsYUU!!!MMM###"""""rrcceZdZddZdZdS)LocationNc0||_||_||_dSr)r7yr8)rr7rkr8s rrzLocation.__init__srcP|jjd|jd|jd|jdS)Nz(x=z, y=z, z=))rBr"r7rkr8rs rr zLocation.__str__s2)-)@)@)@$&&&$&&&RVRXRXRXYYrNNN)r"r#r$rr r&rrriris<ZZZZZrric eZdZdZeZ eZ eZ eZ eZ e Z e Z d!dZ dZ e efdZdZdZdZd Zd Zd Zd"d Zd"d Zd#dZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#e$fdZ%dZ&dZ'dZ(dZ)d Z*dS)$ RegionFilezSA convenience class for extracting NBT files from the Minecraft Beta Region Format.Nc*d|_d|_d|_d|_ ||_|r$||_t |d|_d|_n<|r$t |dr |j|_||_n|jstdi|_ t|j |_ t|j |_ t|_ |||dS)z Read a region file by filename or file object. If a fileobj is specified, it is not closed after use; it is the callers responibility to close it. NFzr+bTnamez@RegionFile(): Need to specify either a filename or a file object)filefilename _closefileclosed chunkclassopenhasattrrr ValueErrorrOrLheaderrc chunk_headersriloc _init_header _parse_header_parse_chunk_headers)rrtfileobjrws rrzRegionFile.__init__s&    M$  a$DMXu--DI"DOO  aw'' - '  DII a_`` `  %T]33  (1?? ::@   !!#####rct|jdt|jS)z! Returns the file size in bytes. r)rsseekr tellrs rget_sizezRegionFile.get_size s, q(###y~~rcBt||\}}|dkr|n|dzS)a Given a size in bytes, return how many sections of length sectorlen are required to contain it. This is equivalent to ceil(bsize/sectorlen), if Python would use floating points for division, and integers for ceil(), rather than the other way around.rr)divmod)bsize sectorlengthsectors remainders r_bytes_to_sectorzRegionFile._bytes_to_sectors. $E<88#q..wwgk9rc||jr4 |jd|_dS#t$rYdSwxYwdS)a Clean up resources after use. Note that the instance is no longer readable nor writable after calling close(). The method is automatically called by garbage collectors, but made public to allow explicit cleanup. TN)rurscloservIOErrorrs rrzRegionFile.closes\ ?   !!!"        s + 99c.|dSr)rrs r__del__zRegionFile.__del__+s rcdtz}|j|kr|j||jd|j|dz||_dS)zLInitialise the file header. This will erase any data previously in the file.rrN)rGsizerstruncaterwrite)r header_lengths r _init_filezRegionFile._init_file/sf- 9} $ $ I  } - - - q  g-...! rctdD].}tdD]}t|||j||f</dS)N )ranger4rOr@s rr~zRegionFile._init_header8sXr 9 9A2YY 9 9%21a%8%8 ac"" 9 9 9rcR||_|jdkrdS|jdtzkrtd|jzt dtdD]`}t |dzdz}t |dzdz}|j||f}|j|tdd|j dz\}}||c|_ |_ |j|tztd |j dd|_ |dkr|dkrt|_|dkrt |_|dkr|dkrt"|_0t|zd z|jkrt$|_Tt&|_b|ddD]A}t+|d kr,|D])}|jt t"t$fvr t,|_*BdS) z=Read the region header and stores: offset, length and status.rNrz@The region file is %d bytes, too small in size to have a header.r>IBr>Ir)rrrGr(rintrOrsrrreadr9r:r;r>r?STATUS_CHUNK_ZERO_LENGTHSTATUS_CHUNK_IN_HEADERSTATUS_CHUNK_OUT_OF_FILESTATUS_CHUNK_OK_sectorsr^STATUS_CHUNK_OVERLAPPING)rindexr7r8rUoffsetr<chunkss rrzRegionFile._parse_header=s MMOO 9>> F Y= ( ( !cfjfo!opp p1mQ// + +EE1H "AE1H r!A ad#A INN5 ! ! !#E549>>!3D3D+DEENFF*0& 'AL!- INN5=0 1 1 1 ty~~a'8'899!Br)rrOr?rrSTATUS_CHUNK_MISMATCHED_LENGTHSrsrr9rGrrr<r=rrrrr:)rr7r8rUr<r=s rrzRegionFile._parse_chunk_headerslsgr ? ?A2YY ? ?M!Q$'8O5M$C$EEEINN1< #=>>>#D$)..*;*;<AH- ? ? ?sBC##C<;C<c||j}dt|D}d|d<d|d<|jD]}|s||kr|jr|jr~|jt|j| z}tt|jdt||D]}|| ||S)zd Return a list of all sectors, each sector is a list of chunks occupying the block. cg|]}gSr&r&).0ss r z'RegionFile._sectors..s111!2111rTrrr) rrrrOvaluesrJr:r9maxrHminappend)r ignore_chunk sectorsizerrUblockendbs rrzRegionFile._sectorss **4955 11uZ00111  %%'' ) )A<<>> q  } ) )<#amQ=M=M=O=O*P*PPs1<33S:5N5NOO))AAJ%%a((((rcF||}d|DS)z7Return a list of booleans, indicating the free sectors.rcg|]}| Sr&r&)ris rrz3RegionFile._locate_free_sectors..s'''!A'''r)r)rrrs r_locate_free_sectorszRegionFile._locate_free_sectorss)--\-::''w''''rrc|rt||||zr|Sd}|t|kr3t||||zrn|dz }|t|k3|S)z Given a list of booleans, find a list of consecutive True values. If no such list is found, return length(free_locations). Assumes first two values are always False. rr)allr^)rfree_locationsrequired_sectors preferredrs r_find_free_locationzRegionFile._find_free_locations  ^Ii@P6P,PQRR   #n%%%%>!A&6$6"6788  FA#n%%%%rcHd|jDS)z Return a list of the metadata of each chunk that is defined in te regionfile. This includes chunks which may not be readable for whatever reason, but excludes chunks that are not yet defined. c:g|]}||Sr&)rJ)rrUs rrz+RegionFile.get_metadata..s%DDDaQ\\^^DDDDr)rOrrs r get_metadatazRegionFile.get_metadatas' ED4=//11DDDDrc*|S)a Return the x,z coordinates and length of the chunks that are defined in te regionfile. This includes chunks which may not be readable for whatever reason. Warning: despite the name, this function does not actually return the chunk, but merely it's metadata. Use get_chunk(x,z) to get the NBTFile, and then Chunk() to get the actual chunk. This method is deprecated. Use :meth:`get_metadata` instead. )get_chunk_coordsrs r get_chunkszRegionFile.get_chunkss$$&&&rcg}tdD]U}tdD]C}|j||f}|r||||jdDV|S)a Return the x,z coordinates and length of the chunks that are defined in te regionfile. This includes chunks which may not be readable for whatever reason. This method is deprecated. Use :meth:`get_metadata` instead. r)r7r8r<)rrOrJrr:)rrr7r8rUs rrzRegionFile.get_chunk_coordssr M MA2YY M MM!A#&<<>>MMMQ]"K"KLLL M rc#K|D]5} ||j|jV&#t$rY2wxYwdS)a% Yield each readable chunk present in the region. Chunks that can not be read for whatever reason are silently skipped. Warning: this function returns a :class:`nbt.nbt.NBTFile` object, use ``Chunk(nbtfile)`` to get a :class:`nbt.chunk.Chunk` instance. N)r get_chunkr7r8rrrUs r iter_chunkszRegionFile.iter_chunksss""$$  A nnQS!#......(      s"= A  A c#K|D]H} |||j|jV9#t $rYEwxYwdS)z Yield each readable chunk present in the region. Chunks that can not be read for whatever reason are silently skipped. This function returns a :class:`nbt.chunk.Chunk` instance. N)rrwrr7r8rrs riter_chunks_classzRegionFile.iter_chunks_classs ""$$  A oodnnQS!#&>&>??????(      s5A AAc*|Sr)rrs rrZzRegionFile.__iter__s!!!rc*|j||fjS)a` Return the timestamp of when this region file was last modified. Note that this returns the timestamp as-is. A timestamp may exist, while the chunk does not, or it may return a timestamp of 0 even while the chunk exists. To convert to an actual date, use `datetime.fromtimestamp()`. )rOr;r@s r get_timestampzRegionFile.get_timestamps}QqS!++rcDt|S)zNReturn the number of defined chunks. This includes potentially corrupt chunks.)r^rrs r chunk_countzRegionFile.chunk_counts4$$&&'''rcd|j||f}|jtkrtd||fz|jtkrt d||fz|jt kr*|jdks |jdkrt d||fz|jtkr3|j dkrt d||fztd||fz|j tzdz|jkrt d||fzd} |j|j tzdzt#|jdz |j|j tzdzz }|j|}|jt&krXt)jt-| }t/|}|nL|jt2krt5j|}n'|jt8krt;d |jz|S#t<$rt>$r}d |z}Yd}~nd}~wwxYw|rW|jt@krtd ||fz|jtBkrtd ||fzt;|dS)aY Return the decompressed binary data representing a chunk. May raise a RegionFileFormatError(). If decompression of the data succeeds, all available data is returned, even if it is shorter than what is specified in the header (e.g. in case of a truncated while and non-compressed data). z$Chunk %d,%d is not present in regionz#Chunk %d,%d is in the region headerrNz4Chunk %d,%d is partially/completely outside the filerzChunk %d,%d has zero lengthrrz%Unknown chunk compression/format (%s)%szXThe length in region header and the length in the header of chunk %d,%d are incompatiblez-Chunk %d,%d is overlapping with another chunk)"rOr?r>r1rr+rr<r=rr:r-r9rGrrsrrrCOMPRESSION_GZIPgzipGzipFiler bytesrCOMPRESSION_ZLIBzlib decompressCOMPRESSION_NONEr/r Exceptionrr) rr7r8rUerrr<chunkfes r get_blockdatazRegionFile.get_blockdata s M!Q$  8/ / /"#IQqE#QRR R X/ / /#$IQqE$QRR R X1 1 1qx1}} Y]H]H]#$Z^_`a]b$bcc c X1 1 1}!!'(E1(MNNN&'D!u'LMMM \M )A - : :#$Z^_`a]b$bcc c  INN1<-7!; < < <AtyAL=4PST4T'UVVFINN6**E !111M'%..999affhh -#333.."222$%Lq}%\]]]L$       (CCCCCC   *x:::&(BFGHIEJ(JKKK555&'VZ[\]Y^'^___$S))) * *sD)H11IIIc|||}t|}d} t|}|jjdkr||jjdzz }|jjdkr||jjdzz }t |||_|S#t$r}d|z}Yd}~nd}~wwxYw|rt|dS)z Return a NBTFile of the specified chunk. Raise InconceivedChunk if the chunk is not included in the file. Nbufferr)r7r8r) rr rr}r7r8rirr/)rr7r8datarnbtrs rget_nbtzRegionFile.get_nbtOs !!!Q''t}} &&&CxzT!!TXZ]"xzT!!TXZ]"a(((CGJ!   (CCCCCC   & %% % & &sA+B B.B))B.c.|||S)a( Return a NBTFile of the specified chunk. Raise InconceivedChunk if the chunk is not included in the file. Note: this function may be changed later to return a Chunk() rather than a NBTFile() object. To keep the old functionality, use get_nbt(). )rr@s rrzRegionFile.get_chunkfs||Aq!!!rc  |tkrwt}tj|}||||d|}~n=|tkrtj |}n|tkrtd|zt|}||dz}|dkrtd|z|jdt"zkr||j||f} || } || || j } |j| t"zkrt|jdt0|j| t"z|jz d z|j| t"zksJ|j| t"z|jt5d |d z|jt5d ||j|t"|z|z dz } |j| d z|jd|d|zzz|jt5d| |d d|jt"d|d|zzzzt7t9j} |jt5d | | | |zt| z dgzt=| | |zD]}d| |<t?tA| !d}|dkrIt"t| |z z|_|j"|j| d| } t=| jtG| j| j$zt| D]N}| |rD|jt"|z|jt"d zOtK| |zt"z|j|_|&|jksJ| | _|| _$tN| _(| | _)|d z| _*t| _+dS)z Compress the data, write it to file, and add pointers in the header so it can be found as chunk(x,z). rrzUnknown compression type %drz3Chunk is too large (%d sectors exceeds 255 maximum)rr)rrrrrrrrNTF),rr rrrrrrrrcompressrrzr^rr/rrGrrOrrr9rsr rrrtimeextendrlistreversedrrrr:rrrr?r;r<r=)rr7r8rr=compressed_filerr<nsectorscurrent free_sectorssectorremaining_lengthr;rtruncate_counts rwrite_blockdatazRegionFile.write_blockdataps * * *%iiO o666A GGDMMM GGIII   # # #"''))D , , ,=&&DD , , ,:[HII IT((!44 s?? !VZb!cdd d 9q & & OO   -1%00g0FF )),GL^)__ 9vm+ + + INN1h ' ' ' IOOVM1DI=H I I I9>>##vm';;;;; vm+,,, T6A:../// T;//000 )83f $ / />* *"n7779P9PQ Qrc|jr d|d|jdSd|t|fzS)N<(z)><%s object at %d>rtr idrs rr zRegionFile.__str__sT = G G!%!2!2!2!2DMMMB B&$//*;*;RXX)FF Frc|jr|d|jdSd|t|fzS)Nr rmr rrs rrDzRegionFile.__repr__sN = G#0000$---@ @&$//*;*;RXX)FF Frrnr)rN)+r"r#r$r%rrrrrrr>rr staticmethodrGrrrrr~rrrrrrrrrrrZrrrrrrrrrr r rDr&rrrprpsv]] 8M'F#m7M3K7M%OD7MG$G$G$G$R   -::::\:   """999 -<-<-<^???4((((( .EEE ' ' '      "   """ , , ,(((B*B*B*H&&&."""7GY/Y/Y/Y/|444 2 2 2DRRRGGG GGGGGrrp)+r%rrrstructrrcollections.abcr ImportError collectionsrrior rosr rGrrrrrrr>rrrrrr(r+r-r/ LookupErrorr1objectr4rLrcrirpr&rrrsC-,,,,,,,$'''''''$$$########$   io"$iFUUVD;<< IHHHHH*HHHDDDDD-DDDYYYYY,YYY(((((*((({ 2$2$2$2$2$F2$2$2$h " " " " "W " " " " " " " "' " " "ZZZZZvZZZ` G` G` G` G` G` G` G` G` G` Gs  ))