Database_Checker.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 44,
   "metadata": {},
   "outputs": [],
   "source": [
    "from collections import *\n",
    "import csv"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {},
   "outputs": [],
   "source": [
    "def federal_state_search():\n",
    "    from pprint import pprint\n",
    "    fed_state_search = ''\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    federal_state_results = []\n",
    "    state_abbreviations = [\n",
    "    'AL','al','AK','ak','AZ','az','AR','ar','CA','ca','CO','co',\n",
    "    'CT','ct','DE','de','FL','fl','GA','ga','HI','hi','ID','id',\n",
    "    'IL','il','IN','in','IA','ia','KS','ks','KY','ky','LA','la',\n",
    "    'ME','me','MD','md','MA','ma','MI','mi','MN','mn','MS','ma',\n",
    "    'MO','mo','MT','mt','NE','ne','NV','nv','NH','nh','NJ','nj',\n",
    "    'NM','nm','NY','ny','NC','nc','ND','nd','OH','oh','OK','ok',\n",
    "    'OR','or','PA','pa','RI','ri','SC','sc','SD','sd','TN','tn',\n",
    "    'TX','tx','UT','ut','VT','vt','VA','va','WA','wa','WV','wv',\n",
    "    'WI','wi','WY','wy']\n",
    "    while True:\n",
    "        fed_state_search = input(\"What state are you looking to find chargers? (Or enter '0' to exit):\\t\")\n",
    "        if fed_state_search == '0':\n",
    "            break\n",
    "        if fed_state_search in state_abbreviations:\n",
    "            break\n",
    "        else:\n",
    "            print('Invalid input. Please enter the abbreviation of the state you are looking for.')\n",
    "            fed_state_search\n",
    "    \n",
    "    fed_state_search = fed_state_search.upper()\n",
    "    \n",
    "    while True:\n",
    "        if fed_state_search == '0':\n",
    "            break\n",
    "\n",
    "        with open('data/alt_fuel_stations (Oct 4 2020).csv', 'r') as file:\n",
    "            stuff = csv.reader(file)\n",
    "            \n",
    "            for line in stuff:\n",
    "                if line[5] in fed_state_search:\n",
    "                    federal_state_results.append(line[4])\n",
    "                    address = line[2],line[4]+', '+line[5]\n",
    "                    address_list.append(address)\n",
    "                    address_list.sort(key = lambda x: x[1]) \n",
    "                    \n",
    "            if len(federal_state_results) > 0:\n",
    "                print('Here are the street addresses of the chargers we found in ' + fed_state_search, '\\n')\n",
    "                pprint(address_list)\n",
    "                break\n",
    "                \n",
    "            if len(federal_state_results) < 1:\n",
    "                print(\"We could not find any chargers in \" + fed_state_search + \". Please try again or enter '0' to exit.\")\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "    while True:\n",
    "        if fed_state_search == '0':\n",
    "            break\n",
    "        count_of_cities_question = input('Would you also like a count of chargers in each city?:\\t')\n",
    "        print('\\n')\n",
    "        if count_of_cities_question in count_yes:\n",
    "            print('Here are those same cities in ' + fed_state_search + ', with the amount of chargers we found in that city.' '\\n')\n",
    "            state_charger_count = Counter(federal_state_results)\n",
    "            print(state_charger_count)\n",
    "            break\n",
    "        if count_of_cities_question in count_no:\n",
    "            break\n",
    "        if count_of_cities_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "                        \n",
    "\n",
    "\n",
    "                \n",
    "# federal_state_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "metadata": {},
   "outputs": [],
   "source": [
    "def federal_zip_search():\n",
    "    from pprint import pprint\n",
    "    fed_zip_search = ''\n",
    "    federal_zip_results = []\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    while True:\n",
    "        \n",
    "        fed_zip_search = input(\"In what zip code are you looking to find chargers? (Or enter '0' to exit.):\\t\")\n",
    "        \n",
    "        if fed_zip_search == '0':\n",
    "            break\n",
    "\n",
    "        with open('data/alt_fuel_stations (Oct 4 2020).csv', 'r') as file:\n",
    "            stuff = csv.reader(file)\n",
    "\n",
    "            \n",
    "            for line in stuff:\n",
    "                if fed_zip_search == line[6]:\n",
    "                    federal_zip_results.append(line[6])\n",
    "                    address = line[2], ', '+line[4]+', '+line[5]\n",
    "                    address_list.append(address)\n",
    "                    address_list.sort(key = lambda x: x[0]) \n",
    "                    \n",
    "            if len(federal_zip_results) > 0:\n",
    "                print('Here are the street addresses of the chargers we found in the zip code ' + fed_zip_search, '\\n')\n",
    "                pprint(address_list)\n",
    "                break\n",
    "                \n",
    "            if len(federal_zip_results) < 1:\n",
    "                print(\"We could not find any chargers in the \" + fed_zip_search + \" zip code. Please try again or enter '0' to exit.\")\n",
    "                \n",
    "                \n",
    "    while True:\n",
    "        if fed_zip_search == '0':\n",
    "            break\n",
    "        count_of_zip_question = input('Would you also like a count of chargers in that zip code?:\\t')\n",
    "        print('\\n')\n",
    "        if count_of_zip_question in count_yes:\n",
    "            print('Here is the count of chargers in the ' + fed_zip_search + ' zip code.' '\\n')\n",
    "            zip_charger_count = Counter(federal_zip_results)\n",
    "            print(zip_charger_count)\n",
    "            break\n",
    "        if count_of_zip_question in count_no:\n",
    "            break\n",
    "        if count_of_zip_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "            \n",
    "            \n",
    "            \n",
    "# federal_zip_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {},
   "outputs": [],
   "source": [
    "def federal_city_search():\n",
    "    from pprint import pprint\n",
    "    fed_city_search = ''\n",
    "    federal_city_results = []\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    \n",
    "    while True:\n",
    "    \n",
    "        fed_city_search = input(\"What city are you looking to find chargers in? (Or enter '0' to exit):\\t\")\n",
    "        \n",
    "        if fed_city_search == '0':\n",
    "            break\n",
    "    \n",
    "        fed_city_search = fed_city_search.title()\n",
    "    \n",
    "        with open('data/alt_fuel_stations (Oct 4 2020).csv', 'r') as file:\n",
    "            stuff = csv.reader(file)\n",
    "            \n",
    "\n",
    "        \n",
    "            for line in stuff:\n",
    "                if fed_city_search == line[4]:\n",
    "                    federal_city_results.append(line[4]+ ', '+ line[5])\n",
    "                    address = line[2], ', '+line[4]+', '+line[5]\n",
    "                    address_list.append(address)\n",
    "                    address_list.sort(key = lambda x: x[1]) \n",
    "                    \n",
    "            if len(federal_city_results) > 0:\n",
    "                print('\\n')\n",
    "                print('Here are the street addresses of the chargers we found in all of the cities with the name \"' + fed_city_search +'\", across the U.S. .' '\\n')\n",
    "                pprint(address_list)\n",
    "                break\n",
    "            if len(federal_city_results) < 1:\n",
    "                print(\"We could not find any chargers in \" + fed_city_search + \". Plese try again or enter '0' to exit.\")\n",
    "                \n",
    "                \n",
    "                \n",
    "    while True:\n",
    "        if fed_city_search == '0':\n",
    "            break\n",
    "            \n",
    "        count_of_zip_question = input('Would you also like a count of chargers in that city?:\\t')\n",
    "        print('\\n')\n",
    "        if count_of_zip_question in count_yes:\n",
    "            print('Here is the count of chargers in each of the cities with the name \"' + fed_city_search + ', across the U.S. .' '\\n')\n",
    "            zip_charger_count = Counter(federal_city_results)\n",
    "            print(zip_charger_count)\n",
    "            break\n",
    "        if count_of_zip_question in count_no:\n",
    "            break\n",
    "        if count_of_zip_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "        \n",
    "                \n",
    "            \n",
    "\n",
    "    \n",
    "    \n",
    "# federal_city_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {},
   "outputs": [],
   "source": [
    "def federal_db_search():\n",
    "    menu = {}\n",
    "    menu['1'] = '1 - Search by state'\n",
    "    menu['2'] = '2 - Search by city'\n",
    "    menu['3'] = '3 - Search by zip code'\n",
    "    menu['4'] = '4 - Exit'\n",
    "\n",
    "    while True:\n",
    "        options = menu.keys()\n",
    "        print('Menu/Options:')\n",
    "        print('-------------')\n",
    "        for entry in options:\n",
    "            print(menu[entry])\n",
    "            \n",
    "        selection = input('Which element would you like to search by?:\\t')\n",
    "        \n",
    "        if selection == '1':\n",
    "            federal_state_search()\n",
    "        if selection == '2':\n",
    "            federal_city_search()\n",
    "        if selection == '3':\n",
    "            federal_zip_search()\n",
    "        if selection == '4':\n",
    "            print('Exiting to main menu..........')\n",
    "            break\n",
    "        if selection not in menu.keys():\n",
    "            print('Invalid option. Please try enter a valid option.')\n",
    "\n",
    "# federal_db_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [],
   "source": [
    "def tesla_state_search():\n",
    "    from pprint import pprint\n",
    "    tesla_state_search = ''\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    tesla_results = []\n",
    "    state_abbreviations = [\n",
    "    'AL','al','AK','ak','AZ','az','AR','ar','CA','ca','CO','co',\n",
    "    'CT','ct','DE','de','FL','fl','GA','ga','HI','hi','ID','id',\n",
    "    'IL','il','IN','in','IA','ia','KS','ks','KY','ky','LA','la',\n",
    "    'ME','me','MD','md','MA','ma','MI','mi','MN','mn','MS','ma',\n",
    "    'MO','mo','MT','mt','NE','ne','NV','nv','NH','nh','NJ','nj',\n",
    "    'NM','nm','NY','ny','NC','nc','ND','nd','OH','oh','OK','ok',\n",
    "    'OR','or','PA','pa','RI','ri','SC','sc','SD','sd','TN','tn',\n",
    "    'TX','tx','UT','ut','VT','vt','VA','va','WA','wa','WV','wv',\n",
    "    'WI','wi','WY','wy']\n",
    "    while True:\n",
    "        tesla_state_search = input(\"What state are you looking to find Superchargers? (Or enter '0' to exit):\\t\")\n",
    "        if tesla_state_search == '0':\n",
    "            break\n",
    "        if tesla_state_search in state_abbreviations:\n",
    "            break\n",
    "        else:\n",
    "            print(\"Invalid input. Please enter the abbreviation of the state you are looking for. (Or enter '0' to exit)\")\n",
    "            tesla_state_search\n",
    "    \n",
    "    tesla_state_search = tesla_state_search.upper()\n",
    "    \n",
    "    while True:\n",
    "        if tesla_state_search == '0':\n",
    "            break\n",
    "\n",
    "        file =  csv.reader(open('data/tesla_superchargers.csv', 'r', encoding = 'utf-8'))\n",
    "        \n",
    "        print('Here are the cities we found with Superchargers in ' + tesla_state_search, '\\n') \n",
    "    \n",
    "        for line in file:\n",
    "            if tesla_state_search in line[2]:\n",
    "                tesla_results.append(line[2])\n",
    "                address = line[1], ', '+line[2]\n",
    "                address_list.append(address)\n",
    "                address_list.sort(key = lambda x: x[1])\n",
    "                \n",
    "        if len(tesla_results) > 0:\n",
    "            pprint(address_list)\n",
    "            break\n",
    "        if len(tesla_results) < 1:\n",
    "            print(\"We could not find any Superchargers in \"+tesla_state_search+ \". Please try again or enter '0' to exit.\")\n",
    "            \n",
    "            \n",
    "    while True:\n",
    "        if tesla_state_search == '0':\n",
    "            break\n",
    "        count_of_cities_question = input('Would you also like a count of chargers in this state?:\\t')\n",
    "        print('\\n')\n",
    "        \n",
    "        if count_of_cities_question in count_yes:\n",
    "            print('In ' + tesla_state_search + ', there are this many charging locations.' '\\n')\n",
    "            state_charger_count = Counter(tesla_results)\n",
    "            print(state_charger_count)\n",
    "            break\n",
    "        if count_of_cities_question in count_no:\n",
    "            break\n",
    "        if count_of_cities_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "        \n",
    "# tesla_state_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [],
   "source": [
    "def tesla_city_search():\n",
    "    from pprint import pprint\n",
    "    tesla_city_search = ''\n",
    "    tesla_results = []\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    \n",
    "    while True:\n",
    "        \n",
    "        tesla_city_search = input(\"What city are you looking to find Superchargers in? (Or enter '0' to exit):\\t\")\n",
    "        tesla_city_search = tesla_city_search.title()\n",
    "        \n",
    "        if tesla_city_search == '0':\n",
    "            break\n",
    "\n",
    "\n",
    "        file =  csv.reader(open('data/tesla_superchargers.csv', 'r', encoding = 'utf-8'))\n",
    "        \n",
    "    \n",
    "        for line in file:\n",
    "            if tesla_city_search in line[1]:\n",
    "                tesla_results.append(line[1])\n",
    "                address = line[4], line[2] + ', ' + line[1]\n",
    "                address_list.append(address)\n",
    "                address_list.sort(key = lambda x: x[1])\n",
    "                \n",
    "                \n",
    "        if len(tesla_results) > 0:\n",
    "            print('Here are the street addresses of the chargers we found in all of the cities with the name \"' + tesla_city_search + '\", across the U.S. .' '\\n')\n",
    "            pprint(address_list)\n",
    "            break\n",
    "            \n",
    "        if len(tesla_results) < 1:\n",
    "            print(\"We could not find any chargers in \"+ tesla_city_search+ \". Please try again or enter '0' to exit.\")\n",
    "                \n",
    "                \n",
    "                \n",
    "    while True:\n",
    "        if tesla_city_search == '0':\n",
    "            break\n",
    "            \n",
    "        count_of_city_question = input('Would you also like a count of chargers in \"'+ tesla_city_search +'\"?:\\t')\n",
    "        print('\\n')\n",
    "        if count_of_city_question in count_yes:\n",
    "            print('Here is the count of chargers in the city with name \"' + tesla_city_search, '\\n')\n",
    "            city_charger_count = Counter(tesla_results)\n",
    "            print(city_charger_count)\n",
    "            break\n",
    "        if count_of_city_question in count_no:\n",
    "            break\n",
    "        if count_of_city_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "            \n",
    "\n",
    "# tesla_city_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "metadata": {},
   "outputs": [],
   "source": [
    "def tesla_zip_search():\n",
    "    from pprint import pprint\n",
    "    tesla_zip_search = ''\n",
    "    tesla_zip_results = []\n",
    "    address_list = []\n",
    "    count_yes = ['Yes', 'yes', 'y', 'YES']\n",
    "    count_no = ['No', 'no', 'n', 'NO']\n",
    "    \n",
    "    while True:\n",
    "        \n",
    "        tesla_zip_search = input(\"In what zip code are you looking to find chargers? (Or enter '0' to exit.):\\t\")\n",
    "        \n",
    "        if tesla_zip_search == '0':\n",
    "            break\n",
    "            \n",
    "        file =  csv.reader(open('data/tesla_superchargers.csv', 'r', encoding = 'utf-8'))\n",
    "        \n",
    "        for line in file:\n",
    "            if tesla_zip_search in line[3]:\n",
    "                tesla_zip_results.append(line[3])\n",
    "                address = line[4] + ', ' + line[2] + ' ' + line[3]\n",
    "                address_list.append(address)\n",
    "                address_list.sort(key = lambda x: x[0])\n",
    "                \n",
    "        if len(tesla_zip_results) > 0:\n",
    "            print('Here are the street addresses of the chargers we found in the zip code ' + tesla_zip_search, '\\n')\n",
    "            pprint(address_list)\n",
    "            print('\\n')\n",
    "            break\n",
    "                \n",
    "        if len(tesla_zip_results) < 1:\n",
    "            print(\"We could not find any chargers in the \" + tesla_zip_search + \" zip code. Please try again or enter '0' to exit.\")\n",
    "            \n",
    "    while True:\n",
    "        if tesla_zip_search == '0':\n",
    "            break\n",
    "        count_of_zip_question = input('Would you also like a count of chargers in that zip code?:\\t')\n",
    "        if count_of_zip_question in count_yes:\n",
    "            print('Here is the count of chargers in the ' + tesla_zip_search + ' zip code.' '\\n')\n",
    "            zip_charger_count = Counter(tesla_zip_results)\n",
    "            print(zip_charger_count)\n",
    "            break\n",
    "        if count_of_zip_question in count_no:\n",
    "            break\n",
    "        if count_of_zip_question not in count_yes or count_no:\n",
    "            print('Invalid entry. Please try again.')\n",
    "                \n",
    "# print(tesla_zip_search())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {},
   "outputs": [],
   "source": [
    "def tesla_db_search():\n",
    "    import csv\n",
    "    menu = {}\n",
    "    menu['1'] = '1 - Search by state'\n",
    "    menu['2'] = '2 - Search by city'\n",
    "    menu['3'] = '3 - Search by zip code'\n",
    "    menu['4'] = '4 - Exit'\n",
    "\n",
    "    while True:\n",
    "        options = menu.keys()\n",
    "        print('Menu/Options:')\n",
    "        print('-------------')\n",
    "        for entry in options:\n",
    "            print(menu[entry])\n",
    "            \n",
    "        selection = input('Which element would you like to search by?:\\t')\n",
    "        \n",
    "        if selection == '1':\n",
    "            tesla_state_search()\n",
    "        if selection == '2':\n",
    "            tesla_city_search()\n",
    "        if selection == '3':\n",
    "            tesla_zip_search()\n",
    "        if selection == '4':\n",
    "            print('Exiting to main menu..........')\n",
    "            break\n",
    "        if selection not in menu.keys():\n",
    "            print('Invalid option. Please try enter a valid option.')\n",
    "\n",
    "\n",
    "# tesla_db_search()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from collections import *\n",
    "\n",
    "menu = {}\n",
    "menu['1'] = '1 - Search the Federal database'\n",
    "menu['2'] = '2 - Search the Tesla database'\n",
    "menu['3'] = '3 - Exit'\n",
    "\n",
    "while True:\n",
    "    options = menu.keys()\n",
    "    print('Menu/Options:')\n",
    "    print('-------------')\n",
    "    for entry in options:\n",
    "        print(menu[entry])\n",
    "\n",
    "    selection = input('Please select an option 1 - 3:\\t')\n",
    "\n",
    "    if selection == '1':\n",
    "        print('Selection 1')\n",
    "        federal_db_search()\n",
    "        \n",
    "    if selection == '2':\n",
    "        print('Selection 2')\n",
    "        tesla_db_search()\n",
    "    \n",
    "    if selection == '3':\n",
    "        print('Exiting program..........')\n",
    "        break\n",
    "    \n",
    "    if selection not in menu.keys():\n",
    "        print('Invalid option. Please try enter a valid option.')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.7.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}