changes

examples/customer-support/customer-support.ipynb

@@ -99,46 +99,50 @@
     "    # Backup - we will use this to \"reset\" our DB in each section\n",
     "    shutil.copy(local_file, backup_file)\n",
     "# Convert the flights to present time for our tutorial\n",
-    "conn = sqlite3.connect(local_file)\n",
-    "cursor = conn.cursor()\n",
+    "def update_dates(file):\n",
+    "    shutil.copy(backup_file, file)\n",
+    "    conn = sqlite3.connect(file)\n",
+    "    cursor = conn.cursor()\n",
     "\n",
-    "tables = pd.read_sql(\n",
-    "    \"SELECT name FROM sqlite_master WHERE type='table';\", conn\n",
-    ").name.tolist()\n",
-    "tdf = {}\n",
-    "for t in tables:\n",
-    "    tdf[t] = pd.read_sql(f\"SELECT * from {t}\", conn)\n",
+    "    tables = pd.read_sql(\n",
+    "        \"SELECT name FROM sqlite_master WHERE type='table';\", conn\n",
+    "    ).name.tolist()\n",
+    "    tdf = {}\n",
+    "    for t in tables:\n",
+    "        tdf[t] = pd.read_sql(f\"SELECT * from {t}\", conn)\n",
     "\n",
-    "example_time = pd.to_datetime(\n",
-    "    tdf[\"flights\"][\"actual_departure\"].replace(\"\\\\N\", pd.NaT)\n",
-    ").max()\n",
-    "current_time = pd.to_datetime(\"now\").tz_localize(example_time.tz)\n",
-    "time_diff = current_time - example_time\n",
+    "    example_time = pd.to_datetime(\n",
+    "        tdf[\"flights\"][\"actual_departure\"].replace(\"\\\\N\", pd.NaT)\n",
+    "    ).max()\n",
+    "    current_time = pd.to_datetime(\"now\").tz_localize(example_time.tz)\n",
+    "    time_diff = current_time - example_time\n",
     "\n",
-    "tdf[\"bookings\"][\"book_date\"] = (\n",
-    "    pd.to_datetime(tdf[\"bookings\"][\"book_date\"].replace(\"\\\\N\", pd.NaT), utc=True)\n",
-    "    + time_diff\n",
-    ")\n",
-    "\n",
-    "datetime_columns = [\n",
-    "    \"scheduled_departure\",\n",
-    "    \"scheduled_arrival\",\n",
-    "    \"actual_departure\",\n",
-    "    \"actual_arrival\",\n",
-    "]\n",
-    "for column in datetime_columns:\n",
-    "    tdf[\"flights\"][column] = (\n",
-    "        pd.to_datetime(tdf[\"flights\"][column].replace(\"\\\\N\", pd.NaT)) + time_diff\n",
+    "    tdf[\"bookings\"][\"book_date\"] = (\n",
+    "        pd.to_datetime(tdf[\"bookings\"][\"book_date\"].replace(\"\\\\N\", pd.NaT), utc=True)\n",
+    "        + time_diff\n",
     "    )\n",
     "\n",
-    "for table_name, df in tdf.items():\n",
-    "    df.to_sql(table_name, conn, if_exists=\"replace\", index=False)\n",
-    "del df\n",
-    "del tdf\n",
-    "conn.commit()\n",
-    "conn.close()\n",
+    "    datetime_columns = [\n",
+    "        \"scheduled_departure\",\n",
+    "        \"scheduled_arrival\",\n",
+    "        \"actual_departure\",\n",
+    "        \"actual_arrival\",\n",
+    "    ]\n",
+    "    for column in datetime_columns:\n",
+    "        tdf[\"flights\"][column] = (\n",
+    "            pd.to_datetime(tdf[\"flights\"][column].replace(\"\\\\N\", pd.NaT)) + time_diff\n",
+    "        )\n",
     "\n",
-    "db = local_file  # We'll be using this local file as our DB in this tutorial"
+    "    for table_name, df in tdf.items():\n",
+    "        df.to_sql(table_name, conn, if_exists=\"replace\", index=False)\n",
+    "    del df\n",
+    "    del tdf\n",
+    "    conn.commit()\n",
+    "    conn.close()\n",
+    "\n",
+    "    return file\n",
+    "\n",
+    "db = update_dates(local_file)"
    ]
   },
   {
@@ -1750,7 +1754,7 @@
     "]\n",
     "\n",
     "# Update with the backup file so we can restart from the original place in each section\n",
-    "shutil.copy(backup_file, db)\n",
+    "db = update_dates(db)\n",
     "thread_id = str(uuid.uuid4())\n",
     "\n",
     "config = {\n",
@@ -2304,7 +2308,7 @@
     "import uuid\n",
     "\n",
     "# Update with the backup file so we can restart from the original place in each section\n",
-    "shutil.copy(backup_file, db)\n",
+    "db = update_dates(db)\n",
     "thread_id = str(uuid.uuid4())\n",
     "\n",
     "config = {\n",
@@ -2908,7 +2912,7 @@
     "import uuid\n",
     "\n",
     "# Update with the backup file so we can restart from the original place in each section\n",
-    "shutil.copy(backup_file, db)\n",
+    "db = update_dates(db)\n",
     "thread_id = str(uuid.uuid4())\n",
     "\n",
     "config = {\n",
@@ -4330,7 +4334,7 @@
     "import uuid\n",
     "\n",
     "# Update with the backup file so we can restart from the original place in each section\n",
-    "shutil.copy(backup_file, db)\n",
+    "db = update_dates(db)\n",
     "thread_id = str(uuid.uuid4())\n",
     "\n",
     "config = {\n",