tunned-geobr 1.0.17__py3-none-any.whl → 1.0.18__py3-none-any.whl

tunned_geobr/read_quilombola_areas.py

@@ -5,13 +5,11 @@ import requests
 from zipfile import ZipFile
 from io import BytesIO
 import urllib3
-import time
-from pathlib import Path
 
-def read_quilombola_areas(simplified=False, local_file=None):
+def read_quilombola_areas(simplified=False):
     """Download Quilombola Areas data from INCRA.
     
-    This function downloads and processes data about Quilombola Areas (Áreas Quilombolas) 
+    This function downloads and processes data about Quilombola Areas (Áreas Quilombolas)
     in Brazil. These are territories recognized and titled to remaining quilombo communities.
     Original source: INCRA - Instituto Nacional de Colonização e Reforma Agrária
     
@@ -19,9 +17,6 @@ def read_quilombola_areas(simplified=False, local_file=None):
     ----------
     simplified : boolean, by default False
         If True, returns a simplified version of the dataset with fewer columns
-    local_file : string, optional
-        Path to a local zip file containing the data, by default None
-        If provided, the function will use this file instead of downloading from INCRA
         
     Returns
     -------
@@ -46,57 +41,54 @@ def read_quilombola_areas(simplified=False, local_file=None):
     
     # Read Quilombola Areas data
     >>> quilombos = read_quilombola_areas()
-    
-    # Or use a local file that was previously downloaded
-    >>> quilombos = read_quilombola_areas(local_file="path/to/Áreas de Quilombolas.zip")
     """
     
     url = "https://certificacao.incra.gov.br/csv_shp/zip/Áreas%20de%20Quilombolas.zip"
     
-    # If a local file is provided, use it instead of downloading
-    if local_file and os.path.exists(local_file):
-        print(f"Using local file: {local_file}")
-        try:
-            with tempfile.TemporaryDirectory() as temp_dir:
-                # Extract the zip file
-                with ZipFile(local_file) as zip_ref:
-                    zip_ref.extractall(temp_dir)
-                
-                # Find the shapefile
-                shp_files = [f for f in os.listdir(temp_dir) if f.endswith('.shp')]
-                if not shp_files:
-                    raise Exception("No shapefile found in the local file")
-                
-                print(f"Found shapefile: {shp_files[0]}")
+    try:
+        # Download the zip file
+        # Disable SSL verification due to INCRA's certificate issues
+        response = requests.get(url, verify=False)
+        if response.status_code != 200:
+            raise Exception("Failed to download data from INCRA")
+            
+        # Suppress the InsecureRequestWarning
+        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+            
+        # 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 = [f for f in os.listdir(temp_dir) if f.endswith('.shp')]
+            if not shp_files:
+                raise Exception("No shapefile found in the downloaded data")
                 
-                # Read the shapefile
-                gdf = gpd.read_file(os.path.join(temp_dir, shp_files[0]))
-                gdf = gdf.to_crs(4674)  # Convert to SIRGAS 2000
+            # Read the shapefile
+            gdf = gpd.read_file(os.path.join(temp_dir, shp_files[0]))
+            gdf = gdf.to_crs(4674)  # Convert to SIRGAS 2000
+            
+            if simplified:
+                # Keep only the most relevant columns
+                columns_to_keep = [
+                    'geometry',
+                    'nome',       # Area name
+                    'municipio',  # Municipality
+                    'uf',        # State
+                    'area_ha',   # Area in hectares
+                    'fase'       # Current phase
+                ]
                 
-                print(f"Successfully loaded {len(gdf)} Quilombola Areas from local file")
-                
-                if simplified:
-                    # Keep only the most relevant columns
-                    columns_to_keep = [
-                        'geometry',
-                        'nome',       # Area name
-                        'municipio',  # Municipality
-                        'uf',        # State
-                        'area_ha',   # Area in hectares
-                        'fase'       # Current phase
-                    ]
-                    
-                    # 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]
-                
-                return gdf
-        except Exception as e:
-            raise Exception(f"Error processing local file: {str(e)}")
+                # 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]
     
-    # If no local file is provided, return a message with download instructions
-    # This is consistent with the approach in read_snci_properties as mentioned in the MEMORY
-    return "O download automático dos dados de Áreas Quilombolas está temporariamente indisponível.\nPor favor, faça o download manual através do link:\n" + url + "\n\nApós o download, você pode usar o parâmetro local_file:\nquilombos = read_quilombola_areas(local_file='caminho/para/Áreas de Quilombolas.zip')"
+    except Exception as e:
+        raise Exception(f"Error downloading Quilombola Areas data: {str(e)}")
+        
+    return gdf
 
 if __name__ == '__main__':
     quilombos = read_quilombola_areas()

{tunned_geobr-1.0.17.dist-info → tunned_geobr-1.0.18.dist-info}/METADATA

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

{tunned_geobr-1.0.17.dist-info → tunned_geobr-1.0.18.dist-info}/RECORD

@@ -1,7 +1,7 @@
-tunned_geobr-1.0.17.dist-info/METADATA,sha256=L_vwL2Yy63eLa0GHgHmVcRwJ9oeSAeb_euYaNWaPrEI,5019
-tunned_geobr-1.0.17.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-tunned_geobr-1.0.17.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-tunned_geobr-1.0.17.dist-info/licenses/LICENSE.txt,sha256=mECZRcbde3HssOKe1Co4zgqBLGVN0OWpTsEy3LIbcRA,75
+tunned_geobr-1.0.18.dist-info/METADATA,sha256=0REjnixXDIlXgmgKPHUUd97T6sEfy3TfM7tV1qH1VIs,5019
+tunned_geobr-1.0.18.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+tunned_geobr-1.0.18.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+tunned_geobr-1.0.18.dist-info/licenses/LICENSE.txt,sha256=mECZRcbde3HssOKe1Co4zgqBLGVN0OWpTsEy3LIbcRA,75
 tunned_geobr/__init__.py,sha256=8STl_F6A8JWqePcb22E5eW-_EZ5Ln_tOnkCsjlrVOFc,8045
 tunned_geobr/data/grid_state_correspondence_table.csv,sha256=FpkBuX_-lRXQ1yBrQODxQgG9oha9Fd8A8zGKfdsDAmk,2660
 tunned_geobr/list_geobr.py,sha256=JXQeF8bKmXlNVOHUyNKaYgOtLZQ9e2mb2b2PSLfzIhU,18426
@@ -118,7 +118,7 @@ tunned_geobr/read_private_aerodromes.py,sha256=Il9sfvBxDM-Xv6fkvOXYfaFLfjOaHlIw-
 tunned_geobr/read_processing_facilities.py,sha256=8iCveDTk7MXm1bmb1pcknzen62HTGYQ3KEzvUGSdWfk,5349
 tunned_geobr/read_production_fields.py,sha256=yw1g6u4kLGgiFwEbIcoj2IxvGGcT6u8sASqM2RRNk1Q,3182
 tunned_geobr/read_public_aerodromes.py,sha256=nq3b9HF5_e-yeNcSfQ5ktdAGHKbSfDD_imj-tOhjKJA,2909
-tunned_geobr/read_quilombola_areas.py,sha256=iY-r4YDRjaGyO-iPRBm1kWDkN_-axjYxMAQyAjIfG68,4288
+tunned_geobr/read_quilombola_areas.py,sha256=hJFcj23Msh7pCaxQgAkgOefbgi228bnezrW7jZgykyY,3385
 tunned_geobr/read_railways.py,sha256=J6eM0yr049CaOL95PMd4sGc7JJHiEinJhqf0ThCOClg,2763
 tunned_geobr/read_region.py,sha256=X7IwsAVxwUl0apsExSuBr9kIK_7IUehPenLXAF-JFDA,2331
 tunned_geobr/read_rppn.py,sha256=nXDzclIiqhutkYWvxlIH_mYSNGdfRVSUzSzi-15X-3w,3963
@@ -146,4 +146,4 @@ tunned_geobr/read_water_bodies_ana.py,sha256=Z-dpTPVgRHVndTeSFxx8uXn7ufMg2jm0Dlz
 tunned_geobr/read_waterways.py,sha256=mEdoVogYWr5EYZ8bE3xMCVWyLrHYU7xTL2lUE0XbDAM,2951
 tunned_geobr/read_weighting_area.py,sha256=m2X5Ua3jRqLlkqCQbIzR2jmo58pzqkyR3UYcGtgy20E,2325
 tunned_geobr/utils.py,sha256=WT9PSGWvcERjj3yhfTvyWSE5ZiEjO4tYK5xIj5jJCg8,8170
-tunned_geobr-1.0.17.dist-info/RECORD,,
+tunned_geobr-1.0.18.dist-info/RECORD,,