tunned-geobr 1.0.5__tar.gz → 1.0.6__tar.gz

{tunned_geobr-1.0.5 → tunned_geobr-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tunned-geobr
-Version: 1.0.5
+Version: 1.0.6
 Summary: Fork personalizado do geobr com funcionalidades extras como download de dados da ANM
 Author: Anderson Stolfi
 License: MIT

{tunned_geobr-1.0.5 → tunned_geobr-1.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tunned-geobr"
-version = "1.0.5"
+version = "1.0.6"
 description = "Fork personalizado do geobr com funcionalidades extras como download de dados da ANM"
 readme = "README.md"
 requires-python = "<4.0,>=3.9"

tunned_geobr-1.0.6/tunned_geobr/read_exploration_blocks.py

@@ -0,0 +1,86 @@
+import geopandas as gpd
+import tempfile
+import os
+import requests
+from zipfile import ZipFile
+from io import BytesIO
+
+def read_exploration_blocks(simplified=False):
+    """Download Brazilian Exploration Blocks data from ANP.
+    
+    This function downloads and processes the Brazilian Exploration Blocks data
+    from ANP (National Petroleum Agency).
+    Original source: ANP - Agência Nacional do Petróleo, Gás Natural e Biocombustíveis
+    
+    Parameters
+    ----------
+    simplified : boolean, by default False
+        If True, returns a simplified version of the dataset with fewer columns
+        
+    Returns
+    -------
+    gpd.GeoDataFrame
+        Geodataframe with Brazilian exploration blocks data
+        
+    Example
+    -------
+    >>> from tunned_geobr import read_exploration_blocks
+    
+    # Read exploration blocks data
+    >>> blocks = read_exploration_blocks()
+    """
+    
+    url = "https://gishub.anp.gov.br/geoserver/BD_ANP/ows?service=WFS&version=1.0.0&request=GetFeature&typeName=BD_ANP%3ABLOCOS_EXPLORATORIOS_SIRGAS&maxFeatures=40000&outputFormat=SHAPE-ZIP"
+    
+    try:
+        # Download the zip file
+        response = requests.get(url)
+        if response.status_code != 200:
+            raise Exception("Failed to download exploration blocks data from ANP")
+            
+        # Create a temporary directory
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Extract the zip file
+            with ZipFile(BytesIO(response.content)) as zip_ref:
+                zip_ref.extractall(temp_dir)
+            
+            # Find the shapefile
+            shp_files = []
+            for root, dirs, files in os.walk(temp_dir):
+                shp_files.extend([os.path.join(root, f) for f in files if f.endswith('.shp')])
+            
+            if not shp_files:
+                raise Exception("No shapefile found in the downloaded data")
+                
+            # Read the shapefile
+            gdf = gpd.read_file(shp_files[0])
+            
+            # Convert to SIRGAS 2000 (EPSG:4674) if not already
+            if gdf.crs is None:
+                gdf.crs = 4674
+            elif gdf.crs.to_epsg() != 4674:
+                gdf = gdf.to_crs(4674)
+            
+            if simplified:
+                # Keep only the most relevant columns
+                # Note: Column names based on typical ANP data structure
+                columns_to_keep = [
+                    'geometry',
+                    'NOME',        # Block name
+                    'OPERADOR',    # Operator
+                    'BACIA',       # Basin
+                    'CONTRATO',    # Contract
+                    'AREA_KM2'     # Area in square kilometers
+                ]
+                
+                # Filter columns that actually exist in the dataset
+                existing_columns = ['geometry'] + [col for col in columns_to_keep[1:] if col in gdf.columns]
+                gdf = gdf[existing_columns]
+    
+    except Exception as e:
+        raise Exception(f"Error downloading exploration blocks data: {str(e)}")
+
+    return gdf
+
+if __name__ == '__main__':
+    read_exploration_blocks()