Skip to content

Instantly share code, notes, and snippets.

@dynamicguy
Forked from pcardune/.gitignore
Created December 17, 2011 13:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dynamicguy/1490202 to your computer and use it in GitHub Desktop.
Save dynamicguy/1490202 to your computer and use it in GitHub Desktop.
Facebook command-line client helper
.fb_access_token
.fbconsole.py
#!/usr/bin/env python
import os.path
import os
import stat
import json
import urllib2
import BaseHTTPServer
import webbrowser
import httplib
import mimetools
import mimetypes
import cookielib
import types
from urlparse import urlparse, parse_qs
from urllib import urlencode
from pprint import pprint
APP_ID = '179745182062082'
SERVER_PORT = 8080
REDIRECT_URI = 'http://127.0.0.1:%s/' % SERVER_PORT
ACCESS_TOKEN = None
LOCAL_FILE = '.fb_access_token'
AUTH_SCOPE = []
__all__ = [
'help',
'authenticate',
'graph',
'graph_post',
'graph_delete',
'shell',
'fql',
'APP_ID',
'SERVER_PORT',
'ACCESS_TOKEN',
'AUTH_SCOPE',
'LOCAL_FILE']
def _get_url(path, args=None, graph=True):
args = args or {}
if ACCESS_TOKEN:
args['access_token'] = ACCESS_TOKEN
subdomain = 'graph' if graph else 'api'
if 'access_token' in args or 'client_secret' in args:
endpoint = "https://%s.facebook.com" % subdomain
else:
endpoint = "http://%s.facebook.com" % subdomain
return endpoint+str(path)+'?'+urlencode(args)
class _MultipartPostHandler(urllib2.BaseHandler):
handler_order = urllib2.HTTPHandler.handler_order - 10 # needs to run first
def http_request(self, request):
data = request.get_data()
if data is not None and not isinstance(data, types.StringTypes):
files = []
params = []
try:
for key, value in data.items():
if isinstance(value, types.FileType):
files.append((key, value))
else:
params.append((key, value))
except TypeError:
raise TypeError("not a valid non-string sequence or mapping object")
if len(files) == 0:
data = urlencode(params)
else:
boundary, data = self.multipart_encode(params, files)
contenttype = 'multipart/form-data; boundary=%s' % boundary
request.add_unredirected_header('Content-Type', contenttype)
request.add_data(data)
return request
https_request = http_request
def multipart_encode(self, params, files, boundary=None, buffer=None):
boundary = boundary or mimetools.choose_boundary()
buffer = buffer or ''
for key, value in params:
buffer += '--%s\r\n' % boundary
buffer += 'Content-Disposition: form-data; name="%s"' % key
buffer += '\r\n\r\n' + value + '\r\n'
for key, fd in files:
file_size = os.fstat(fd.fileno())[stat.ST_SIZE]
filename = fd.name.split('/')[-1]
contenttype = mimetypes.guess_type(filename)[0] or 'application/octet-stream'
buffer += '--%s\r\n' % boundary
buffer += 'Content-Disposition: form-data; '
buffer += 'name="%s"; filename="%s"\r\n' % (key, filename)
buffer += 'Content-Type: %s\r\n' % contenttype
fd.seek(0)
buffer += '\r\n' + fd.read() + '\r\n'
buffer += '--%s--\r\n\r\n' % boundary
return boundary, buffer
class _RequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
def do_GET(self):
global ACCESS_TOKEN
self.send_response(200)
self.send_header("Content-type", "text/html")
self.end_headers()
params = parse_qs(urlparse(self.path).query)
ACCESS_TOKEN = params.get('access_token', [None])[0]
if ACCESS_TOKEN:
data = {'scope': AUTH_SCOPE,
'access_token': ACCESS_TOKEN}
open(LOCAL_FILE,'w').write(json.dumps(data))
self.wfile.write("You have successfully logged in to facebook with fbconsole. "
"You can close this window now.")
else:
self.wfile.write('<html><head>'
'<script>location = "?"+location.hash.slice(1);</script>'
'</head></html>')
def help():
"""Print out some helpful information"""
print '''
The following commands are available:
help() - display this help message
authenticate() - authenticate with facebook. Optionally provide list
of permissions to request
graph(path, params) - call the graph api with the given path and query parameters
graph_post(path, data) - post data to the graph api with the given path
graph_delete(path, params) - send a delete request
fql(query) - make an fql request
'''
def authenticate():
"""Authenticate with facebook so you can make api calls that require auth.
Alternatively you can just set the ACCESS_TOKEN global variable in this
module to an access token you get from facebook.
If you want to request certain permissions, set the AUTH_SCOPE global
variable to the list of permissions you want.
"""
global ACCESS_TOKEN
needs_auth = True
if os.path.exists(LOCAL_FILE):
data = json.loads(open(LOCAL_FILE).read())
if set(data['scope']).issuperset(AUTH_SCOPE):
ACCESS_TOKEN = data['access_token']
needs_auth = False
if needs_auth:
print "Logging you in to facebook..."
webbrowser.open('https://www.facebook.com/dialog/oauth?' +
urlencode({'client_id':APP_ID,
'redirect_uri':REDIRECT_URI,
'response_type':'token',
'scope':','.join(AUTH_SCOPE)}))
httpd = BaseHTTPServer.HTTPServer(('127.0.0.1', SERVER_PORT), _RequestHandler)
while ACCESS_TOKEN is None:
httpd.handle_request()
def graph(path, params=None):
"""Send a GET request to the graph api.
For example:
>>> graph('/me')
>>> graph('/me', {'fields':'id,name'})
"""
return json.load(urllib2.urlopen(_get_url(path, args=params)))
def graph_post(path, params=None):
"""Send a POST request to the graph api.
You can also upload files using this function. For example:
>>> graph_post('/me/photos',
... {'name': 'My Photo',
... 'source': open("myphoto.jpg")})
"""
opener = urllib2.build_opener(
urllib2.HTTPCookieProcessor(cookielib.CookieJar()),
_MultipartPostHandler)
return json.load(opener.open(_get_url(path), params))
def graph_delete(path, params=None):
"""Send a DELETE request to the graph api.
For example:
>>> msg_id = graph_post('/me/feed', {'message':'hello world'})['id']
>>> graph_delete('/'+msg_id)
"""
if not params:
params = {}
params['method'] = 'delete'
return graph_post(path, params)
def fql(query):
"""Make an fql request.
For example:
>>> fql('SELECT name FROM user WHERE uid = me()')
"""
url = _get_url('/method/fql.query',
args={'query': query, 'format': 'json'},
graph=False)
return json.load(urllib2.urlopen(url))
INTRO_MESSAGE = '''\
__ _ _
/ _| | | |
| |_| |__ ___ ___ _ __ ___ ___ | | ___
| _| '_ \ / __|/ _ \| '_ \/ __|/ _ \| |/ _ \\
| | | |_) | (__| (_) | | | \__ \ (_) | | __/
|_| |_.__/ \___|\___/|_| |_|___/\___/|_|\___|
Type help() for a list of commands.
quick start:
>>> authenticate()
>>> print "Hello", graph('/me')['name']
'''
def shell():
try:
from IPython.Shell import IPShellEmbed
IPShellEmbed()(INTRO_MESSAGE)
except ImportError:
import code
code.InteractiveConsole(globals()).interact(INTRO_MESSAGE)
if __name__ == '__main__':
shell()
#!/usr/bin/env python
from urllib import urlretrieve
import imp
urlretrieve('https://raw.github.com/gist/1194123/fbconsole.py', '.fbconsole.py')
fb = imp.load_source('fb', '.fbconsole.py')
# Your script goes below here
fb.AUTH_SCOPE = ['publish_stream']
fb.authenticate()
print "Hello", fb.graph('/me')['name']
status = raw_input("Update your status: ")
fb.graph_post('/me/feed', {'message': status})
print "your status has been updated"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment