Screenshot tool.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d7c125d0",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Screenshot Tool Installer\n",
    "pip install keyboard\n",
    "pip install pyautogui"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "4d74e89e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1/2 Enter the number of screenshots to take: 2\n",
      "2/2 Enter the number of seconds to pause in between each screenshot\n",
      "Set this equal to the Auto Skip Interval under View > Preferences: 3\n",
      "Taking 2 screenshots. Press 'Esc' to stop early.\n",
      "Starting in 10 seconds...\n",
      "Starting in 9 seconds...\n",
      "Starting in 8 seconds...\n",
      "Starting in 7 seconds...\n",
      "Starting in 6 seconds...\n",
      "Starting in 5 seconds...\n",
      "Starting in 4 seconds...\n",
      "Starting in 3 seconds...\n",
      "Starting in 2 seconds...\n",
      "Starting in 1 seconds...\n",
      "Screenshot saved to C:\\Users\\JoelTencer\\Pictures\\Screenshots\\2024-08-20_13-43-49\\screenshot_2024-08-20_13-43-59.png\n",
      "Screenshot saved to C:\\Users\\JoelTencer\\Pictures\\Screenshots\\2024-08-20_13-43-49\\screenshot_2024-08-20_13-44-03.png\n"
     ]
    },
    {
     "ename": "KeyboardInterrupt",
     "evalue": "Interrupted by user",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
      "Cell \u001b[1;32mIn[2], line 93\u001b[0m\n\u001b[0;32m     90\u001b[0m keyboard\u001b[38;5;241m.\u001b[39munhook_all()\n\u001b[0;32m     92\u001b[0m \u001b[38;5;66;03m# Infinite pause after Esc is pressed and text is printed\u001b[39;00m\n\u001b[1;32m---> 93\u001b[0m \u001b[38;5;28minput\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mPress Enter to exit...\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
      "File \u001b[1;32m~\\anaconda3\\Lib\\site-packages\\ipykernel\\kernelbase.py:1262\u001b[0m, in \u001b[0;36mKernel.raw_input\u001b[1;34m(self, prompt)\u001b[0m\n\u001b[0;32m   1260\u001b[0m     msg \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mraw_input was called, but this frontend does not support input requests.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[0;32m   1261\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m StdinNotImplementedError(msg)\n\u001b[1;32m-> 1262\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_input_request(\n\u001b[0;32m   1263\u001b[0m     \u001b[38;5;28mstr\u001b[39m(prompt),\n\u001b[0;32m   1264\u001b[0m     \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_parent_ident[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mshell\u001b[39m\u001b[38;5;124m\"\u001b[39m],\n\u001b[0;32m   1265\u001b[0m     \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mget_parent(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mshell\u001b[39m\u001b[38;5;124m\"\u001b[39m),\n\u001b[0;32m   1266\u001b[0m     password\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mFalse\u001b[39;00m,\n\u001b[0;32m   1267\u001b[0m )\n",
      "File \u001b[1;32m~\\anaconda3\\Lib\\site-packages\\ipykernel\\kernelbase.py:1305\u001b[0m, in \u001b[0;36mKernel._input_request\u001b[1;34m(self, prompt, ident, parent, password)\u001b[0m\n\u001b[0;32m   1302\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mKeyboardInterrupt\u001b[39;00m:\n\u001b[0;32m   1303\u001b[0m     \u001b[38;5;66;03m# re-raise KeyboardInterrupt, to truncate traceback\u001b[39;00m\n\u001b[0;32m   1304\u001b[0m     msg \u001b[38;5;241m=\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mInterrupted by user\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m-> 1305\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m \u001b[38;5;167;01mKeyboardInterrupt\u001b[39;00m(msg) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m\n\u001b[0;32m   1306\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mException\u001b[39;00m:\n\u001b[0;32m   1307\u001b[0m     \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mlog\u001b[38;5;241m.\u001b[39mwarning(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mInvalid Message:\u001b[39m\u001b[38;5;124m\"\u001b[39m, exc_info\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mTrue\u001b[39;00m)\n",
      "\u001b[1;31mKeyboardInterrupt\u001b[0m: Interrupted by user"
     ]
    }
   ],
   "source": [
    "# v0.7\n",
    "# added longer delay countdown\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "try:\n",
    "    # Get the number of screenshots from the user\n",
    "    num_screenshots = int(input(\"1/2 Enter the number of screenshots to take: \"))\n",
    "except ValueError:\n",
    "    print(\"Invalid input. Please enter a valid number.\")\n",
    "    exit(1)\n",
    "\n",
    "try:\n",
    "    # Get the delay between screenshots in seconds from the user\n",
    "    screenshot_delay_seconds = float(input(\"2/2 Enter the number of seconds to pause in between each screenshot: \"))\n",
    "except ValueError:\n",
    "    print(\"Invalid input. Please enter a valid number.\")\n",
    "    exit(1)\n",
    "\n",
    "# Delay time for screenshots in 0.1-second intervals\n",
    "screenshot_delay = int(screenshot_delay_seconds * 10)\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "def on_esc_pressed(event):\n",
    "    if event.name == 'esc':\n",
    "        print(\"Esc key pressed. Exiting...\")\n",
    "        global exit_flag\n",
    "        exit_flag = True\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    # Countdown for the initial 10-second delay\n",
    "    for countdown in range(10, 0, -1):\n",
    "        print(f\"Starting in {countdown} seconds...\")\n",
    "        time.sleep(1)\n",
    "    \n",
    "    # Register the handler for the Esc key press event\n",
    "    keyboard.on_press(on_esc_pressed)\n",
    "    \n",
    "    exit_flag = False\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        if exit_flag:\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for the user-defined delay, checking for the exit flag\n",
    "        for _ in range(screenshot_delay):  # checks in 0.1-second intervals\n",
    "            if exit_flag:\n",
    "                break\n",
    "            time.sleep(0.1)\n",
    "    \n",
    "    # Unregister the event handler\n",
    "    keyboard.unhook_all()\n",
    "    \n",
    "    # Infinite pause after Esc is pressed and text is printed\n",
    "    input(\"Press Enter to exit...\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0ba8ec41",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.6\n",
    "# Let user define screenshot delay\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "try:\n",
    "    # Get the number of screenshots from the user\n",
    "    num_screenshots = int(input(\"1/2 Enter the number of screenshots to take: \"))\n",
    "except ValueError:\n",
    "    print(\"Invalid input. Please enter a valid number.\")\n",
    "    exit(1)\n",
    "\n",
    "try:\n",
    "    # Get the delay between screenshots in seconds from the user\n",
    "    screenshot_delay_seconds = int(input(\"2/2 Enter the number of seconds to pause in between each screenshot\\nSet this equal to the Auto Skip Interval under View > Preferences: \"))\n",
    "except ValueError:\n",
    "    print(\"Invalid input. Please enter a valid number.\")\n",
    "    exit(1)\n",
    "\n",
    "# Delay time for screenshots in 0.1-second intervals\n",
    "screenshot_delay = screenshot_delay_seconds * 10\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "def on_esc_pressed(event):\n",
    "    if event.name == 'esc':\n",
    "        print(\"Esc key pressed. Exiting...\")\n",
    "        global exit_flag\n",
    "        exit_flag = True\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    # Countdown for the initial 2-second delay\n",
    "    for countdown in range(2, 0, -1):\n",
    "        print(f\"Starting in {countdown} seconds...\")\n",
    "        time.sleep(1)\n",
    "    \n",
    "    # Register the handler for the Esc key press event\n",
    "    keyboard.on_press(on_esc_pressed)\n",
    "    \n",
    "    exit_flag = False\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        if exit_flag:\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for the user-defined delay, checking for the exit flag\n",
    "        for _ in range(screenshot_delay):  # checks in 0.1-second intervals\n",
    "            if exit_flag:\n",
    "                break\n",
    "            time.sleep(0.1)\n",
    "    \n",
    "    # Unregister the event handler\n",
    "    keyboard.unhook_all()\n",
    "    \n",
    "   \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "70924e30",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.5 \n",
    "# fix esc press end so message is displayed\n",
    "# Add counter to start\n",
    "# Defined and moved screenshot delay to top\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "\n",
    "# Delay time for screenshots\n",
    "# 50 = 5s, checks in 5 seconds (0.1-second intervals)\n",
    "screenshot_delay = 50\n",
    "\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "def on_esc_pressed(event):\n",
    "    if event.name == 'esc':\n",
    "        print(\"Esc key pressed. Exiting...\")\n",
    "        global exit_flag\n",
    "        exit_flag = True\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    try:\n",
    "        # Get the number of screenshots from the user\n",
    "        num_screenshots = int(input(\"Enter the number of screenshots to take: \"))\n",
    "    except ValueError:\n",
    "        print(\"Invalid input. Please enter a valid number.\")\n",
    "        exit(1)\n",
    "\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    # Countdown for the initial 2-second delay\n",
    "    for countdown in range(2, 0, -1):\n",
    "        print(f\"Starting in {countdown} seconds...\")\n",
    "        time.sleep(1)\n",
    "    \n",
    "    # Register the handler for the Esc key press event\n",
    "    keyboard.on_press(on_esc_pressed)\n",
    "    \n",
    "    exit_flag = False\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        if exit_flag:\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for 5 seconds, checking for the exit flag\n",
    "        for _ in range(screenshot_delay):  # 50 checks in 5 seconds (0.1-second intervals)\n",
    "            if exit_flag:\n",
    "                break\n",
    "            time.sleep(0.1)\n",
    "    \n",
    "    # Unregister the event handler\n",
    "    keyboard.unhook_all()\n",
    "    \n",
    "    # Infinite pause after Esc is pressed and text is printed\n",
    "    input(\"Press Enter to exit...\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "7392ff9e",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.5 \n",
    "# fix esc press end so message is displayed\n",
    "# Add counter to start\n",
    "# Defined and moved screenshot delay to top\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "\n",
    "# Delay time for screenshots\n",
    "# 50 = 5s, checks in 5 seconds (0.1-second intervals)\n",
    "screenshot_delay = 50\n",
    "\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "def on_esc_pressed(event):\n",
    "    if event.name == 'esc':\n",
    "        print(\"Esc key pressed. Exiting...\")\n",
    "        global exit_flag\n",
    "        exit_flag = True\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    try:\n",
    "        # Get the number of screenshots from the user\n",
    "        num_screenshots = int(input(\"Enter the number of screenshots to take: \"))\n",
    "    except ValueError:\n",
    "        print(\"Invalid input. Please enter a valid number.\")\n",
    "        exit(1)\n",
    "\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    # Countdown for the initial 2-second delay\n",
    "    for countdown in range(2, 0, -1):\n",
    "        print(f\"Starting in {countdown} seconds...\")\n",
    "        time.sleep(1)\n",
    "    \n",
    "    # Register the handler for the Esc key press event\n",
    "    keyboard.on_press(on_esc_pressed)\n",
    "    \n",
    "    exit_flag = False\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        if exit_flag:\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for 5 seconds, checking for the exit flag\n",
    "        for _ in range(screenshot_delay):  # 50 checks in 5 seconds (0.1-second intervals)\n",
    "            if exit_flag:\n",
    "                break\n",
    "            time.sleep(0.1)\n",
    "    \n",
    "    # Unregister the event handler\n",
    "    keyboard.unhook_all()\n",
    "    \n",
    "    # Infinite pause after Esc is pressed and text is printed\n",
    "    input(\"Press Enter to exit...\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e0c27d87",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.3 added 2s delay at start\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "def on_esc_pressed(event):\n",
    "    if event.name == 'esc':\n",
    "        print(\"Esc key pressed. Exiting...\")\n",
    "        global exit_flag\n",
    "        exit_flag = True\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    try:\n",
    "        # Get the number of screenshots from the user\n",
    "        num_screenshots = int(input(\"Enter the number of screenshots to take: \"))\n",
    "    except ValueError:\n",
    "        print(\"Invalid input. Please enter a valid number.\")\n",
    "        exit(1)\n",
    "\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    # Add a 2-second delay before starting to take screenshots\n",
    "    print(\"Starting in 2 seconds...\")\n",
    "    time.sleep(2)\n",
    "    \n",
    "    # Register the handler for the Esc key press event\n",
    "    keyboard.on_press(on_esc_pressed)\n",
    "    \n",
    "    exit_flag = False\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        if exit_flag:\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for 5 seconds, checking for the exit flag\n",
    "        for _ in range(50):  # 50 checks in 5 seconds (0.1-second intervals)\n",
    "            if exit_flag:\n",
    "                break\n",
    "            time.sleep(0.1)\n",
    "    \n",
    "    # Unregister the event handler\n",
    "    keyboard.unhook_all()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2e704947",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.2\n",
    "# saves to C:/user/Pictures/Screenshot/[date time]\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "def take_screenshot(folder_path):\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string for the screenshot filename\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder_path, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    try:\n",
    "        # Get the number of screenshots from the user\n",
    "        num_screenshots = int(input(\"Enter the number of screenshots to take: \"))\n",
    "    except ValueError:\n",
    "        print(\"Invalid input. Please enter a valid number.\")\n",
    "        exit(1)\n",
    "\n",
    "    # Get the current date and time for the folder name\n",
    "    now = datetime.now()\n",
    "    folder_name = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Get the current user's home directory\n",
    "    home_dir = os.path.expanduser('~')\n",
    "    \n",
    "    # Define the folder path\n",
    "    folder_path = os.path.join(home_dir, \"Pictures\", \"Screenshots\", folder_name)\n",
    "    \n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder_path):\n",
    "        os.makedirs(folder_path)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        # Check if the Esc key is pressed\n",
    "        if keyboard.is_pressed(\"esc\"):\n",
    "            print(\"Esc key pressed. Exiting...\")\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot(folder_path)\n",
    "        \n",
    "        # Wait for 5 seconds, checking for the Esc key press\n",
    "        for _ in range(50):  # 50 checks in 5 seconds (0.1-second intervals)\n",
    "            if keyboard.is_pressed(\"esc\"):\n",
    "                print(\"Esc key pressed. Exiting...\")\n",
    "                exit(0)\n",
    "            time.sleep(0.1)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "886305de",
   "metadata": {},
   "outputs": [],
   "source": [
    "# v0.1 screenshot tool\n",
    "# take screenshot and saves to a folder with date and time\n",
    "# waits 5s between each screenshot\n",
    "# Asks user for # of screenshots to take\n",
    "\n",
    "import pyautogui\n",
    "import os\n",
    "import time\n",
    "from datetime import datetime\n",
    "import keyboard\n",
    "\n",
    "def take_screenshot():\n",
    "    # Get the current date and time\n",
    "    now = datetime.now()\n",
    "    # Format the date and time string\n",
    "    timestamp = now.strftime(\"%Y-%m-%d_%H-%M-%S\")\n",
    "    \n",
    "    # Define the folder to save the screenshot\n",
    "    folder = \"screenshots\"\n",
    "    # Create the folder if it doesn't exist\n",
    "    if not os.path.exists(folder):\n",
    "        os.makedirs(folder)\n",
    "    \n",
    "    # Define the file path\n",
    "    file_path = os.path.join(folder, f\"screenshot_{timestamp}.png\")\n",
    "    \n",
    "    # Take the screenshot\n",
    "    screenshot = pyautogui.screenshot()\n",
    "    \n",
    "    # Save the screenshot to the file\n",
    "    screenshot.save(file_path)\n",
    "    print(f\"Screenshot saved to {file_path}\")\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    try:\n",
    "        # Get the number of screenshots from the user\n",
    "        num_screenshots = int(input(\"Enter the number of screenshots to take: \"))\n",
    "    except ValueError:\n",
    "        print(\"Invalid input. Please enter a valid number.\")\n",
    "        exit(1)\n",
    "\n",
    "    print(f\"Taking {num_screenshots} screenshots. Press 'Esc' to stop early.\")\n",
    "    \n",
    "    for i in range(num_screenshots):\n",
    "        # Check if the Esc key is pressed\n",
    "        if keyboard.is_pressed(\"esc\"):\n",
    "            print(\"Esc key pressed. Exiting...\")\n",
    "            break\n",
    "        \n",
    "        # Take a screenshot\n",
    "        take_screenshot()\n",
    "        \n",
    "        # Wait for 5 seconds, checking for the Esc key press\n",
    "        for _ in range(50):  # 50 checks in 5 seconds (0.1-second intervals)\n",
    "            if keyboard.is_pressed(\"esc\"):\n",
    "                print(\"Esc key pressed. Exiting...\")\n",
    "                exit(0)\n",
    "            time.sleep(0.1)\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}