Coverage for agentos/marketplace/skills/spreadsheet/spreadsheet.py: 68%

1""" 

2spreadsheet — CSV-based spreadsheet operations. 

3 

4Actions: read, filter, sort, aggregate, columns 

5Lightweight CSV manipulation without pandas. 

6""" 

7 

8import csv 

9import os 

10from typing import Any 

11from io import StringIO 

12 

13 

14def run(action: str = "read", file_path: str = "", column: str = "", condition: str = "", 

15 sort_by: str = "", aggregate: str = "", delimiter: str = ",", **kwargs: Any) -> str: 

16 if not file_path or not os.path.exists(file_path): 

17 return f"[spreadsheet] File not found: {file_path}" 

18 

19 try: 

20 with open(file_path, newline="", encoding="utf-8") as f: 

21 reader = csv.DictReader(f, delimiter=delimiter) 

22 rows = list(reader) 

23 except Exception as e: 

24 return f"[spreadsheet] Read error: {e}" 

25 

26 headers = reader.fieldnames or [] 

27 

28 if action == "columns": 

29 return f"Columns ({len(headers)}): {', '.join(headers)}" 

30 

31 if action == "read": 

32 return _format(rows[:20], headers) 

33 

34 if action == "filter": 

35 if not column or column not in headers: 

36 return f"[spreadsheet] Column '{column}' not found. Available: {headers}" 

37 filtered = [r for r in rows if condition.lower() in r.get(column, "").lower()] 

38 return _format(filtered[:30], headers) 

39 

40 if action == "sort": 

41 if not sort_by or sort_by not in headers: 

42 return f"[spreadsheet] Column '{sort_by}' not found." 

43 descending = kwargs.get("desc", False) 

44 sorted_rows = sorted(rows, key=lambda r: r.get(sort_by, ""), reverse=descending) 

45 return _format(sorted_rows[:30], headers) 

46 

47 if action == "aggregate": 

48 if column not in headers: 

49 return f"[spreadsheet] Column '{column}' not found." 

50 values = [] 

51 for r in rows: 

52 try: 

53 values.append(float(r.get(column, 0))) 

54 except (ValueError, TypeError): 

55 pass 

56 if not values: 

57 return f"[spreadsheet] No numeric values in '{column}'." 

58 if aggregate == "sum": 

59 return f"Sum of {column}: {sum(values)}" 

60 if aggregate == "avg": 

61 return f"Average of {column}: {sum(values)/len(values):.2f}" 

62 if aggregate == "min": 

63 return f"Min of {column}: {min(values)}" 

64 if aggregate == "max": 

65 return f"Max of {column}: {max(values)}" 

66 if aggregate == "count": 

67 return f"Count of {column}: {len(values)}" 

68 return f"Sum: {sum(values)}, Avg: {sum(values)/len(values):.2f}, Min: {min(values)}, Max: {max(values)}, Count: {len(values)}" 

69 

70 return f"[spreadsheet] Unknown action: {action}. Available: read, filter, sort, aggregate, columns" 

71 

72 

73def _format(rows, headers): 

74 if not rows: 

75 return "[spreadsheet] No matching rows." 

76 result = "\t".join(headers) + "\n" 

77 result += "\n".join("\t".join(str(r.get(h, "")) for h in headers) for r in rows) 

78 return result 

79 

80 

81__all__ = ["run"]