FC_ExporterTool_v1.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Created directory: D:\\Rameez\\FC_Exports\n",
      "Feature class exported to CSV format: D:\\Rameez\\FC_Exports\\feature_class_export_20241121_105746.csv\n",
      "Invalid export type. Please choose 'csv', 'json', or 'geojson'.\n",
      "Invalid export type. Please choose 'csv', 'json', or 'geojson'.\n",
      "Exiting the program.\n"
     ]
    }
   ],
   "source": [
    "import arcpy\n",
    "import json\n",
    "import os\n",
    "import csv\n",
    "from datetime import datetime\n",
    "\n",
    "def get_export_type():\n",
    "    \"\"\"\n",
    "    Prompt the user for the export type and validate the input.\n",
    "    \n",
    "    Returns:\n",
    "    str: The validated export type ('csv', 'json', 'geojson').\n",
    "    \"\"\"\n",
    "    while True:\n",
    "        export_type = input(\"Enter export type (csv, json, geojson), or 'exit' to quit: \").strip().lower()\n",
    "        if export_type in ['csv', 'json', 'geojson']:\n",
    "            return export_type\n",
    "        elif export_type == 'exit':\n",
    "            print(\"Exiting the program.\")\n",
    "            return None\n",
    "        else:\n",
    "            print(\"Invalid export type. Please choose 'csv', 'json', or 'geojson'.\")\n",
    "\n",
    "def export_feature_class(fc, export_type, output_dir):\n",
    "    \"\"\"\n",
    "    Export the feature class to the specified format (csv, json, geojson).\n",
    "    \n",
    "    Parameters:\n",
    "    fc (str): Path to the feature class.\n",
    "    export_type (str): The export format ('csv', 'json', 'geojson').\n",
    "    output_dir (str): Directory where the output file will be saved.\n",
    "    \"\"\"\n",
    "    current_time = datetime.now().strftime(\"%Y%m%d_%H%M%S\")\n",
    "    output_file = os.path.join(output_dir, f\"feature_class_export_{current_time}.{export_type}\")\n",
    "\n",
    "    if not os.path.exists(output_dir):\n",
    "        os.makedirs(output_dir)\n",
    "        print(f\"Created directory: {output_dir}\")\n",
    "\n",
    "    fields = [field.name for field in arcpy.ListFields(fc)] + [\"SHAPE@\"]\n",
    "\n",
    "    features = []\n",
    "\n",
    "    try:\n",
    "        with arcpy.da.SearchCursor(fc, fields) as cursor:\n",
    "            for row in cursor:\n",
    "                feature = {field: value for field, value in zip(fields[:-1], row[:-1])}\n",
    "                geometry = row[-1]\n",
    "                feature[\"geometry\"] = geometry.__geo_interface__ if geometry else None\n",
    "                features.append(feature)\n",
    "\n",
    "        if export_type == 'geojson':\n",
    "            geojson_output = {\n",
    "                \"type\": \"FeatureCollection\",\n",
    "                \"features\": [\n",
    "                    {\n",
    "                        \"type\": \"Feature\",\n",
    "                        \"geometry\": feature[\"geometry\"],\n",
    "                        \"properties\": {key: feature[key] for key in feature if key != \"geometry\"}\n",
    "                    }\n",
    "                    for feature in features\n",
    "                ]\n",
    "            }\n",
    "            with open(output_file, 'w', encoding='utf-8') as f:\n",
    "                json.dump(geojson_output, f, ensure_ascii=False, indent=4)\n",
    "            print(f\"Feature class exported to GeoJSON format: {output_file}\")\n",
    "\n",
    "        elif export_type == 'json':\n",
    "            with open(output_file, 'w', encoding='utf-8') as f:\n",
    "                json.dump(features, f, ensure_ascii=False, indent=4)\n",
    "            print(f\"Feature class exported to JSON format: {output_file}\")\n",
    "\n",
    "        elif export_type == 'csv':\n",
    "            with open(output_file, 'w', newline='', encoding='utf-8') as f:\n",
    "                writer = csv.DictWriter(f, fieldnames=features[0].keys())\n",
    "                writer.writeheader()\n",
    "                for feature in features:\n",
    "                    writer.writerow(feature)\n",
    "            print(f\"Feature class exported to CSV format: {output_file}\")\n",
    "\n",
    "    except Exception as e:\n",
    "        print(f\"Error occurred during export: {e}\")\n",
    "\n",
    "def run():\n",
    "    fc = r\"D:\\ArcPro Projects\\EOP\\EOP.gdb\\Edition5_EOP\"\n",
    "    output_dir = r\"D:\\Rameez\\FC_Exports\"\n",
    "\n",
    "    while True:\n",
    "        export_type = get_export_type()\n",
    "        if export_type is None:\n",
    "            break\n",
    "        export_feature_class(fc, export_type, output_dir)\n",
    "\n",
    "run()\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}