build: add setup.py to fetch data set dependencies

.gitignore

@@ -1,2 +1,3 @@
 venv/
 __pycache__
+data/

scripts/setup.py

@@ -0,0 +1,44 @@
+from core import Parser
+
+import os
+import requests
+
+# Files to download into the data directory.
+FILES = [
+    "https://raw.githubusercontent.com/kipr/opencv/master/data/haarcascades/haarcascade_frontalface_default.xml",
+    "https://raw.githubusercontent.com/kipr/opencv/master/data/haarcascades/haarcascade_eye_tree_eyeglasses.xml",
+]
+
+def setup(dataDir):
+    """Downloads the required files into the data directory."""
+
+    # Create the data directory if it doesn't exist.
+    if not os.path.exists(dataDir):
+        os.mkdir(dataDir)
+
+    for file in FILES:
+        # Send a GET request to the file URL.
+        print("Downloading file {}...".format(file))
+        response = requests.get(file)
+        if response.status_code != 200:
+            print("Failed to download file {}, error {}".format(file, response.status_code))
+            continue
+
+        # Write the file to the data directory.
+        filename = os.path.basename(file)
+        path = os.path.join(dataDir, filename)
+        open(path, "wb").write(response.content)
+
+if __name__ == "__main__":
+    # Default data directory is `../data` relative to this file.
+    defaultDataDir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..", "data")
+
+    parser = Parser()
+    parser.add("help", "Prints this help message.", default=False)
+    parser.add("data", "Path to the data directory.", default=defaultDataDir)
+    options = parser.parse()
+
+    if options["help"]:
+        print(parser.help())
+    else:
+        setup(options["data"])