crawl and save.py

from danbooru import pages
from PIL import Image
import requests
from io import BytesIO


def get(size=(300, 300)) :
    ''':returns PIL.Image, 0~2 raiting level. 0 is safe, 1 is questionable, 2 is explicit.'''
    p = pages(limit=100)
    while True:
        post = next(p)
        if 'file_url' in post.keys():
            response = requests.get('https://danbooru.donmai.us{}'.format(post['file_url']))
            img = Image.open(BytesIO(response.content))
            img = img.resize(size, Image.ANTIALIAS)
            loc = '{}\\{}.png'.format('sqe'.index(post['rating']), post['id'])
            print('downloading {}'.format(loc))
            img.save(loc)

get()