Create a Data Helper Class in Python

import json
import csv
from pathlib import Path

class DataHelper:
    def __init__(self, base_path="."):
        self.base_path = Path(base_path)
        self.base_path.mkdir(exist_ok=True)

    def save_json(self, data, filename):
        path = self.base_path / filename
        with open(path, "w") as f:
            json.dump(data, f, indent=2)
        print(f"Saved JSON to {path}")

    def save_csv(self, rows, filename, headers=None):
        path = self.base_path / filename
        if headers is None:
            headers = list(rows[0].keys()) if rows else []
        with open(path, "w", newline="") as f:
            writer = csv.DictWriter(f, fieldnames=headers)
            writer.writeheader()
            writer.writerows(rows)
        print(f"Saved CSV to {path}")

    def load_json(self, filename):
        path = self.base_path / filename
        with open(path) as f:
            return json.load(f)

    def load_csv(self, filename):
        path = self.base_path / filename
        with open(path) as f:
            reader = csv.DictReader(f)
            return [dict(row) for row in reader]

if __name__ == "__main__":
    helper = DataHelper("data_examples")
    sample_data = [
        {"name": "Alice", "age": 30},
        {"name": "Bob", "age": 25}
    ]
    helper.save_json(sample_data, "people.json")
    helper.save_csv(sample_data, "people.csv")
    
    loaded_json = helper.load_json("people.json")
    loaded_csv = helper.load_csv("people.csv")
    
    print(f"JSON loaded: {loaded_json[0]['name']}")
    print(f"CSV loaded: {loaded_csv[1]['age']}")