o 0Of@sdZdZddlmZmZmZmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlZe eZdZdZdZdZd Zd Zd Zd Zd ZdZdZdZ dZ!dZ"dZ#ddddddddddddd d!d"Z$dZ%dZ&d#Z'dZ(d$Z)d Z*d%d&d'd(d)d*d+Z+e j,ddkZ-e-re.Z/e0Z1dd,l2m3Z3m4Z4dd-l5m6Z6dd.l7m8Z8m9Z9ndd/l:m1Z1dd,l3m3Z3m4Z4dd-l;m6Z6dd.l;m8Z8m9Z9dd0gZdkd6d7Z?d8d9Z@ndkd:d5Z>dkd;d7Z?dd?ZAnd@d?ZAGdAdBdBe j ZBdldDdEZCdFdGZDdHdIZEdJdKZFdLdMZGdNdOZHdPdQZIdldRdSZJdTdUZKdmdVdWZLGdXdYdYeMZNGdZd[d[eOZPGd\d]d]eMZQGd^d_d_eOZRGd`dadaeOZSGdbdcdceTZUGdddedeeMZVGdfdgdgeMZWdhdiZXedjkr eXZYe ZeYdSdS)nz shapefile.py Provides read and write support for ESRI Shapefiles. authors: jlawheadgeospatialpython.com maintainer: karim.bahgat.norwaygmail.com Compatible with Python versions 2.7-3.x z2.3.1)packunpackcalcsizeerrorStructN)dateT NULLPOINTPOLYLINEPOLYGON MULTIPOINTPOINTZ POLYLINEZPOLYGONZ MULTIPOINTZPOINTM POLYLINEMPOLYGONM MULTIPOINTM MULTIPATCH)rrr r r r r rrrrrrrTRIANGLE_STRIP TRIANGLE_FAN OUTER_RING INNER_RING FIRST_RINGRING)rrr#r r$r )urlparse urlunparse) HTTPError)urlopenRequest)izipgJutf-8strictcC@t|tr |||St|tr|S|durdSt|||S)N) isinstancestrencodebytesvencodingencodingErrorsr>B/home/ubuntu/webapp/venv/lib/python3.10/site-packages/shapefile.pybg   r@cCr4Nr1)r6r9decoder7r:r>r>r?uurArDcC t|tSN)r6r7r;r>r>r? is_string rHcCr4rB)r6unicoder8r9r:r>r>r?r@rAcCr4rB)r6r9rCrJr:r>r>r?rDrAcCrErF)r6 basestringrGr>r>r?rHrI)r cCst|tjr t|S|SrF)r6osPathLikefsdecodepathr>r>r? pathlike_objs  rRcCs8t|r|St|dr|Szt|WS|YS)N __fspath__)rHhasattrrSr7rPr>r>r?rRs  c@seZdZdZddZdS)_ArrayzmConverts python tuples to lists of the appropriate type. Used to unpack different shapefile header parts.cCs t|SrF)r7tolistselfr>r>r?__repr__ z_Array.__repr__N)__name__ __module__ __qualname____doc__rYr>r>r>r?rUs rUFcsltttt|dd\ddtfddtdt|D}|r2|S|dS)a Return the signed area enclosed by a ring using the linear time algorithm. A value >= 0 indicates a counter-clockwise oriented ring. A faster version is possible by setting 'fast' to True, which returns 2x the area, e.g. if you're only interested in the sign of the area. Nr#rc3s0|]}||d|dVqdS)rNr>).0ixsysr>r? s.zsigned_area..g@)maplistzipappendsumrangelen)coordsfastarea2r>rar? signed_areas"rocCst|dd}|dkS)zkReturns True if a polygon ring has clockwise orientation, determined by a negatively signed area. Trmr)ro)rlrnr>r>r?is_cws rqcCs tt|S)z0Returns the input coords in reversed order. )rfreversedrlr>r>r?rewinds rtcCs,t|\}}t|t|t|t|f}|S)z7Calculates and returns the bounding box of a ring. )rgminmax)rlrbrcbboxr>r>r? ring_bboxs rxc Cs<|\}}}}|\}}}} ||ko||ko|| ko||k} | S)zBTests whether two bounding boxes overlap, returning a boolean r>) bbox1bbox2xmin1ymin1xmax1ymax1xmin2ymin2xmax2ymax2overlapr>r>r? bbox_overlap   rc Cs<|\}}}}|\}}}} ||ko||ko||ko|| k} | S)zBTests whether bbox1 fully contains bbox2, returning a boolean r>) ryrzr{r|r}r~rrrrcontainsr>r>r? bbox_containsrrc Cs|\}}|d}|d|k}d}|ddD]C}|d|k}||krU|d|k} | |d|kkr6| r5| }n|d|d||d|d|d|d|krU| }|}|}q|S)aFast point-in-polygon crossings algorithm, MacMartin optimization. Adapted from code by Eric Haynes http://www.realtimerendering.com/resources/GraphicsGems//gemsiv/ptpoly_haines/ptinpoly.c Original description: Shoot a test ray along +X axis. The strategy, from MacMartin, is to compare vertex Y values to the testing point's Y and quickly discard edges which are entirely to one side of the test ray. rrFNr>) rlptxtyvtx0yflag0 inside_flagvtx1yflag1xflag0r>r>r?ring_contains_points"    8rc sg}fdd}|D]x}||vr||t|dkr|dd|dd|dd|dd|dd|dd|dd|ddk}|s~||dg}t| }||kr~t|\}} t|dt| d} } t| | fr~| | fS|dq td) aMReturn a sample point guaranteed to be within a ring, by efficiently finding the first centroid of a coordinate triplet whose orientation matches the orientation of the ring and passes the point-in-ring test. The orientation of the ring is assumed to be clockwise, unless ccw (counter-clockwise) is set to True. c3s D]}|VqdVdS)Nrr>)rrsr>r? itercoords%szring_sample..itercoordsr rrr#g@z5Unexpected error: Unable to find a ring sample point.)rhrkrqrgrirpop Exception) rlccwtripletrris_straight_lineclosed_triplet triplet_ccwrbrcxmeanymeanr>rsr? ring_samples$    `    rcstfdd|DS)zFReturns True if all vertexes in coords2 are fully inside coords1. c3s|]}t|VqdSrF)r)r_p2coords1r>r?rdLsz%ring_contains_ring..)all)rcoords2r>rr?ring_contains_ringIsrcsgg}|D]}t|r|q||qtdkr*dg|}|g}|Stdkr=|sDg}D] }|g}||q7|Stddtt|D}ddD}|D]} t|| } t|D]\} } t| | rx|| | qhq\| D]2\} } t| dkrt||  }t || |d}g}| D]} t | |}|r|| q||| <q~| D]\} } t| dkrt | fddd d} | g|| <qg}t | D]\} } | s|| || =qqg}tD]+\} }|g}g}t | D]\} } | d| kr||| q||7}||q|D]} || }|g}||q|r;|d ur;t||d <|S|d urHt||d <|d dD}|S)aOrganize a list of coordinate rings into one or more polygons with holes. Returns a list of polygons, where each polygon is composed of a single exterior ring, and one or more interior holes. If a return_errors dict is provided (optional), any errors encountered will be added to it. Rings must be closed, and cannot intersect each other (non-self-intersecting polygon). Rings are determined as exteriors if they run in clockwise direction, or interior holes if they run in counter-clockwise direction. This method is used to construct GeoJSON (multi)polygons from the shapefile polygon shape type, which does not explicitly store the structure of the polygons beyond exterior/interior ring orientation. rrcSsg|]}|gfqSr>r>)r_hole_ir>r>r? {z*organize_polygon_rings..cSg|]}t|qSr>)rx)r_ringr>r>r?r|r)rcstt|ddS)NTrp)absro)x exteriorsr>r?rz(organize_polygon_rings..)keyNpolygon_orphaned_holespolygon_only_holescSsg|]}|gqSr>r>)r_extr>r>r?r)rqrhrkdictxrangekeysrx enumerateritemsrrsortedrf)rings return_errorsholesrpolypolysrhole_exteriorsexterior_bboxesr hole_bboxext_iext_bboxexterior_candidatesr hole_samplenew_exterior_candidateshole_in_exterior orphan_holes poly_holesr>rr?organize_polygon_ringsNs                 rc@sXeZdZeddddfddZeddZeddZedd Z ed d Z d d Z dS)ShapeNcCsF||_|pg|_|p g|_|r||_i|_|dur||_dSd|_dS)aaStores the geometry of the different shape types specified in the Shapefile spec. Shape types are usually point, polyline, or polygons. Every shape type except the "Null" type contains points at some level for example vertices in a polygon. If a shape type has multiple shapes containing points within a single geometry record then those shapes are called parts. Parts are designated by their starting index in geometry record's list of shapes. For MultiPatch geometry, partTypes designates the patch type of each of the parts. N) shapeTypepointsparts partTypes_errors _Shape__oid)rXrrrroidr>r>r?__init__s    zShape.__init__c Cs||jtttfvrt|jdkrdtdSdt|jddS|jttt fvr>t|jdkr3dgdSddd|jDdS|jt t t fvrt|j dkrRdgdSt|j dkrddd d|jDdSd}g}|j D]}|dkrt|}qk|d d|j||D|}qk|d d|j|dDd |dS|jtttfvr5t|j dkrd gdSg}tt|j D]0}|j |}z |j |d}Wntyt|j}Ynwdd|j||D}||qt||j} tr"|jr"d|j} |jdd} | r| d} t| |jdd} | r"| d} t| t| dkr0d | ddSd| dStdt|j)NrPoint)type coordinates MultiPointcSrr>tupler_rr>r>r?rrz+Shape.__geo_interface__.. LineStringrcSrr>rrr>r>r?rrcSrr>rrr>r>r?rrcSrr>rrr>r>r?rrMultiLineStringPolygoncSrr>rrr>r>r?r,rzAPossible issue encountered when converting Shape #{} to GeoJSON: ra^Shapefile format requires that all polygon interior holes be contained by an exterior ring, but the Shape contained interior holes (defined by counter-clockwise orientation in the shapefile format) that were orphaned, i.e. not contained by any exterior rings. The rings were still included but were encoded as GeoJSON exterior rings instead of holes.raShapefile format requires that polygons contain at least one exterior ring, but the Shape was entirely made up of interior holes (defined by counter-clockwise orientation in the shapefile format). The rings were still included but were encoded as GeoJSON exterior rings instead of holes. MultiPolygonz1Shape type "%s" cannot be represented as GeoJSON.)rrrrrkrrrr!rrrrrrhrr rr IndexErrorrrVERBOSEformatrgetloggerwarningrSHAPETYPE_LOOKUP)rXpsrpartrr`startendrrheaderorphansmsg only_holesr>r>r?__geo_interface__s~             zShape.__geo_interface__c CsBt}|r |dnd}|dkrt}n0|dkrt}n)|dkr t}n"|dkr't}n|dkr.t}n|dkr5t}n |dkrr>r? _from_geojsonTs  5 2            zShape._from_geojsoncC|jS)z9The index position of the shape in the original shapefile)rrWr>r>r?rz Shape.oidcC t|jSrFrrrWr>r>r? shapeTypeName zShape.shapeTypeNamecCsd|j|jS)Nz Shape #{}: {})rrrrWr>r>r?rYzShape.__repr__) r[r\r]rrpropertyr staticmethodrrrrYr>r>r>r?rs p Q   rc@s`eZdZdZdddZddZddZd d Zd d Ze d dZ dddZ ddZ ddZ dS)_Recordat A class to hold a record. Subclasses list to ensure compatibility with former work and to reuse all the optimizations of the builtin list. In addition to the list interface, the values of the record can also be retrieved using the field's name. For example if the dbf contains a field ID at position 0, the ID can be retrieved with the position, the field name as a key, or the field name as an attribute. >>> # Create a Record with one field, normally the record is created by the Reader class >>> r = _Record({'ID': 0}, [0]) >>> print(r[0]) >>> print(r['ID']) >>> print(r.ID) NcCs,||_|dur ||_nd|_t||dS)z A Record should be created by the Reader class :param field_positions: A dict mapping field names to field positions :param values: A sequence of values :param oid: The object id, an int (optional) Nr)_Record__field_positions _Record__oidrfr)rXfield_positionsvaluesrr>r>r?rs z_Record.__init__cCsPz |j|}t||WStytd|ty'td|w)a __getattr__ is called if an attribute is used that does not exist in the normal sense. For example r=Record(...), r.ID calls r.__getattr__('ID'), but r.index(5) calls list.index(r, 5) :param item: The field name, used as attribute :return: Value of the field :raises: AttributeError, if item is not a field of the shapefile and IndexError, if the field exists but the field's corresponding value in the Record does not exist {} is not a field namez4{} found as a field but not enough values available.)rrf __getitem__KeyErrorAttributeErrorrrrXitemrr>r>r? __getattr__s   z_Record.__getattr__cCsP|dr t|||Sz |j|}t|||WSty'td|w)z Sets a value of a field attribute :param key: The field name :param value: the value of that field :return: None :raises: AttributeError, if key is not a field of the shapefile _r) startswithrf __setattr__r __setitem__rr rrXrvaluerr>r>r?rs   z_Record.__setattr__c Csjzt||WSty#z|j|}Wn ty d}YnwYnw|dur.t||Std|)z Extends the normal list item access with access using a fieldname For example r['ID'], r[0] :param item: Either the position of the value or the name of a field :return: the value of the field Nz'"{}" is not a field name and not an int)rfr TypeErrorrrrrr r>r>r?rs    z_Record.__getitem__cCsTzt|||WSty)|j|}|dur"t|||YStd|w)z Extends the normal list item access with access using a fieldname For example r['ID']=2, r[0]=2 :param key: Either the position of the value or the name of a field :param value: the new value of the field Nz%{} is not a field name and not an int)rfrrrrrrrr>r>r?r s   z_Record.__setitem__cCr)z:The index position of the record in the original shapefile)rrWr>r>r?rrz _Record.oidFcsXtfddjD}|r*|D]\}}t|tr)d|j|j|j||<q|S)zi Returns this Record as a dictionary using the field names as keys :return: dict c3s |] \}}||fVqdSrFr>)r_fr`rWr>r?rd%z"_Record.as_dict..{:04d}{:02d}{:02d}) rrrr6rryearmonthday)rX date_stringsdctkr;r>rWr?as_dict s z_Record.as_dictcCsd|jt|S)NzRecord #{}: {})rrrfrWr>r>r?rY,sz_Record.__repr__cCs&ttt|}t|j}||S)z Helps to show the field names in an interactive environment like IPython. See: http://ipython.readthedocs.io/en/stable/config/integrating.html :return: List of method names and fields )rfdirrrr)rXdefaultfnamesr>r>r?__dir__/sz_Record.__dir__rFF)r[r\r]r^rr rrrrrrrYr!r>r>r>r?rs    rc@s&eZdZdZdddZeddZdS) ShapeRecordzA ShapeRecord object containing a shape along with its attributes. Provides the GeoJSON __geo_interface__ to return a Feature dictionary.NcCs||_||_dSrFrrecord)rXrr%r>r>r?r=s zShapeRecord.__init__cCs.d|jjdd|jjtkrddS|jjdS)NFeatureT)r)r propertiesgeometry)r%rrrrrrWr>r>r?rAs  zShapeRecord.__geo_interface__NN)r[r\r]r^rrrr>r>r>r?r#:s  r#c@$eZdZdZddZeddZdS)Shapesa$A class to hold a list of Shape objects. Subclasses list to ensure compatibility with former work and to reuse all the optimizations of the builtin list. In addition to the list interface, this also provides the GeoJSON __geo_interface__ to return a GeometryCollection dictionary.cCdt|S)Nz Shapes: {}rrfrWr>r>r?rYMzShapes.__repr__cCddd|Dd}|S)NGeometryCollectioncSg|]}|jqSr>r)r_rr>r>r?rUrz,Shapes.__geo_interface__..)r geometriesr>rX collectionr>r>r?rPs zShapes.__geo_interface__Nr[r\r]r^rYrrr>r>r>r?r+G r+c@r*) ShapeRecordsa)A class to hold a list of ShapeRecord objects. Subclasses list to ensure compatibility with former work and to reuse all the optimizations of the builtin list. In addition to the list interface, this also provides the GeoJSON __geo_interface__ to return a FeatureCollection dictionary.cCr,)NzShapeRecords: {}r-rWr>r>r?rY^r.zShapeRecords.__repr__cCr/)NFeatureCollectioncSr1r>r2)r_shaperecr>r>r?rdrz2ShapeRecords.__geo_interface__..)rfeaturesr>r4r>r>r?ras zShapeRecords.__geo_interface__Nr6r>r>r>r?r8Xr7r8c@seZdZdZdS)ShapefileExceptionz3An exception to handle shapefile specific problems.N)r[r\r]r^r>r>r>r?r<gsr<c@sFeZdZdZddZddZddZdd Zd d Zd d Z e ddZ e ddZ dHddZ ddZddZddZddZddZdd Zd!d"Zd#d$ZdId%d&Zd'd(Zd)d*ZdHd+d,ZdJd.d/ZdHd0d1ZdHd2d3Zd4d5ZdHd6d7ZdHd8d9ZdHd:d;Z dJdd?Z"dHd@dAZ#dKdBdCZ$dIdDdEZ%dIdFdGZ&dS)LReaderacReads the three files of a shapefile as a unit or separately. If one of the three files (.shp, .shx, .dbf) is missing no exception is thrown until you try to call a method that depends on that particular file. The .shx index file is used if available for efficiency but is not required to read the geometry from the .shp file. The "shapefile" argument in the constructor is the name of the file you want to open, and can be the path to a shapefile on a local filesystem, inside a zipfile, or a url. You can instantiate a Reader without specifying a shapefile and then specify one later with the load() method. Only the shapefile headers are read upon loading. Content within each file is only accessed when required and as efficiently as possible. Shapefiles are usually not large but they can be. c Os.d|_d|_d|_g|_d|_g|_d|_d|_d|_g|_ d|_ i|_ | dd|_ | dd|_t|dkrt|d}t|rd|vr<|ddkrVtd ||dr`|}d}n|d|dd }||dd dd}|d rt|d d id}t|}tjdddd}|||dnt|dd}t |dj} |sdd| !D} t| dkrtdt| dkr| d}ntd| t"j#$|d}dD]1} z*| |d| } tjddd} | | | dt%|| | |j&| WqYqWdn 1swYz|'WnY|js0|jr6|(dStd||d rt)|}|d}t"j#$|\}}t"j#*|}dD]K} z>t+|}|d| |d<t,|}t|d d id}t|}tjddd} | || dt%|| | |j&| WqZt-yYqZw|js|jr|(dStd||(|dSd|.vrEt/|dd r|d|_z|jdWn%t0t1j2fyt13|j|_Ynwt"j#$|d\}} |4|d!|.vrEt/|d!d r6|d!|_z|jdWn%t0t1j2fy5t13|j|_Ynwt"j#$|d!\}} |5|d"|.vrt/|d"d rx|d"|_z|jdWn%t0t1j2fywt13|j|_Ynwt"j#$|d"\}} |6||js|jr|(dSdS)#Nz Not specifiedrr<r2r=r3z.ziprz:Reading from multiple nested zipfiles is not supported: %sr$httpz User-agentzwMozilla/5.0 (Macintosh; Intel Mac OS X 10_9_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/35.0.1916.47 Safari/537.36)headerszw+bT)modesuffixdeleterb)r@rcSsg|] }|dr|qS).shp)endswith)r_namer>r>r?rs z#Reader.__init__..z'Zipfile does not contain any shapefileszZipfile contains more than one shapefile: %s. Please specify the full path to the shapefile you would like to open.)shpshxdbf.)r@rBz'No shp or dbf file found in zipfile: %sr#z#No shp or dbf file found at url: %srHreadrIrJ)7rHrIrJ_files_to_close shapeName_offsets shpLength numRecords numShapesfields_Reader__dbfHdrLength_Reader__fieldLookuprr<r=rkrRrHcountr<rFfindrr/r.tempfileNamedTemporaryFilewriterLseekopenzipfileZipFilenamelistrMrQsplitextsetattrrhcloseloadr+basenamerfr,r-rrT NameErrorioUnsupportedOperationBytesIOload_shpload_shxload_dbf)rXargskwargsrQzpath shapefilereqresp zipfileobjarchive shapefilesrmemberfileobjurlinfourlpathr _urlinfo_pathbaseNamer>r>r?rs                           zReader.__init__cCsTdg}|jr|dt|t|j|jr%|dt|t|jd|S)zC Use some general info on the shapefile as __str__ zshapefile Readerz {} shapes (type '{}')z {} records ({} fields) ) rHrhrrkrrrJrSjoin)rXinfor>r>r?__str__Ds zReader.__str__cC|Sz1 Enter phase of context manager. r>rWr>r>r? __enter__QzReader.__enter__cC |dS)zD Exit phase of context manager, close opened files. NrbrXexc_typeexc_valexc_tbr>r>r?__exit__W zReader.__exit__c Cs|jr|jdur ||jS|jrz|jr!|jdur||jS|jdurw|j}|}|dd|}|dt dj }g}|}||krj| ||| d\}}|dd|7}||||ksKt ||_||_|||jSdS)z6Returns the number of shapes/records in the shapefile.Nrr#d>2ir )rJrQ_Reader__dbfHeaderrHrIrR_Reader__shxHeadertellr[rrrhrLrkrO) rXrH checkpointrProffsetsposrecNum recLengthr>r>r?__len__]s:          zReader.__len__ccs|D]}|VqdS)z5Iterates through the shapes/records in the shapefile.N)iterShapeRecords)rXr:r>r>r?__iter__s zReader.__iter__cCs |}|j}t|j|d<|S)Nrw) shapeRecordsrrfrw)rX shaperecords fcollectionr>r>r?rszReader.__geo_interface__cCrrFrrWr>r>r?rrzReader.shapeTypeNameNcCs|r*tj|\}}||_|||||||js*|js*t d||f|jr1| |jr8| |j rA| dSdS)zOpens a shapefile from a filename or file-like object. Normally this method would be called by the constructor with the file name as an argument.z Unable to open %s.dbf or %s.shp.N)rMrQr`rNrirjrkrHrJr<_Reader__shpHeaderrrIr)rXrorNrr>r>r?rcs     z Reader.loadc Cd}ztd||fd|_|j|jWdSty@ztd||fd|_|j|jWYdSty?YYdSww)zX Attempts to load file with .shp extension as both lower and upper case rH%s.%srCN)r\rHrMrhIOErrorupper)rXshapefile_nameshp_extr>r>r?ri  zReader.load_shpc Cr)zX Attempts to load file with .shx extension as both lower and upper case rIrrCN)r\rIrMrhrr)rXrshx_extr>r>r?rjrzReader.load_shxc Cr)zX Attempts to load file with .dbf extension as both lower and upper case rJrrCN)r\rJrMrhrr)rXrdbf_extr>r>r?rkrzReader.load_dbfcC |dSrFrrWr>r>r?__del__rZzReader.__del__c Cs@|jD]}t|drz|WqtyYqwqg|_dS)Nrb)rMrTrbr)rX attributer>r>r?rbs     z Reader.closecCsD|std|jr|jdur||jr t|jdkr ||S)zqChecks to see if the requested shapefile file object is available. If not a ShapefileException is raised.z:Shapefile Reader requires a shapefile or file-like object.Nr)r<rHrPrcrJrkrS)rXrr>r>r? __getFileObjszReader.__getFileObjcCs>|jr|jd}t||krtd|dkrt|j|}|S)zqProvides list-like handling of a record index with a clearer error message if the index is out of bounds.rz#Shape or Record index out of range.r)rQrrrj)rXr`rmaxr>r>r?__restrictIndexs   zReader.__restrictIndexcCs|jstd|j}|dtd|ddd|_|dtd|dd|_td td |d|_td td |d |_ g|_ td td |d D]}|t kr_|j |qR|j d qRd S)z.Reads the header information from a .shp file.zMShapefile Reader requires a shapefile or file-like object. (no shp file found>ir$rr# r>r? __shpHeaders  zReader.__shpHeadercCs||j}t|d}d}}}}} } td|d\} } |d| } td|dd}||_|dkr>g|_n!|d vr_td td |d |_ |dur_t ||j s_| | dS|d vrmtd|dd}|d vr{td|dd}|rtdtd|||d|_ |dkrtdtd|||d|_ |rtdd||d|}ttt|fd|_|dvrtd|d\}}td td|||d|_|dvr1| |dkrtd|d\} } | ||dkr'g|_td td|||dD]}|tkr|j|q|jdqn ddt|D|_|dvrCtd td|dg|_|dkrSttd|d|_|dvr{| |dkrktd|d\}nt}|tkrw|g|_ndg|_| | |S)z8Returns the header info and geometry for a single shape.)rNrr r#rr$r r r r r rrrrrrrrrr r r rrrrr`z<%sir<%sdrr rrrrr rrrrrrcSg|]}dqSrFr>r_r r>r>r?rOz"Reader.__shape..rr rr r>r?__shapesh       "       zReader.__shapecCsF|j}|s td|dtd|dddd}|d|_d S) z.Reads the header information from a .shx file.MShapefile Reader requires a shapefile or file-like object. (no shx file foundrrr$rr#rr N)rIr<r[rrLrR)rXrIshxRecordLengthr>r>r? __shxHeadergs  zReader.__shxHeadercCsf|j}|s td|dtd|d|jd}tjdkr$|dd|d d dD|_ d S) z1Reads the shape offset positions from a .shx filerrr`r#r$bigcSsg|]}d|qSr#r>)r_elr>r>r?r|rz'Reader.__shxOffsets..N) rIr<r[rUrLrRsys byteorderbyteswaprO)rXrI shxRecordsr>r>r? __shxOffsetsqs  zReader.__shxOffsetscCs.|j}|r |dkr dS|js||j|S)zbReturns the offset in a .shp file for a shape based on information in the .shx index file.N)rIrO_Reader__shxOffsets)rXr`rIr>r>r? __shapeIndex~s   zReader.__shapeIndexrc Cs||j}||}||}|s_|dd|}|dtdj}d}|}||krS||kr5n||d\}} |dd| 7}|||d7}||ks0||kr_t d |||||j ||dS) zReturns a shape object for a shape in the geometry record file. If the 'bbox' arg is given (list or tuple of xmin,ymin,xmax,ymax), returns None if the shape is not within that region. rr#rrr rzFShape index {} is out of bounds; the .shp file only contains {} shapesrrw) rrH_Reader__restrictIndex_Reader__shapeIndexr[rrrrLr<r_Reader__shape) rXr`rwrHoffsetrPr_irrr>r>r?rs,         z Reader.shapecCst}||j|d|S)zReturns all shapes in a shapefile. To only read shapes within a given spatial region, specify the 'bbox' arg as a list or tuple of xmin,ymin,xmax,ymax. rw)r+r iterShapes)rXrwshapesr>r>r?rsz Reader.shapesccs||j}|dd|}|d|jr/t|jD]}|j||d}|r,|VqdSd}g}|}||krX|||j||d}|}|rP|V|d7}||ks;|t|ks`J||_||_ dS)zReturns a generator of shapes in a shapefile. Useful for handling large shapefiles. To only read shapes within a given spatial region, specify the 'bbox' arg as a list or tuple of xmin,ymin,xmax,ymax. rr#rrrN) rrHr[rrRrrrhrkrO)rXrwrHrPr`rrrr>r>r?rs4       zReader.iterShapesc Csr|jstd|j}|dtd|d\|_|_|_|jdd}t|D]S}t td|d}d}d}d||vrF|| d}nt ||d}||d |||<t |||j |j||<||||<t |dd |d<|j|q(|d}|d krtd |jdd tddt|jD|_dd|jdd D}||\} } } | |_| |_d S)zzReads a dbf header. Xbase-related code borrows heavily from ActiveState Python Cookbook Recipe 362715 by Raymond HettingerzNShapefile Reader requires a shapefile or file-like object. (no dbf file found)rz r_r`rr>r>r?rdrz%Reader.__dbfHeader..cSsg|]}|dqSrr>)r_rr>r>r?rrz&Reader.__dbfHeader..)rJr<r[rrLrQrT_Reader__recordLengthrjrfrrkrDr<r=lstriprSrhinsertrrrU_Reader__recordFields_Reader__fullRecStruct_Reader__fullRecLookup) rXrJ numFieldsfield fieldDescrGidx terminator fieldnames fieldTuples recLookup recStructr>r>r? __dbfHeaders8     zReader.__dbfHeadercs|jdur |dd|jD}dur"fddt|j|D}d|}t|}||jkr=|d7}|d7}||jks0||fS)zCalculates the format and size of a .dbf record. Optional 'fields' arg specifies which fieldnames to unpack and which to ignore. Note that this always includes the DeletionFlag at index 0, regardless of the 'fields' arg. NcSsg|]}d|dqS)z%dsr#r>)r_ fieldinfor>r>r?rsz&Reader.__recordFmt..cs8g|]\}}|dvs|ddkr|nd|dqS)rrz%dxr#r>)r_rcoderSr>r?rs  r1rr)rQrrSrgr}rr)rXrS structcodesfmtfmtSizer>rr? __recordFmt s      zReader.__recordFmtc Cs|durPtt|}|j|d\}}t|}|D]}||jvs#|dkr*td|qg}|jddD]}|d}||vrC||q4t ddt |D}n |jdd}|j }|j }|||fS) aFReturns the necessary info required to unpack a record's fields, restricted to a subset of fieldnames 'fields' if specified. Returns a list of field info tuples, a name-index lookup dict, and a Struct instance for unpacking these fields. Note that DeletionFlag is not a valid field. Nrrz"{}" is not a valid field namerrcsrrr>rr>r>r?rd=rz(Reader.__recordFields..) rfset_Reader__recordFmtrrU ValueErrorrrSrhrrrr) rXrSrrrrGrrrr>r>r?__recordFields$s&   zReader.__recordFieldsc Cs||j}|||j}|ddkrdS|dd}t|t|kr1tdt|t|g}t||D]\\}} } } } | dvr| dd} | dd } | d krXd} n| rlzt | } Wnt ykd} Ynwzt | } Wnt yzt t | } Wn t yd} YnwYnow| d kr| dd  dd  d d sd} nYz"t | dd t | d d t | d d} }}t| ||} Wn6t| } Yn+| dkr| dkrd} n | dvrd} n| dvrd} nd} nt| |j|j} | d} || q8t|||S)zReads and returns a dbf record row as a list of values. Requires specifying a list of field info tuples 'fieldTuples', a record name-index dict 'recLookup', and a Struct instance 'recStruct' for unpacking these fields. r NrznNumber of record values ({}) is different from the requested number of fields ({})NFr*r5D0r$rLr LsYyTt1TsNnFf0F)rrJrrLsizerkr<rrgsplitreplacefloatr intrrDstripr<r=rstriprhr)rXrrrrrrecordContentsr%rGtyprdeciryrrr>r>r?__recordEsh         4  zReader.__recordcCsn||j}|jdur|||}|j}|d||j||||\}}}|j ||||dS)zReturns a specific dbf record based on the supplied index. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. Nrrrrr) rrJrQrrrr[rTr_Reader__record)rXr`rSrrecSizerrrr>r>r?r%s    z Reader.recordc Csr|jdur |g}||j}||j||\}}}t|jD]}|j||||d}|r6| |q$|S)zReturns all records in a dbf file. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. Nr ) rQrrrJr[rTrrjr!rh) rXrSrecordsrrrrr`rDr>r>r?r#s    zReader.recordsccsl|jdur |||j}||j||\}}}t|jD]}|j||||d}|r3|Vq#dS)zReturns a generator of records in a dbf file. Useful for large shapefiles or dbf files. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. Nr ) rQrrrJr[rTrrr!)rXrSrrrrr`rDr>r>r? iterRecordss   zReader.iterRecordscCs:||}|j||d}|r|j||d}t||dSdS)a^Returns a combination geometry and attribute record for the supplied record index. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. If the 'bbox' arg is given (list or tuple of xmin,ymin,xmax,ymax), returns None if the shape is not within that region. rrr$N)rrr%r#)rXr`rSrwrr%r>r>r? shapeRecords  zReader.shapeRecordcCst|j||dS)a`Returns a list of combination geometry/attribute records for all records in a shapefile. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. To only read entries within a given spatial region, specify the 'bbox' arg as a list or tuple of xmin,ymin,xmax,ymax. )rSrw)r8r)rXrSrwr>r>r?rszReader.shapeRecordsccst|durt||j|dD] \}}t||dVqdS|j|dD]}|r7|j|j|d}t||dVq$dS)aeReturns a generator of combination geometry/attribute records for all records in a shapefile. To only read some of the fields, specify the 'fields' arg as a list of one or more fieldnames. To only read entries within a given spatial region, specify the 'bbox' arg as a list or tuple of xmin,ymin,xmax,ymax. Nrr$r)r`rS)r0rr$r#r%r)rXrSrwrr%r>r>r?rszReader.iterShapeRecordsrFr)r)rNN)'r[r\r]r^rrrrrrrrrrcrirjrkrrbrrrrrrrrrrrrrr!r%r#r$r%rrr>r>r>r?r=sN" 1      O % * *  ! T     r=c@sVeZdZdZdVddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZeddZddZddZd d!ZdWd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6ZdXd7d8ZdYd:d;Zdd?Z!d@dAZ"dBdCZ#dDdEZ$dFdGZ%dHdIZ&dJdKZ'dLdMZ(dNdOZ)dPdQZ*dZdTdUZ+dS)[Writerz+Provides write support for ESRI Shapefiles.NFcKs||_||_g|_||_d|_|_|_g|_|rYt|}t |s.t d t |t ||tj|dd|_|tj|dd|_|tj|dd|_n=|dsh|dsh|dr|d|d|d}}}|r|||_|r|||_|r|||_nt d |jr|jd |jr|jd d|_d|_d|_d|_d|_d|_|d d |_|d d|_dS)NzHThe target filepath {} must be of type str/unicode or path-like, not {}.rrEz.shxz.dbfrHrIrJzYEither the target filepath, or any of shp, shx, or dbf must be set to create a shapefile.sd9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999r<r2r=r3)target autoBalancerSrrHrIrJrMrRrHrrreprr_Writer__getFileObjrMrQr`rrZrshpNum_bbox_zbox_mbox deletionFlagrr<r=)rXr'rr(rmrHrIrJr>r>r?rsB"   zWriter.__init__cCst|j|jS)zReturns the current number of features written to the shapefile. If shapes and records are unbalanced, the length is considered the highest of the two.)rvrr+rWr>r>r?r#szWriter.__len__cCrrr>rWr>r>r?r)rzWriter.__enter__cCr)zT Exit phase of context manager, finish writing and close the files. Nrrr>r>r?r/rzWriter.__exit__cCrrFrrWr>r>r?r5rZzWriter.__del__c Csl|jo t|jdo |jj }|jot|jdo|jj }|jo)t|jdo(|jj }|jrK|rK|jrK|rK|jr;||j|jkrKt d|j|jf|jrX|rX|j |jdd|jre|re|j |jdd|jrn|rn| |j|j|jfD]}t|drt|dr|jsz| Wqvt yYqvwqv|jD]}t|drz|Wqt yYqwqg|_dS) zL Write final shp, shx, and dbf headers, close opened files. closedzpWhen saving both the dbf and shp file, the number of records (%s) must correspond with the number of shapes (%s)rH) headerTyperIflushrbN)rHrTr0rIrJr(balancerr+r<_Writer__shapefileHeader_Writer__dbfHeaderr2rrMrb)rXshp_openshx_opendbf_openrr>r>r?rb8sB            z Writer.closecCs^|stdt|dr |Stj|d}|r"tj|s"t|t|d}|j ||S)z*Safety handler to verify file-like objectszNo file-like object available.rZrzwb+) r<rTrMrQrexistsmakedirsr\rMrh)rXrpthfpr>r>r?rds    zWriter.__getFileObjcCs:|j}|jdd|j}|d}|j||S)z+Calculates the file length of the shp file.rr#)rHrr[)rXrrr>r>r?__shpFileLengthrs   zWriter.__shpFileLengthcCsg}g}t|jdkr#tt|jdd\}}||||ntd|jt|t|t|t|g}|j rdt|d|j dt|d|j dt|d|j dt|d|j dg|_ |S||_ |S)Nrr#zjCannot create bbox. Expected a valid shape with at least one point. Got a shape of type '%s' and 0 points.rr ) rkrrfrgrrrrurvr,)rXsrrpxpyrwr>r>r?__bboxs  Nz Writer.__bboxc Csg}|jD]}z ||dWqty|dYqwt|t|g}|jrAt|d|jdt|d|jdg|_|S||_|S)Nr#rr)rrhrrurvr-)rXr>rrrr>r>r?__zboxs  *z Writer.__zboxc Cs|jdvrdnd}g}|jD]}z||dur|||Wqty(Yqw|s0|tt|t|g}|jrRt|d|jdt|d|jdg|_|S||_|S)N)r r rrrr r#rr)rrrhrrrurvr.)rXr>mposrrrr>r>r?__mboxs$    *z Writer.__mboxcCrrFrrWr>r>r?rrzWriter.shapeTypeNamecCr)zReturns the current bounding box for the shapefile which is the lower-left and upper-right corners. It does not contain the elevation or measure extremes.)r,rWr>r>r?rwsz Writer.bboxcCr)z1Returns the current z extremes for the shapefile.)r-rWr>r>r?rz Writer.zboxcCr)z1Returns the current m extremes for the shapefile.)r.rWr>r>r?rrEz Writer.mboxrHc Cs||}|d|tddddddd|dkr&|td|n|dkr9|tdd|jdd |jd urAt|_|td d |j|jdkruz|}|d ur]gd }|tdg|RWnt ytt dw|tddddd|jdvr| }|d urddg}nddg}|jdvr| }|d urddg}nddg}z|td|d|d|d|dWd St yt dw)zWrites the specified header type to the specified file-like object. Several of the shapefile formats are so similar that a single generic method to read or write them is warranted.rz>6ii 'rHrrIrr r#Nz<2ii)rrrrrz8Failed to write shapefile bounding box. Floats required.)r r rr) r r rrrrrrrrzHFailed to write shapefile elevation and measure values. Floats required.) r*r[rZr_Writer__shpFileLengthr+rrrwrr<rr)rXfileObjr1rrwrrr>r>r?__shapefileHeadersJ       , zWriter.__shapefileHeaderc Cs:||j}|dd}tdd\}}}|d8}dd|jD}|s*td|j}t|}|dd } | d kr?td t d d|Dd } t d|||||| | } | | |D]9} | \} }}}t | |j |j} | dd} | ddddd} t |d}t|}t d| |||}| |q\| ddS)z,Writes the dbf header and field descriptors.rr NilcSsg|] }|ddkr|qS)rrr>r_rr>r>r?rz&Writer.__dbfHeader..z3Shapefile dbf file must contain at least one field.rriz3Shapefile dbf header length exceeds maximum length.cSsg|]}t|dqSr)rrIr>r>r?rsrz r>r?r s<        zWriter.__dbfHeadercCs~|jr |j|jkr |t|ts+t|dr|j}t|tr%t |}nt d|| |\}}|j r=| ||dSdS)NrzbCan only write Shape objects, GeoJSON dictionaries, or objects with the __geo_interface__, not: %r)r(rr+r3r6rrTrrrr_Writer__shpRecordrI_Writer__shxRecord)rXr>rlengthr>r>r?r0s    z Writer.shapecs||j}|jd7_td|jd}|jdur-|jtkr-|j|_|jtkrB|j|jkrBtd|j|jftd|j|jdvrU| ||jdvryztdg| |RWnt yxt d |jw|jd vrtdt |j |jdvrtdt |j|jd vr|j D] }td|q|jd kr|jD] }td|q|jdvrz fd d |jDWnt yt d|jw|jdvr9ztdg||RWnt yt d|jwz$t|drtdt |jg|jRn fdd |jDWnt y8t d|jw|jdvrztdg||RWnt y^t d|jwz4t|dr}tdt |jgdd |jDRn|jdvrdndfdd |jDWnt yt d|jw|jdvrztd|jdd|jddWnt yt d|jw|jdkr4||t|drz|jsd |_td!|jdWn@t yt d"|jwz!t |jddkr|jddtd!|jddWnt y3t d"|jw|jd#vr||t|drrz|jrR|jddurVtf|_td$|jdWn^t yqt d%|jwz?|jdkr{dndt |jddkr|jdtn|jddurt|jd<td$|jdWnt yt d%|jw}||d}|d&td'||||fS)(Nrrrz@The shape's type (%s) must match the type of the shapefile (%s).rrrrzr?rms*z&Writer.__shpRecord..z6Failed to write points for record %s. Expected floats.rrzBFailed to write elevation extremes for record %s. Expected floats.rrcs0g|]}tdt|dkr|dndqS)rr#r)rZrrkrrZr>r?r}s0z@Failed to write elevation values for record %s. Expected floats.rz?Failed to write measure extremes for record %s. Expected floatsrcSsg|] }|dur |ntqSrF)r)r_rr>r>r?rrJr r#cs<g|]}tdt|kr|dur|ntqS)rN)rZrrkrrrrCr>r?rs<z=Failed to write measure values for record %s. Expected floatsz5Failed to write point for record %s. Expected floats.r rrz?Failed to write elevation value for record %s. Expected floats.)r rz<1dz=Failed to write measure value for record %s. Expected floats.r$r)r*rHrr+rZrrrr _Writer__bboxrr<rkrrr _Writer__zboxrTr _Writer__mboxrrhrr[)rXr>rrrptfinishrYr>r[r? __shpRecordCs                 $   . *        zWriter.__shpRecordcCsP||j}z |td|dWn tytdw|td|dS)zWrites the shx records.rr#zThe .shp file has reached its file size limit > 4294967294 bytes (4.29 GB). To fix this, break up your file into multiple smaller ones.N)r*rIrZrrr<)rXrrYrr>r>r? __shxRecords  zWriter.__shxRecordcOs|jr |j|jkr |tdd|jD}|r/t|}t||kr.|dt||ks#n<|rbg}|jD]*}|ddkr?q6|d|vr[||d}|durU|dq6||q6|dq6n ddt |D}| |dS) aCreates a dbf attribute record. You can submit either a sequence of field values or keyword arguments of field names and values. Before adding records you must add fields for the record values using the field() method. If the record values exceed the number of fields the extra ones won't be added. In the case of using keyword arguments to specify field/value pairs only fields matching the already registered fields will be added.css |] }|ddkrdVqdS)rrrNr>rIr>r>r?rdrz Writer.record..r1rrNcSr)r1r>rr>r>r?rrz!Writer.record..) r(rr+r3rirSrfrkrhrj_Writer__dbfRecord)rX recordList recordDict fieldCountr%rvalr>r>r?r%s.            z Writer.recordc Cs||j}|jdkr||d|jd7_dd|jD}t||D]\\}}}}}|}t|}|dvr}|t vrEd|}n|sjzt|}Wnt y\tt |}Ynwt |dd | |}nyt |}t |d |d | |}nf|d krt|trd |j|j|j}nRt|trt|d krd j |}nA|t vrd}n:t|rt|dkrn/td|dkr|t vrd}n |dvrd}n|dvrd}nd}nt||j|jd ||}t|tst|d|j}t||krtdt|||f||q(d S)zWrites the dbf records.rr rcss |] }|ddkr|VqdS)rrNr>rIr>r>r?rd rz%Writer.__dbfRecord..r rrNz.%sfrrr s00000000r zaDate values must be either a datetime.date object, a list, a YYYYMMDD string, or a missing value.r)TrT)FrFrzZShapefile Writer unable to pack incorrect sized value (size %d) into field '%s' (size %d).)r*rJrr5rZrSrgrrMISSINGr rrrjustr6rrrrrfrkrHr<r@r<r=rOr9) rXr%rrS fieldNamerTrrrr>r>r? __dbfRecordsd           zWriter.__dbfRecordcCsH|j|jkr||j|jks|j|jkr"||j|jksdSdS)zAdds corresponding empty attributes or null geometry records depending on which type of record was created to make sure all three files are in synch.N)rr+nullr%rWr>r>r?r3C s  zWriter.balancecCs|ttdS)zCreates a null shape.N)rrrrWr>r>r?rnM sz Writer.nullcCs*t}t|}|j||g||dS)zCreates a POINT shape.N)rrrrhr)rXrrr pointShaper>r>r?pointR sz Writer.pointcCs,t}t|}|j|||g||dS)z[Creates a POINTM shape. If the m (measure) value is not set, it defaults to NoData.N)rrrrhr)rXrrrrror>r>r?pointmY sz Writer.pointmrcCs.t}t|}|j||||g||dS)zCreates a POINTZ shape. If the z (elevation) value is not set, it defaults to 0. If the m (measure) value is not set, it defaults to NoData.N)rrrrhr)rXrrrrrror>r>r?pointza sz Writer.pointzcCt}|g}|j||ddS)zBCreates a MULTIPOINT shape. Points is a list of xy values.rrN)r _shapepartsrXrrr>r>r? multipointk szWriter.multipointcCrs)zCreates a MULTIPOINTM shape. Points is a list of xym values. If the m (measure) value is not included, it defaults to None (NoData).rtN)r!rurvr>r>r? multipointmr szWriter.multipointmcCrs)zCreates a MULTIPOINTZ shape. Points is a list of xyzm values. If the z (elevation) value is not included, it defaults to 0. If the m (measure) value is not included, it defaults to None (NoData).rtN)rrurvr>r>r? multipointzz szWriter.multipointzcCt}|j||ddS)zfCreates a POLYLINE shape. Lines is a collection of lines, each made up of a list of xy values.rtN)rrurXlinesrr>r>r?line sz Writer.linecCrz)zCreates a POLYLINEM shape. Lines is a collection of lines, each made up of a list of xym values. If the m (measure) value is not included, it defaults to None (NoData).rtN)rrur{r>r>r?linem sz Writer.linemcCrz)zCreates a POLYLINEZ shape. Lines is a collection of lines, each made up of a list of xyzm values. If the z (elevation) value is not included, it defaults to 0. If the m (measure) value is not included, it defaults to None (NoData).rtN)rrur{r>r>r?linez z Writer.linezcCrz)aCreates a POLYGON shape. Polys is a collection of polygons, each made up of a list of xy values. Note that for ordinary polygons the coordinates must run in a clockwise direction. If some of the polygons are holes, these must run in a counterclockwise direction.rtN)rrurXrrr>r>r?r rz Writer.polycCrz)apCreates a POLYGONM shape. Polys is a collection of polygons, each made up of a list of xym values. Note that for ordinary polygons the coordinates must run in a clockwise direction. If some of the polygons are holes, these must run in a counterclockwise direction. If the m (measure) value is not included, it defaults to None (NoData).rtN)r rurr>r>r?polym sz Writer.polymcCrz)aCreates a POLYGONZ shape. Polys is a collection of polygons, each made up of a list of xyzm values. Note that for ordinary polygons the coordinates must run in a clockwise direction. If some of the polygons are holes, these must run in a counterclockwise direction. If the z (elevation) value is not included, it defaults to 0. If the m (measure) value is not included, it defaults to None (NoData).rtN)rrurr>r>r?polyz sz Writer.polyzcCspt}t|}g|_g|_|D]}|jt|j|D]}t|ts&t|}|j|qq||_| |dS)aCreates a MULTIPATCH shape. Parts is a collection of 3D surface patches, each made up of a list of xyzm values. PartTypes is a list of types that define each of the surface patches. The types can be any of the following module constants: TRIANGLE_STRIP, TRIANGLE_FAN, OUTER_RING, INNER_RING, FIRST_RING, or RING. If the z (elevation) value is not included, it defaults to 0. If the m (measure) value is not included, it defaults to None (NoData).N) r"rrrrhrkr6rfrr)rXrrr polyShaperrpr>r>r? multipatch s zWriter.multipatchcCst|}g|_g|_|dvr"|D]}|d|dkr!||dq|D]}|jt|j|D]}t|tsr>r?ru s  zWriter._shapepartsr50cCsP|dkr d}d}n|dkrd}d}t|jdkrtd|j||||fdS) z-Adds a dbf field descriptor to the shapefile.r8rr1iz8Shapefile Writer reached maximum number of fields: 2046.N)rkrSr<rh)rXrGrTrrUr>r>r?r sz Writer.field)NNF)rHrFr)rrr),r[r\r]r^rrrrrrbr*rFr\r]r^rrrwrrr4r5rrWrXr%rcr3rnrprqrrrwrxryr}r~rrrrrrurr>r>r>r?r&sV ',   :# #E        r&c sddld_|dd}|dkrtdddlGfdddj}j||d}tdd }j | d  d d id ddd}Wdn1sRwY| |\}}|rg|d|S|dkrqtd|S|dkrz|||S)NrrverbosezRunning doctests...cs&eZdZfddZfddZdS)ztest..Py23DocCheckercs@tjddkrdd|}dd|}j||||}|S)Nrr#zu'(.*?)'z'\1'zu"(.*?)"z"\1")r version_infosub OutputChecker check_output)rXwantgot optionflagsresdoctestrer>r?rO s z)test..Py23DocChecker.check_outputcsjddS)NT)r summarizerW)rr>r?rU rz&test..Py23DocChecker.summarizeN)r[r\r]rrr>rr>r?Py23DocCheckerN sr)checkerrz README.mdrCutf8z r|README)stringglobsrGfilenamelinenoTzAll test passed successfully)rNORMALIZE_WHITESPACErprintrr DocTestRunnerr\ DocTestParser get_doctestrLrCrrunr)rm verbosityrrunnerfobjtest failure_count test_countr>rr?rE s*  ,  r__main__)r2r3r"rF)[r^ __version__structrrrrrrMrrMarrayrXloggingrfdatetimerr] getLoggerr[rrrrrrrrrrrrrr r!r"rr%r&r'r(r)r*PARTTYPE_LOOKUPrPYTHON3rjrrgr0 urllib.parser+r, urllib.errorr-urllib.requestr.r/ itertoolsurllib2rjrr@rDrHrRrUrorqrtrxrrrrrrobjectrrfrr#r+r8rr<r=r&rrexitr>r>r>r?s              *, yl  (qQ $