code tidy

Author: nickzoic

countess/core/plugins.py

@@ -231,6 +231,10 @@ def execute(
         ]
         if len(tablenames_filenames_and_params) == 0:
             return None
+
+        for num, (tablename, filename, _) in enumerate(tablenames_filenames_and_params, 1):
+            logger.debug("DuckdbLoadFilePlugin.execute File #%d %s %s", num, tablename, repr(filename))
+
         row_limit_per_file = (row_limit // len(tablenames_filenames_and_params)) if row_limit else None
         logger.debug("row_limit_per_file %s", row_limit_per_file)
 

countess/plugins/csv.py

@@ -101,6 +101,7 @@ class SaveCsvOrderParameter(TabularMultiParam):
     order_by = ColumnChoiceParam("Order By")
     descending = BooleanParam("Descending")
 
+
 class SaveCsvPlugin(DuckdbSaveFilePlugin):
     name = "CSV Save"
     description = "Save data as CSV or similar delimited text files"
@@ -126,10 +127,11 @@ def execute(
     ) -> None:
         filename = self.filename.value
 
-        if len(self.sorting):
+        if source is None:
+            return
+        elif len(self.sorting):
             order_by = ",".join(
-                duckdb_escape_identifier(sp.order_by.value) + (" desc" if sp.descending else "")
-                for sp in self.sorting
+                duckdb_escape_identifier(sp.order_by.value) + (" desc" if sp.descending else "") for sp in self.sorting
             )
             logger.debug("SaveCsvPlugin.execute order_by %s", order_by)
             table = source.order(order_by)

countess/plugins/frequency.py

@@ -25,7 +25,7 @@ def prepare(self, ddbc: DuckDBPyConnection, source: Optional[DuckDBPyRelation])
         # set default values for converting "count" columns
         if not any(cp.value for cp in self.columns):
             for cp in self.columns:
-                if 'count' in cp.label:
+                if "count" in cp.label:
                     cp.value = True
 
     def sql(self, table_name: str, columns: Iterable[str]) -> Optional[str]: