Spaces:
Sleeping
Sleeping
File size: 3,618 Bytes
196c707 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 | """
Export Utilities - Handle various export formats
"""
import json
import csv
import io
from typing import Dict, List
from datetime import datetime
def save_json_file(data: Dict, prefix: str = "export") -> str:
"""
Save data to JSON file and return filepath.
Args:
data: Data to save
prefix: Filename prefix
Returns:
Path to saved file
"""
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"{prefix}_{timestamp}.json"
with open(filename, 'w', encoding='utf-8') as f:
json.dump(data, f, indent=2, ensure_ascii=False)
return filename
def survey_to_csv(survey_data: Dict) -> str:
"""
Convert survey to CSV format (one row per question).
Args:
survey_data: Survey dictionary
Returns:
Path to CSV file
"""
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"survey_{timestamp}.csv"
with open(filename, 'w', newline='', encoding='utf-8') as f:
writer = csv.writer(f)
# Write header
writer.writerow(['Question ID', 'Question Text', 'Type', 'Options', 'Required', 'Help Text'])
# Write questions
for q in survey_data.get('questions', []):
writer.writerow([
q.get('id', ''),
q.get('question_text', ''),
q.get('question_type', ''),
'; '.join(q.get('options', [])) if q.get('options') else '',
'Yes' if q.get('required', False) else 'No',
q.get('help_text', '')
])
return filename
def responses_to_csv(responses: List[Dict], filename_prefix: str = "responses") -> str:
"""
Convert responses to CSV format.
Args:
responses: List of response dictionaries
filename_prefix: Prefix for filename
Returns:
Path to CSV file
"""
if not responses:
return None
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"{filename_prefix}_{timestamp}.csv"
# Get all unique keys from all responses
all_keys = set()
for response in responses:
if isinstance(response, dict):
all_keys.update(response.keys())
fieldnames = sorted(all_keys)
with open(filename, 'w', newline='', encoding='utf-8') as f:
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer.writeheader()
for response in responses:
if isinstance(response, dict):
writer.writerow(response)
return filename
def analysis_to_markdown_file(analysis_report: str, prefix: str = "analysis_report") -> str:
"""
Save analysis report to markdown file.
Args:
analysis_report: Markdown formatted report
prefix: Filename prefix
Returns:
Path to markdown file
"""
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
filename = f"{prefix}_{timestamp}.md"
with open(filename, 'w', encoding='utf-8') as f:
f.write(analysis_report)
return filename
def create_survey_package(survey_data: Dict) -> Dict[str, str]:
"""
Create a complete package of survey files (JSON, CSV, etc.).
Args:
survey_data: Survey dictionary
Returns:
Dictionary mapping format to filepath
"""
package = {}
# Save JSON
package['json'] = save_json_file(survey_data, "survey")
# Save CSV
package['csv'] = survey_to_csv(survey_data)
return package
|