new_exercise_getter.py

# -*- coding: UTF-8 -*-
"""Gets updated version of exercises.

Checks for existing version and won't overwrite.
"""
import os
import requests

LOCAL = os.path.dirname(os.path.realpath(__file__))


def get_the_updates():
    """Decide if the other functions should download each file."""
    base = ("https://raw.githubusercontent.com/"
            "notionparallax/code1161base/master")
    new_files = [
        "/week2/exercise0.py",
        "/week4/IOexamples.py",
        "/week5/exercise1.py",
        "/week5/exercise2.py"
        ""
    ]

    for f in new_files:
        save_path = "./" + f
        if not os.path.isfile(save_path) and f is not "":
            url = base + f
            print(("downloading", url))
            download_and_save(url, save_path)
        elif f is "":
            pass  # do nothing, it's padding
        else:
            print(("You already have {f}".format(f=f)))
            print("If you really want to update that file, "
                  "delete it locally and rerun this script.")


def get_file_text(url):
    """Pull the raw file and return it as a string."""
    r = requests.get(url)
    return r.text.encode('utf-8')


def download_and_save(url, save_path):
    """Save a string as a file."""
    f = open(os.path.join(LOCAL, save_path), 'w')
    f.write(get_file_text(url))
    f.close()


get_the_updates()