Create a Data Helper Class for Beginners in Python

import json
from pathlib import Path

class DataHelper:
    """Simple helper for reading and writing common data files."""
    
    def __init__(self, directory="data"):
        self.directory = Path(directory)
        self.directory.mkdir(exist_ok=True)
    
    def save_json(self, filename, data):
        filepath = self.directory / f"{filename}.json"
        with filepath.open("w") as f:
            json.dump(data, f, indent=2)
        return filepath
    
    def load_json(self, filename):
        filepath = self.directory / f"{filename}.json"
        with filepath.open("r") as f:
            return json.load(f)
    
    def save_csv(self, filename, headers, rows):
        filepath = self.directory / f"{filename}.csv"
        with filepath.open("w") as f:
            f.write(",".join(headers) + "\n")
            for row in rows:
                f.write(",".join(str(item) for item in row) + "\n")
        return filepath

if __name__ == "__main__":
    helper = DataHelper()
    
    users = [{"id": 1, "name": "Alice", "role": "admin"}, 
             {"id": 2, "name": "Bob", "role": "user"}]
    
    json_path = helper.save_json("users", users)
    loaded = helper.load_json("users")
    
    csv_path = helper.save_csv("logs", 
                               ["timestamp", "level", "message"],
                               [["2024-01-01", "INFO", "Started"],
                                ["2024-01-01", "WARN", "Retry"]])
    
    print(f"JSON saved: {json_path}")
    print(f"JSON loaded: {loaded}")
    print(f"CSV saved: {csv_path}")