app.py

# -*- coding: utf-8 -*-

import os
import sys
import requests
import requests.packages.urllib3
import json
import urllib
import warnings
import time
from tqdm import tqdm

# hide ssl warning
requests.packages.urllib3.disable_warnings()

# load config file
try:
	with open('config.json') as json_data_file:
		data = json.load(json_data_file)
except IOError as e:
	print "config.json not found!"
	exit()

# Parameters of your app
FACEBOOK_APP_ID     = data['FACEBOOK_APP_ID']
FACEBOOK_APP_SECRET = data['FACEBOOK_APP_SECRET']

#albumUrl = "https://www.facebook.com/1118241991550593/photos/?tab=album&album_id=1127894767251982"


try:
	albumUrl = str(sys.argv[1])
	albumId = albumUrl.split("album_id=",1)
	album = albumId[1]
except Exception as e:
	print "Album ID not found"
	exit()

def generateToken():
	# Trying to get an access token.
	oauth_args = dict(client_id     = FACEBOOK_APP_ID,
	                  client_secret = FACEBOOK_APP_SECRET,
	                  grant_type    = 'client_credentials')
	oauthURL = 'https://graph.facebook.com/oauth/access_token?' + urllib.urlencode(oauth_args)

	try:
		loadOauth = requests.get(oauthURL)
		parseOauth = json.loads(loadOauth.content)
		facebook_access_token = parseOauth['access_token']
	except KeyError:
	    print "Unable to grab an access token!"
	    exit()

	return facebook_access_token

def setAlbumName(albumId, facebook_token):
	# Trying get Album Name
	setUrlAlbum = "https://graph.facebook.com/v2.8/%s/?access_token=%s" % (albumId, facebook_token)
	getAlbumName = requests.get(setUrlAlbum)
	loadAlbumName = json.loads(getAlbumName.content)
	albumName = loadAlbumName['name']

	return albumName

def setAlbumUrl(albumId, facebook_token):
	setUrlAlbum = "https://graph.facebook.com/v2.8/%s/photos?access_token=%s&limit=100" % (albumId, facebook_token)
	getPaging = requests.get(setUrlAlbum)
	loadURL = json.loads(getPaging.content)
	albumURL = []
	nextPage = loadURL['paging']
	albumURL.append(setUrlAlbum)

	if "next" in nextPage:
		albumURL.append(loadURL['paging']['next'])

	while "next" in nextPage:
		getPaging = requests.get(setUrlAlbum)
		loadURL = json.loads(getPaging.content)
		nextPage = loadURL['paging']
		if "next" in nextPage:
			albumURL.append(loadURL['paging']['next'])
			setUrlAlbum = loadURL['paging']['next']

	return albumURL


def getPhotosId(setAlbumUrl):
	PhotosId = []
	for albumURL in setAlbumUrl:
		try:
			getPhotos = requests.get(albumURL)
			setIdPhotosJson = json.loads(getPhotos.content)

			for id in setIdPhotosJson['data']:
				idPhotos = id['id']
				PhotosId.append(idPhotos)
		except requests.exceptions.RequestException as e:  # This is the correct syntax
		    print e
		    sys.exit(1)

	return PhotosId

def getPhotosURL(getPhotosId, facebook_token):
	photosURL = []
	for photosId in getPhotosId:
		photoId = photosId
		setUrlPhoto = "https://graph.facebook.com/v2.8/%s/picture?access_token=%s" % (photoId, facebook_token)
		photosURL.append(setUrlPhoto)
	return photosURL

def downloadAlbum(albumId):
	facebook_token = generateToken()
	albumURL = setAlbumUrl(albumId, facebook_token)
	albumName = setAlbumName(albumId, facebook_token)

	print "Fetch data From Facebook..."
	print "Download Album %s" % (albumName)

	photoId = getPhotosId(albumURL)
	photoURL = getPhotosURL(photoId, facebook_token)

	print "Download %i photos" %(len(photoId))
	number = 0

	if not os.path.exists(albumName):
		print "Create Folder %s" %(albumName)
		os.makedirs(albumName)

	for image in photoURL:
		try:

			number += 1
			chekImageJPG = "%s/%s.jpg" %(albumName,number)
			chekImagePNG = "%s/%s.png" %(albumName,number)
			if os.path.exists(chekImageJPG) or os.path.exists(chekImagePNG):
				continue

			response = requests.get(image, stream=True)
			contentType = response.headers['Content-Type']

			if contentType.startswith('image/png'):
				imageType = "png"
			elif contentType.startswith('image/jpeg'):
				imageType = "jpg"

			nameImage = "%s/%s.%s" %(albumName,number,imageType)

			print "%s  downloading...."%(nameImage)
			# progress(number, len(photoId), status=nameImage)

			# Total size in bytes.
			total_size = int(response.headers.get('content-length', 0));
			with open(nameImage, 'wb') as handle:
			        if not response.ok:
			            print response

			        for block in tqdm(response.iter_content(), total=total_size, unit='B', unit_scale=True):

			            if not block:
			                break

			            handle.write(block)

		except requests.exceptions.RequestException as e:  # This is the correct syntax
			print e
			os.remove(nameImage)
			sys.exit(1)

	sys.stdout.write("Finish Download %i photos" %(len(photoId)))
	sys.stdout.write("\n")
	return "finish"

if __name__ == "__main__":

	try:
		downloadAlbum(album)
	except requests.exceptions.HTTPError as err:
		print err
		exit()
	except KeyboardInterrupt:
		print "Good Bye~"
		pass