Added notebook

Author: viragtripathi

README.md

@@ -222,4 +222,4 @@ python tests/test_runner.py
 
 ---
 
-📓 [Try it in a Jupyter notebook](https://raw.githubusercontent.com/viragtripathi/crdb-sql-audit/main/notebooks/demo_crdb_sql_audit.ipynb)
+📓 [Try it in a Jupyter notebook](notebooks/demo_crdb_sql_audit.ipynb)

notebooks/demo_crdb_sql_audit.ipynb

@@ -8,26 +8,36 @@
     {
       "cell_type": "code",
       "metadata": {},
+      "outputs": [],
+      "execution_count": null,
       "source": "# Install the tool (if not already installed in the environment)\n!pip install crdb-sql-audit"
     },
     {
       "cell_type": "code",
       "metadata": {},
+      "outputs": [],
+      "execution_count": null,
       "source": "# Step 1: Load the rules YAML\nimport yaml\n\nwith open(\"rules/test_rules.yaml\") as f:\n    rules = yaml.safe_load(f)\n\nprint(f\"\u2705 Loaded {len(rules)} rules\")"
     },
     {
       "cell_type": "code",
       "metadata": {},
+      "outputs": [],
+      "execution_count": null,
       "source": "# Step 2: Load a test log file (SQL lines)\nwith open(\"tests/sample_logs/basic_pg_chunk.log\") as f:\n    raw_lines = [line.strip() for line in f if line.strip()]\n\nprint(f\"\u2705 Loaded {len(raw_lines)} lines\")"
     },
     {
       "cell_type": "code",
       "metadata": {},
+      "outputs": [],
+      "execution_count": null,
       "source": "# Step 3: Use rule engine directly\nimport re\nfrom crdb_sql_audit.rules_engine import apply_rules\n\nall_matches = []\nfor line in raw_lines:\n    matches = apply_rules(line, rules)\n    all_matches.extend(matches)\n\nprint(f\"\ud83d\udea8 Total matches: {len(all_matches)}\")"
     },
     {
       "cell_type": "code",
       "metadata": {},
+      "outputs": [],
+      "execution_count": null,
       "source": "# Step 4: Visualize results\nimport pandas as pd\n\nif all_matches:\n    df = pd.DataFrame(all_matches)\n    display(df[['Rule_ID', 'SQL_Type', 'Issue', 'Example']])\nelse:\n    print(\"No matches found.\")"
     }
   ],
@@ -43,5 +53,5 @@
     }
   },
   "nbformat": 4,
-  "nbformat_minor": 2
+  "nbformat_minor": 5
 }