tunned-geobr 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

tunned_geobr/read_quilombola_areas.py

@@ -4,82 +4,100 @@ import os
 import requests
 from zipfile import ZipFile
 from io import BytesIO
+import urllib3
+import time
+from pathlib import Path
 
-def read_settlements(simplified=False):
-    """Download official settlements data from INCRA.
+def read_quilombola_areas(simplified=False, local_file=None):
+    """Download Quilombola Areas data from INCRA.
     
-    This function downloads and processes data about settlements (assentamentos) 
-    from INCRA (Instituto Nacional de Colonização e Reforma Agrária).
-    Original source: INCRA - Certificação de Imóveis Rurais
+    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
     
     Parameters
     ----------
     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
     -------
     gpd.GeoDataFrame
-        Geodataframe with settlements data
+        Geodataframe with Quilombola Areas data
+        Columns:
+        - geometry: Geometry of the area
+        - nome: Area name
+        - municipio: Municipality
+        - uf: State
+        - area_ha: Area in hectares
+        - fase: Current phase in the titling process
+        - familias: Number of families
+        - portaria: Ordinance number
+        - decreto: Decree number
+        - titulo: Title number
+        - data_titulo: Title date
         
     Example
     -------
-    >>> from geobr import read_settlements
+    >>> from tunned_geobr import read_quilombola_areas
     
-    # Read settlements data
-    >>> settlements = read_settlements()
+    # 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/Assentamento%20Brasil.zip"
+    url = "https://certificacao.incra.gov.br/csv_shp/zip/Áreas%20de%20Quilombolas.zip"
     
-    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
-        import urllib3
-        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")
+    # 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]}")
                 
-            # Read the shapefile
-            gdf = gpd.read_file(os.path.join(temp_dir, shp_files[0]))
-            
-            if simplified:
-                # Keep only the most relevant columns
-                columns_to_keep = [
-                    'geometry',
-                    'NOME_PROJE',  # Nome do Projeto de Assentamento
-                    'MUNICIPIO',   # Município
-                    'UF',          # Estado
-                    'AREA_HA',     # Área em hectares
-                    'NUM_FAMILI',  # Número de famílias
-                    'CAPACIDADE',  # Capacidade de famílias
-                    'DT_CRIACAO',  # Data de criação
-                    'SITUACAO'     # Situação do assentamento
-                ]
+                # Read the shapefile
+                gdf = gpd.read_file(os.path.join(temp_dir, shp_files[0]))
+                gdf = gdf.to_crs(4674)  # Convert to SIRGAS 2000
                 
-                # 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]
+                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)}")
     
-    except Exception as e:
-        raise Exception(f"Error downloading settlements data: {str(e)}")
-        
-    return gdf
+    # 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')"
 
 if __name__ == '__main__':
-    settlements = read_settlements()
-    print(settlements)
+    quilombos = read_quilombola_areas()
+    print(quilombos)

{tunned_geobr-0.2.4.dist-info → tunned_geobr-0.2.5.dist-info}/METADATA

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

{tunned_geobr-0.2.4.dist-info → tunned_geobr-0.2.5.dist-info}/RECORD

@@ -1,7 +1,7 @@
-tunned_geobr-0.2.4.dist-info/METADATA,sha256=sSyUKlIydReoJnsxk_ZbvhK64SIKrvLHDhPyftmrSVY,5018
-tunned_geobr-0.2.4.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-tunned_geobr-0.2.4.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-tunned_geobr-0.2.4.dist-info/licenses/LICENSE.txt,sha256=mECZRcbde3HssOKe1Co4zgqBLGVN0OWpTsEy3LIbcRA,75
+tunned_geobr-0.2.5.dist-info/METADATA,sha256=qV-puwV243sopHV1F8Ec0h5FmeZOhXv2zmmkrys1ZXQ,5018
+tunned_geobr-0.2.5.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+tunned_geobr-0.2.5.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+tunned_geobr-0.2.5.dist-info/licenses/LICENSE.txt,sha256=mECZRcbde3HssOKe1Co4zgqBLGVN0OWpTsEy3LIbcRA,75
 tunned_geobr/__init__.py,sha256=6kbBMfBy0NidzGz_heNFEFPLLVVaNphTqGOFV-6qzqI,7354
 tunned_geobr/data/grid_state_correspondence_table.csv,sha256=FpkBuX_-lRXQ1yBrQODxQgG9oha9Fd8A8zGKfdsDAmk,2660
 tunned_geobr/list_geobr.py,sha256=l-sXzMr94uHZfeYDKgcdlnwaVbB50rrw02I0ehRIWAg,17305
@@ -113,8 +113,7 @@ tunned_geobr/read_ports.py,sha256=dOFOhQ2kim-_VJ_bC1ZiABqD9-FCOelkrTAaLD_yAmY,28
 tunned_geobr/read_private_aerodromes.py,sha256=Il9sfvBxDM-Xv6fkvOXYfaFLfjOaHlIw-tTGhUJ_TpM,2918
 tunned_geobr/read_processing_facilities.py,sha256=BABgyK2FBlHyVSBBzuYN5kRyq9H6LzinPbVOcVsgvgg,5294
 tunned_geobr/read_public_aerodromes.py,sha256=nq3b9HF5_e-yeNcSfQ5ktdAGHKbSfDD_imj-tOhjKJA,2909
-tunned_geobr/read_quilombola_areas.py,sha256=C47Wj4DhSDa-pSFfYK4uGDwtu4sUwqPMr-CuuxS95xg,3060
-tunned_geobr/read_quilombola_areas_temp.py,sha256=iY-r4YDRjaGyO-iPRBm1kWDkN_-axjYxMAQyAjIfG68,4288
+tunned_geobr/read_quilombola_areas.py,sha256=iY-r4YDRjaGyO-iPRBm1kWDkN_-axjYxMAQyAjIfG68,4288
 tunned_geobr/read_railways.py,sha256=J6eM0yr049CaOL95PMd4sGc7JJHiEinJhqf0ThCOClg,2763
 tunned_geobr/read_region.py,sha256=qHbmj3uS-W2Vk6Z1d4vVUA9d03gqGqoujIWPqWk-L8Y,955
 tunned_geobr/read_rppn.py,sha256=nXDzclIiqhutkYWvxlIH_mYSNGdfRVSUzSzi-15X-3w,3963
@@ -137,4 +136,4 @@ tunned_geobr/read_water_bodies_ana.py,sha256=e8wQukpQABjyFCdqSWcFXXMdD-jmguELVJa
 tunned_geobr/read_waterways.py,sha256=mEdoVogYWr5EYZ8bE3xMCVWyLrHYU7xTL2lUE0XbDAM,2951
 tunned_geobr/read_weighting_area.py,sha256=fsV9pXWOw1X7XLS9SAUHVhKy6sw97EEXF5kWEEpFaZ8,2324
 tunned_geobr/utils.py,sha256=WT9PSGWvcERjj3yhfTvyWSE5ZiEjO4tYK5xIj5jJCg8,8170
-tunned_geobr-0.2.4.dist-info/RECORD,,
+tunned_geobr-0.2.5.dist-info/RECORD,,

tunned_geobr/read_quilombola_areas_temp.py

@@ -1,103 +0,0 @@
-import geopandas as gpd
-import tempfile
-import os
-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):
-    """Download Quilombola Areas data from INCRA.
-    
-    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
-    
-    Parameters
-    ----------
-    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
-    -------
-    gpd.GeoDataFrame
-        Geodataframe with Quilombola Areas data
-        Columns:
-        - geometry: Geometry of the area
-        - nome: Area name
-        - municipio: Municipality
-        - uf: State
-        - area_ha: Area in hectares
-        - fase: Current phase in the titling process
-        - familias: Number of families
-        - portaria: Ordinance number
-        - decreto: Decree number
-        - titulo: Title number
-        - data_titulo: Title date
-        
-    Example
-    -------
-    >>> from tunned_geobr import read_quilombola_areas
-    
-    # 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]}")
-                
-                # Read the shapefile
-                gdf = gpd.read_file(os.path.join(temp_dir, shp_files[0]))
-                gdf = gdf.to_crs(4674)  # Convert to SIRGAS 2000
-                
-                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)}")
-    
-    # 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')"
-
-if __name__ == '__main__':
-    quilombos = read_quilombola_areas()
-    print(quilombos)