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

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 

11 

12 

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

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

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

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

17 

18 try: 

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

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

21 rows = list(reader) 

22 except Exception as e: 

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

24 

25 headers = reader.fieldnames or [] 

26 

27 if action == "columns": 

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

29 

30 if action == "read": 

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

32 

33 if action == "filter": 

34 if not column or column not in headers: 

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

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

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

38 

39 if action == "sort": 

40 if not sort_by or sort_by not in headers: 

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

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

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

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

45 

46 if action == "aggregate": 

47 if column not in headers: 

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

49 values = [] 

50 for r in rows: 

51 try: 

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

53 except (ValueError, TypeError): 

54 pass 

55 if not values: 

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

57 if aggregate == "sum": 

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

59 if aggregate == "avg": 

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

61 if aggregate == "min": 

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

63 if aggregate == "max": 

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

65 if aggregate == "count": 

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

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

68 

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

70 

71 

72def _format(rows, headers): 

73 if not rows: 

74 return "[spreadsheet] No matching rows." 

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

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

77 return result 

78 

79 

80__all__ = ["run"]