From 5a35e4a9baa753e5847b4d9333506713c2f5b304 Mon Sep 17 00:00:00 2001
From: Robbie Trencheny <me@robbiet.us>
Date: Sat, 26 Mar 2016 18:52:03 -0700
Subject: [PATCH] Data source validation

---
 homeassistant/components/sensor/gtfs.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/sensor/gtfs.py b/homeassistant/components/sensor/gtfs.py
index bccf0ef2abb..9e198e7ef11 100644
--- a/homeassistant/components/sensor/gtfs.py
+++ b/homeassistant/components/sensor/gtfs.py
@@ -142,8 +142,17 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
         _LOGGER.error("Data must be set in the GTFS configuration!")
         return False
 
+    gtfs_dir = hass.config.path("gtfs")
+
+    if not os.path.exists(gtfs_dir):
+      os.makedirs(gtfs_dir)
+
+    if not os.path.exists(os.path.join(gtfs_dir, config["data"])):
+      _LOGGER.error("The given GTFS data file/folder was not found!")
+      return False
+
     dev = []
-    dev.append(GTFSDepartureSensor(config["data"], hass.config.path("gtfs"),
+    dev.append(GTFSDepartureSensor(config["data"], gtfs_dir,
                                    config["origin"], config["destination"]))
     add_devices(dev)