mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-06-25 21:11:11 +08:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
"""
|
|
Convert to json module
|
|
"""
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def convert_to_json(data: dict, filename: str, position: str = None):
|
|
"""
|
|
Convert data to JSON format and save it to a file.
|
|
|
|
Args:
|
|
data (dict): Data to save.
|
|
filename (str): Name of the file to save without .json extension.
|
|
position (str): Directory where the file should be saved. If None,
|
|
the directory of the caller script will be used.
|
|
|
|
Raises:
|
|
ValueError: If filename contains '.json'.
|
|
FileNotFoundError: If the specified directory does not exist.
|
|
PermissionError: If the program does not have permission to write to the directory.
|
|
"""
|
|
if ".json" in filename:
|
|
filename = filename.replace(".json", "") # Remove .csv extension
|
|
|
|
# Get the directory of the caller script
|
|
if position is None:
|
|
# Get directory of the main script
|
|
caller_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
|
|
position = caller_dir
|
|
|
|
try:
|
|
os.makedirs(position, exist_ok=True)
|
|
with open(os.path.join(position, f"{filename}.json"), "w", encoding="utf-8") as f:
|
|
f.write(json.dumps(data))
|
|
except FileNotFoundError as fnfe:
|
|
raise FileNotFoundError(
|
|
f"The specified directory '{position}' does not exist.") from fnfe
|
|
except PermissionError as pe:
|
|
raise PermissionError(
|
|
f"You don't have permission to write to '{position}'.") from pe
|