#!/usr/local/bin/python2.6
# Quick upload to my flickr public set. Default permission set to private.
# Works on my Windows and UNIX machines.
# You need to change the pset value to your flickr set that you'd like to use.
# -Edward.C <e@os.sg>
 
import flickrapi
import sys

api_key = '728217d9f0ea18a46b4b464b1090b537'
api_secret = '5123df017887d689'
flickr = flickrapi.FlickrAPI(api_key, api_secret)
(token, frob) = flickr.get_token_part_one(perms='write')

if not token:
    raw_input("Press ENTER after you authorized this program on Flickr website.")
    flickr.get_token_part_two((token, frob))

def upload(file):	
	photo = file
	_photo = flickr.upload(filename=photo, is_public=u'1')
	pid = _photo.find('photoid').text
	pset = 72157627755596054
	flickr.photosets_addPhoto(photoset_id=pset, photo_id=pid)
	print('[*] %s uploaded successfully.' % sys.argv[1])
	
def main():
	if len(sys.argv) != 2:
		sys.exit('[*] Usage: %s <filename>' % sys.argv[0])		
	try:
		upload(sys.argv[1])
	except IOError:
		sys.exit('[*] Error: File not found.')
	except flickrapi.exceptions.FlickrError:
		sys.exit('[*] Error: Filetype was not recognised.')
	except KeyboardInterrupt:
		sys.exit('[*] User interrupted! Exiting....')			

if __name__ == '__main__':
    main()