Wallpaper script

wallpaper_script.py

from urllib import urlretrieve
import collections
import hashlib
import json
import os
import pickle
import urllib2

url = "http://www.reddit.com/r/wallpapers/.json?limit=100"
url_cache = '/home/brodul/.wallpaper_cache'
wallpaper_dir = '/home/brodul/.wallpapers/'


def get(url):
    """@todo: Docstring for get
    """
    req = urllib2.Request(url)
    response = urllib2.urlopen(req)
    the_page = response.read()
    return the_page

if os.path.isfile(url_cache):
    with open(url_cache) as f:
        feched_urls = pickle.load(f)
else:
    feched_urls = collections.deque([], 100)


resp = get(url)

data = json.loads(resp)
for elem in data['data']['children']:
    url = elem['data']['url']
    if url in feched_urls:
        continue
    exten = url.split(".")[-1]
    if exten not in ['jpg', 'jpeg', 'png']:
        continue
    hash_ = hashlib.sha224(url).hexdigest().replace('/', '')
    feched_urls.append(url)
    print url
    urlretrieve(
        url,
        os.path.join(wallpaper_dir, hash_ + '.' + exten)
    )


with open(url_cache, 'wb') as f:
    pickle.dump(feched_urls, f)