From 2ddbe3776416e94898cdc38e8bd0b3894dba8ebd Mon Sep 17 00:00:00 2001
From: Roland Meertens <rolandmeertens@gmail.com>
Date: Mon, 18 Dec 2017 16:58:30 +0100
Subject: [PATCH] Input now downloads file from github

---
 ipynb/Advent 2017.ipynb | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/ipynb/Advent 2017.ipynb b/ipynb/Advent 2017.ipynb
index b6bec63..f16a8be 100644
--- a/ipynb/Advent 2017.ipynb	
+++ b/ipynb/Advent 2017.ipynb	
@@ -38,6 +38,9 @@
     "%matplotlib inline\n",
     "import matplotlib.pyplot as plt\n",
     "\n",
+    "import os\n",
+    "import urllib.request\n",
+    "\n",
     "import re\n",
     "import numpy as np\n",
     "import math\n",
@@ -65,7 +68,16 @@
     "\n",
     "def Input(day, year=2017):\n",
     "    \"Open this day's input file.\"\n",
-    "    return open('data/advent{}/input{}.txt'.format(year, day))\n",
+    "    directory = 'advent{}/'.format(year)\n",
+    "    filename = directory+'input{}.txt'.format(day)\n",
+    "    try:\n",
+    "        return open(filename)\n",
+    "    except FileNotFoundError:\n",
+    "        if not os.path.exists(directory):\n",
+    "            os.makedirs(directory)\n",
+    "\n",
+    "        urllib.request.urlretrieve(\"https://raw.githubusercontent.com/norvig/pytudes/master/data/\" + filename, filename)\n",
+    "        return Input(day)\n",
     "    \n",
     "def array(lines):\n",
     "    \"Parse an iterable of str lines into a 2-D array. If `lines` is a str, splitlines.\"\n",