Write result after each search query about google-maps-scraper HOT 1 CLOSED

def write(data):
    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
    csv_filename = f"data_{timestamp}.csv"
    json_filename = f"data_{timestamp}.json"

    # Check if the CSV file already exists
    if os.path.exists(csv_filename):
        # Read the existing data from the CSV file
        existing_data = pd.read_csv(csv_filename)

        # Append new data to the existing data
        new_data = pd.DataFrame(data)
        updated_data = pd.concat([existing_data, new_data], ignore_index=True)

        # Write the updated data to the CSV file
        updated_data.to_csv(csv_filename, index=False, mode='a', header=not existing_data.empty)

    else:
        # File does not exist, create a new file and write the data
        new_data = pd.DataFrame(data)
        new_data.to_csv(csv_filename, index=False)

    # Check if the JSON file already exists
    if os.path.exists(json_filename):
        # Read the existing data from the JSON file
        with open(json_filename, 'r') as file:
            existing_data = json.load(file)

        # Append new data to the existing data
        existing_data.append(data)

        # Write the updated data to the JSON file
        with open(json_filename, 'w') as file:
            json.dump(existing_data, file)

    else:
        # File does not exist, create a new file and write the data
        with open(json_filename, 'w') as file:
            json.dump([data], file)
    print("Data saved to CSV and JSON files")