u1.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1: VER, Current points: 284, Theoretical max points: 500, Can win: Yes\n",
      "2: PER, Current points: 191, Theoretical max points: 407, Can win: Yes\n",
      "3: LEC, Current points: 186, Theoretical max points: 402, Can win: Yes\n",
      "4: SAI, Current points: 171, Theoretical max points: 387, Can win: Yes\n",
      "5: RUS, Current points: 170, Theoretical max points: 386, Can win: Yes\n",
      "6: HAM, Current points: 146, Theoretical max points: 362, Can win: Yes\n",
      "7: NOR, Current points: 76, Theoretical max points: 292, Can win: Yes\n",
      "8: OCO, Current points: 64, Theoretical max points: 280, Can win: No\n",
      "9: ALO, Current points: 51, Theoretical max points: 267, Can win: No\n",
      "10: BOT, Current points: 46, Theoretical max points: 262, Can win: No\n",
      "11: MAG, Current points: 22, Theoretical max points: 238, Can win: No\n",
      "12: VET, Current points: 20, Theoretical max points: 236, Can win: No\n",
      "13: RIC, Current points: 19, Theoretical max points: 235, Can win: No\n",
      "14: GAS, Current points: 18, Theoretical max points: 234, Can win: No\n",
      "15: MSC, Current points: 12, Theoretical max points: 228, Can win: No\n",
      "16: TSU, Current points: 11, Theoretical max points: 227, Can win: No\n",
      "17: ZHO, Current points: 5, Theoretical max points: 221, Can win: No\n",
      "18: ALB, Current points: 4, Theoretical max points: 220, Can win: No\n",
      "19: STR, Current points: 4, Theoretical max points: 220, Can win: No\n",
      "20: LAT, Current points: 0, Theoretical max points: 216, Can win: No\n",
      "21: HUL, Current points: 0, Theoretical max points: 216, Can win: No\n"
     ]
    }
   ],
   "source": [
    "\"\"\"Who can still win the drivers WDC?\n",
    "======================================\n",
    "Calculates which drivers still has chance to win the WDC.\n",
    "Simplified since it doesn't compare positions if points are equal.\n",
    "This example implements 3 functions that it then uses to calculate\n",
    "it's result.\n",
    "\"\"\"\n",
    "\n",
    "import requests\n",
    "import fastf1\n",
    "\n",
    "fastf1.Cache.enable_cache(\"./cache\")  # replace with your cache directory\n",
    "\n",
    "\n",
    "##############################################################################\n",
    "# We need a function to get the current driver standings from\n",
    "# Ergast and returns as list of drivers.\n",
    "# Reference https://ergast.com/mrd/methods/standings/\n",
    "def get_drivers_standings():\n",
    "    url = \"https://ergast.com/api/f1/current/driverStandings.json\"\n",
    "    response = requests.get(url)\n",
    "    data = response.json()\n",
    "    drivers_standings = data['MRData']['StandingsTable']['StandingsLists'][0]['DriverStandings']  # noqa: E501\n",
    "    return drivers_standings\n",
    "\n",
    "\n",
    "##############################################################################\n",
    "# We need a function to calculates the maximum amount of points possible if a\n",
    "# driver wins everything left of the season.\n",
    "# https://en.wikipedia.org/wiki/List_of_Formula_One_World_Championship_points_scoring_systems\n",
    "def calculate_max_points_for_remaining_season():\n",
    "    POINTS_FOR_SPRINT = 8 + 25 + 1  # Winning the sprint, race and fastest lap\n",
    "    POINTS_FOR_CONVENTIONAL = 25 + 1  # Winning the race and fastest lap\n",
    "\n",
    "    events = fastf1.events.get_events_remaining()\n",
    "    # Count how many sprints and conventional races are left\n",
    "    sprint_events = \\\n",
    "        len(events.loc[events[\"EventFormat\"] == \"sprint\"])\n",
    "    conventional_events = \\\n",
    "        len(events.loc[events[\"EventFormat\"] == \"conventional\"])\n",
    "\n",
    "    # Calculate points for each\n",
    "    sprint_points = sprint_events * POINTS_FOR_SPRINT\n",
    "    conventional_points = conventional_events * POINTS_FOR_CONVENTIONAL\n",
    "\n",
    "    return sprint_points + conventional_points\n",
    "\n",
    "\n",
    "##############################################################################\n",
    "# For each driver we will see if there is a chance to get more points than\n",
    "# the current leader. We assume the leader gets no more points and the\n",
    "# driver gets the theoretical maximum amount of points.\n",
    "#\n",
    "# We currently don't consider the case of two drivers getting equal points\n",
    "# since its more complicated and would require comparing positions.\n",
    "def calculate_who_can_win(driver_standings, max_points):\n",
    "    LEADER_POINTS = int(driver_standings[0]['points'])\n",
    "\n",
    "    for _, driver in enumerate(driver_standings):\n",
    "        driver_max_points = int(driver[\"points\"]) + max_points\n",
    "        can_win = 'No' if driver_max_points < LEADER_POINTS else 'Yes'\n",
    "\n",
    "        print(f\"{driver['position']}: \\\n",
    "{driver['Driver']['code']}, \\\n",
    "Current points: {driver['points']}, \\\n",
    "Theoretical max points: {driver_max_points}, \\\n",
    "Can win: {can_win}\")\n",
    "\n",
    "\n",
    "##############################################################################\n",
    "# Now using the 3 functions above we can use them to calculate who\n",
    "# can still win.\n",
    "\n",
    "# Get the current drivers standings\n",
    "driver_standings = get_drivers_standings()\n",
    "\n",
    "# Get the maximum amount of points\n",
    "points = calculate_max_points_for_remaining_season()\n",
    "\n",
    "# Print which drivers can still win\n",
    "calculate_who_can_win(driver_standings, points)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3.9.13 64-bit",
   "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.9.13"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "b0fa6594d8f4cbf19f97940f81e996739fb7646882a419484c72d19e05852a7e"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}