ccfx 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

ccfx/ccfx.py

@@ -23,7 +23,9 @@ import pickle
 import time
 from shapely.geometry import box, Point
 import geopandas, pandas
-from osgeo import gdal
+from osgeo import gdal, ogr, osr
+import py7zr
+import subprocess
 
 
 
@@ -132,9 +134,6 @@ def watchFileCount(path:str="./", extension:str = ".*", interval:float = 0.2, du
     return None
 
 
-
-
-
 def pythonVariable(filename, option, variable=None):
     '''
     option: save, load or open
@@ -233,6 +232,57 @@ def createPath(pathName, v = False):
     return pathName
 
 
+def renameNetCDFvariable(input_file: str, output_file: str, old_var_name: str, new_var_name: str, v = False) -> None:
+    """
+    Renames a variable in a NetCDF file using CDO if it exists.
+    If the variable does not exist, the file is copied without modification.
+    
+    :param input_file: Path to the input NetCDF file
+    :param output_file: Path to the output NetCDF file
+    :param old_var_name: Name of the variable to rename
+    :param new_var_name: New name for the variable
+    """
+    try:
+        # Check if the variable exists in the input file using `cdo showname`
+        result = subprocess.run(
+            ["cdo", "showname", input_file],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        
+        # Check if the old variable name is in the output
+        if old_var_name in result.stdout:
+            # Rename the variable using `cdo chname`
+            subprocess.run(
+                ["cdo", f"chname,{old_var_name},{new_var_name}", input_file, output_file],
+                check=True
+            )
+            if v: print(f"Variable '{old_var_name}' renamed to '{new_var_name}' in '{output_file}'.")
+        else:
+            # Copy the file without renaming
+            shutil.move(input_file, output_file)
+            if v: print(f"Variable '{old_var_name}' not found; '{input_file}' moved to '{output_file}' without modification.")
+    
+    except subprocess.CalledProcessError as e:
+        print(f"Error: {e.stderr}")
+
+def compressTo7z(input_dir: str, output_file: str):
+    """
+    Compresses the contents of a directory to a .7z archive with maximum compression.
+    
+    :param input_dir: Path to the directory to compress
+    :param output_file: Output .7z file path
+    """
+    # Create the .7z archive with LZMA2 compression
+    with py7zr.SevenZipFile(output_file, 'w', filters=[{'id': py7zr.FILTER_LZMA2, 'preset': 9}]) as archive:
+        # Add each item in the input directory, avoiding the top-level folder in the archive
+        for root, _, files in os.walk(input_dir):
+            for file in files:
+                file_path = os.path.join(root, file)
+                # Add file to the archive with a relative path to avoid including the 'tmp' folder itself
+                archive.write(file_path, arcname=os.path.relpath(file_path, start=input_dir))
+
 
 def moveDirectory(srcDir:str, destDir:str, v:bool = False) -> bool:
     '''
@@ -315,6 +365,99 @@ def clipRasterByExtent(inFile: str, outFile: str, bounds: tuple) -> str:
     ds = None
     return outFile
 
+def clipVectorByExtent(inFile: str, outFile: str, bounds: tuple) -> str:
+    '''
+    Clips a vector using GeoPandas
+    inFile: input vector path
+    outFile: output path
+    bounds: tuple (minx, miny, maxx, maxy)
+    return: output path
+    '''
+    # Load the vector file as a GeoDataFrame
+    gdf = geopandas.read_file(inFile)
+    bbox = box(bounds[0], bounds[1], bounds[2], bounds[3])
+    clipped = gdf.clip(bbox)
+    clipped.to_file(outFile)
+    
+    return outFile
+
+def reprojectRaster(inFile: str, outFile: str, dstProjection: str, resamplingMethod: str = 'mode') -> str:
+    '''
+    Reprojects a raster to a new projection
+    inFile: input raster path
+    outFile: output raster path
+    dstProjection: target projection in "AUTH:CODE" format (e.g., "EPSG:3395")
+    resamplingMethod: resampling method to use (default is 'mode')
+    return: output path
+    '''
+    # Open the input raster
+    ds = gdal.Open(inFile)
+    
+    # Define resampling method
+    resampling_methods = {
+        'nearest': gdal.GRA_NearestNeighbour,
+        'bilinear': gdal.GRA_Bilinear,
+        'cubic': gdal.GRA_Cubic,
+        'cubicspline': gdal.GRA_CubicSpline,
+        'lanczos': gdal.GRA_Lanczos,
+        'average': gdal.GRA_Average,
+        'mode': gdal.GRA_Mode,
+        'max': gdal.GRA_Max,
+        'min': gdal.GRA_Min,
+        'med': gdal.GRA_Med,
+        'q1': gdal.GRA_Q1,
+        'q3': gdal.GRA_Q3
+    }
+    
+    resampling = resampling_methods.get(resamplingMethod, gdal.GRA_Mode)
+    gdal.Warp(outFile, ds, dstSRS=dstProjection, resampleAlg=resampling)
+    ds = None
+    
+    return outFile
+
+def rasterizeRaster(inFile: str, outFile: str, targetField: str, targetResolution: float) -> str:
+    '''
+    Rasterizes a vector layer to a raster file
+    inFile: input vector file path
+    outFile: output raster file path
+    targetField: the field in the vector layer to use as the raster value
+    targetResolution: resolution of the output raster (in units of the vector CRS)
+    return: output raster path
+    '''
+    # Open the vector file
+    vector_ds = ogr.Open(inFile)
+    layer = vector_ds.GetLayer()
+    
+    # Get the extent of the vector layer
+    x_min, x_max, y_min, y_max = layer.GetExtent()
+    
+    # Calculate the raster size based on target resolution
+    x_res = int((x_max - x_min) / targetResolution)
+    y_res = int((y_max - y_min) / targetResolution)
+    
+    # Create the raster dataset
+    target_ds = gdal.GetDriverByName('GTiff').Create(outFile, x_res, y_res, 1, gdal.GDT_Int16)
+    target_ds.SetGeoTransform((x_min, targetResolution, 0, y_max, 0, -targetResolution))
+    
+    # Set the projection from the vector layer
+    srs = layer.GetSpatialRef()
+    target_ds.SetProjection(srs.ExportToWkt())
+    
+    # Set the no-data value to -999
+    band = target_ds.GetRasterBand(1)
+    band.SetNoDataValue(-999)
+
+    # Rasterize the vector layer
+    gdal.RasterizeLayer(target_ds, [1], layer, options=["ATTRIBUTE=" + targetField])
+    
+    # Close the datasets
+    band = None
+    target_ds = None
+    vector_ds = None
+    
+    return outFile
+
+
 def getVectorBounds(grid_gdf: geopandas.GeoDataFrame) -> tuple:
     '''
     This function gets the bounds of a GeoDataFrame
@@ -339,7 +482,6 @@ def getVectorBounds(grid_gdf: geopandas.GeoDataFrame) -> tuple:
 
     return minx, miny, maxx, maxy
 
-
 def ignoreWarnings(ignore:bool = True, v:bool = False) -> None:
     '''
     Ignore warnings
@@ -667,6 +809,27 @@ def showProgress(count: int, end: int, message: str, barLength: int = 100) -> No
     if count == end: print()
 
 
+def listAllFiles(folder, extension="*"):
+    list_of_files = []
+    # Getting the current work directory (cwd)
+    thisdir = folder
+
+    # r=root, d=directories, f = files
+    for r, d, f in os.walk(thisdir):
+        for file in f:
+            if extension == "*":
+                list_of_files.append(os.path.join(r, file))
+            elif "." in extension:
+                if file.endswith(extension[1:]):
+                    list_of_files.append(os.path.join(r, file))
+                    # print(os.path.join(r, file))
+            else:
+                if file.endswith(extension):
+                    list_of_files.append(os.path.join(r, file))
+                    # print(os.path.join(r, file))
+
+    return list_of_files
+
 
 def createPointGeometry(coords: list, proj: str = "EPSG:4326") -> geopandas.GeoDataFrame:
     '''

ccfx/sqliteConnection.py

@@ -33,19 +33,29 @@ class sqliteConnection:
 
     def updateValue(self, table_name, col_name, new_value, col_where1, val_1, v=False):
         """
-        does not work yet!
+        Updates a single value in a specified table where the condition matches.
+        
+        Args:
+            table_name (str): Name of the table to update
+            col_name (str): Name of the column to update
+            new_value: Value to set (can be None)
+            col_where1 (str): Column name for WHERE clause
+            val_1: Value to match in WHERE clause
+            v (bool): Verbose flag for logging
         """
-        if not new_value is None:
-            new_value = str(new_value)
-            self.cursor.execute("UPDATE " + table_name + " SET " + col_name +
-                                " = '" + new_value + "' WHERE " + col_where1 + " = " + val_1 + ";")
-        if new_value is None:
-            self.cursor.execute("UPDATE " + table_name + " SET " + col_name +
-                                " = ? " + " WHERE " + col_where1 + " = ?", (new_value, val_1))
-        # self.cursor.execute(sql_str)
-        if v:
-            self.report("\t -> updated {1} value in {0}".format(
-                self.db_name.split("/")[-1].split("\\")[-1], table_name))
+        try:
+            # Use parameterized queries for ALL cases to prevent SQL injection
+            query = f"UPDATE {table_name} SET {col_name} = ? WHERE {col_where1} = ?"
+            self.cursor.execute(query, (new_value, val_1))
+            
+            if v:
+                self.report(f"\t -> updated value in {self.db_name.split('/')[-1].split('\\')[-1]} table: {table_name}")
+            
+            # Commit the transaction if needed (add self.conn.commit() if not in a transaction)
+            
+        except Exception as e:
+            raise Exception(f"Error updating value: {str(e)}")
+        
 
     def createTable(self, table_name, initial_field_name, data_type):
         '''

{ccfx-0.1.0.dist-info → ccfx-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ccfx
-Version: 0.1.0
+Version: 0.2.0
 Summary: Your package description here
 Author-email: Celray James CHAWANDA <celray@chawanda.com>
 License: MIT
@@ -20,7 +20,8 @@ Requires-Dist: pandas
 Requires-Dist: xlsxwriter
 Requires-Dist: pyodbc
 Requires-Dist: sqlalchemy
-Requires-Dist: docx
+Requires-Dist: python-docx
+Requires-Dist: py7zr
 
 # ccfx
 

ccfx-0.2.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+ccfx/__init__.py,sha256=VmBeF3oj6JTJ_793d4i8PvhyF8_FxaxA1L_FmHWqitc,142
+ccfx/ccfx.py,sha256=lB6hTCywvbMpHH6lMgw0UJuoiU-7vA22yvttvKFG5Gc,28450
+ccfx/excel.py,sha256=cQ4TQW49XqbMB3sSS0IOhO3-WArIolEBIrvOvhFyPtI,4757
+ccfx/mssqlConnection.py,sha256=TwyZXhHHI7zy6BSfH1pszuHVJ5cmndRC5dVxvEtSTks,7904
+ccfx/sqliteConnection.py,sha256=BsS3jzHSevXLDmtPIFPVuzKqB981su0VpYadTk4xFEQ,11231
+ccfx/word.py,sha256=AGa64jX5Zl5qotZh5L0QmrsjTnktIBhmj_ByRKZ88vw,3061
+ccfx-0.2.0.dist-info/LICENSE,sha256=2-M3fBUS3FmrSIrqd3cZDmxXxojWVJtZY-SHSRE6RxM,1098
+ccfx-0.2.0.dist-info/METADATA,sha256=2GD2kIkcVUo6FG4Th3IEAucmYmhZ4DvdbtGPTI328-c,5419
+ccfx-0.2.0.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+ccfx-0.2.0.dist-info/top_level.txt,sha256=_cSvSA1WX2K8TgoV3iBJUdUZZqMKJbOPLNnKLYSLHaw,5
+ccfx-0.2.0.dist-info/RECORD,,

{ccfx-0.1.0.dist-info → ccfx-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

ccfx-0.1.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-ccfx/__init__.py,sha256=VmBeF3oj6JTJ_793d4i8PvhyF8_FxaxA1L_FmHWqitc,142
-ccfx/ccfx.py,sha256=7vDYc8X1bQGHuF8ZWEH3cWkGDJGlZKrG5kcoeG9IZac,22110
-ccfx/excel.py,sha256=cQ4TQW49XqbMB3sSS0IOhO3-WArIolEBIrvOvhFyPtI,4757
-ccfx/mssqlConnection.py,sha256=TwyZXhHHI7zy6BSfH1pszuHVJ5cmndRC5dVxvEtSTks,7904
-ccfx/sqliteConnection.py,sha256=7xET_tV8uu4LEln4nsWzp3YR5HOma15aaILjrv2F5ZM,10866
-ccfx/word.py,sha256=AGa64jX5Zl5qotZh5L0QmrsjTnktIBhmj_ByRKZ88vw,3061
-ccfx-0.1.0.dist-info/LICENSE,sha256=2-M3fBUS3FmrSIrqd3cZDmxXxojWVJtZY-SHSRE6RxM,1098
-ccfx-0.1.0.dist-info/METADATA,sha256=hZSUkqy3F-ZWJBH8w6pPELTR_1rNSahFTA4G28PZfiw,5390
-ccfx-0.1.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-ccfx-0.1.0.dist-info/top_level.txt,sha256=_cSvSA1WX2K8TgoV3iBJUdUZZqMKJbOPLNnKLYSLHaw,5
-ccfx-0.1.0.dist-info/RECORD,,