first real commit

Author: mirlan

app.yaml

@@ -0,0 +1,17 @@
+application: yammerfall
+version: 1-3
+runtime: python
+api_version: 1
+
+handlers:
+- url: /styles
+  static_dir: static/styles
+
+- url: /images
+  static_dir: static/images
+
+- url: /js
+  static_dir: static/js
+      
+- url: .*
+  script: main.py

index.yaml

@@ -0,0 +1,11 @@
+indexes:
+
+# AUTOGENERATED
+
+# This index.yaml is automatically updated whenever the dev_appserver
+# detects that a new type of query is run.  If you want to manage the
+# index.yaml file manually, remove the above marker line (the line
+# saying "# AUTOGENERATED").  If you want to manage some indexes
+# manually, move them above the marker line.  The index.yaml file is
+# automatically uploaded to the admin console when you next deploy
+# your application using appcfg.py.

main.py

@@ -0,0 +1,192 @@
+#!/usr/bin/env python
+
+import os
+import logging
+import urllib
+from time import time
+from random import getrandbits
+from google.appengine.api import urlfetch
+from google.appengine.ext.webapp import RequestHandler, WSGIApplication, template, wsgiref
+from django.utils import simplejson
+
+from yammerConfig import CONSUMER_KEY, CONSUMER_SECRET
+
+REQUEST_TOKEN_URL = 'https://www.yammer.com/oauth/request_token'
+ACCESS_TOKEN_URL = 'https://www.yammer.com/oauth/access_token'
+USER_AUTH_URL = 'https://www.yammer.com/oauth/authorize'
+DEFAULT_API_PREFIX = 'https://www.yammer.com/api/v1/'
+DEFAULT_API_SUFFIX = '.json'
+
+HEADERS = {
+	'Content-Type': 'application/x-www-form-urlencoded',
+	'Accept': '*/*',
+	'Accept-Language': 'en-us',
+	'Accept-Encoding': 'gzip, deflate',
+	'Connection': 'keep-alive',
+}
+
+def getOAuthHeaders(token=None, token_secret='', verifier=None):
+	OAuthHeaders = 'OAuth realm=""'
+	OAuthHeaders += ', oauth_consumer_key="%s"' % CONSUMER_KEY
+	
+	if token:
+		OAuthHeaders += ', oauth_token="%s"' % token
+	
+	OAuthHeaders += ', oauth_signature_method="PLAINTEXT"'
+	OAuthHeaders += ', oauth_signature="%s%%26%s"' % (CONSUMER_SECRET, token_secret)
+	currentTime = int(time())
+	OAuthHeaders += ', oauth_timestamp="%s"' % currentTime
+	OAuthHeaders += ', oauth_nonce="%s"' % currentTime
+	
+	if verifier:
+		OAuthHeaders += ', oauth_verifier="%s"' % verifier
+	
+	OAuthHeaders += ', oauth_version="1.0"'
+	return str(OAuthHeaders)
+
+class MainHandler(RequestHandler):
+
+	def get_cookie(self):
+		return self.request.cookies.get('oauth.yammerfall', '')
+	
+	def set_cookie(self, value, path='/'):
+		self.response.headers.add_header(
+			'Set-Cookie',
+			'%s=%s; path=%s; expires="Fri, 31-Dec-2021 23:59:59 GMT"' %
+			('oauth.yammerfall', value, path)
+			)
+	
+	def expire_cookie(self, path='/'):
+		self.response.headers.add_header(
+			'Set-Cookie',
+			'%s=; path=%s; expires="Fri, 31-Dec-1999 23:59:59 GMT"' %
+			('oauth.yammerfall', path)
+			)
+	
+	def get(self):
+		
+		if self.request.get('expire_cookie', None) == 'now':
+			logging.debug('EXPIRING COOKIE')
+			self.expire_cookie()
+			self.redirect('/')
+			return
+		
+		if self.get_cookie():
+			logging.debug('GOT COOKIE! %s' % self.get_cookie())
+			template_values = {
+				'messagesUrl': '/messages?%s' % self.get_cookie(),
+			}
+			path = os.path.join(os.path.dirname(__file__), "templates/home.html")
+			self.response.out.write(template.render(path, template_values))
+			return
+		
+		logging.debug('NO COOKIE!')
+		
+		HEADERS['Authorization'] = getOAuthHeaders()
+		result = urlfetch.fetch(
+			url=REQUEST_TOKEN_URL,
+			payload='',
+			method=urlfetch.POST,
+			headers=HEADERS,
+		)
+		
+		data = {}
+		for nv in result.content.split('&'):
+			(n, v) = nv.split('=')
+			data[n] = v
+		
+		template_values = {
+			'user_auth_url': USER_AUTH_URL,
+			'oauth_token': data['oauth_token'],
+			'oauth_token_secret': data['oauth_token_secret'],
+		}
+		path = os.path.join(os.path.dirname(__file__), "templates/login.html")
+		self.response.out.write(template.render(path, template_values))
+		return
+		
+		
+		
+	def post(self):
+		
+		logging.debug('AUTHORIZE WITH VERIFY CODE: %s' % self.request.get('oauth_verifier'))
+		
+		HEADERS['Authorization'] = getOAuthHeaders(
+			token = self.request.get('oauth_token'), 
+			token_secret = self.request.get('oauth_token_secret'),
+			verifier = self.request.get('oauth_verifier'))
+
+		result = urlfetch.fetch(
+			url=ACCESS_TOKEN_URL,
+			method=urlfetch.POST,
+			headers=HEADERS,
+		)
+		
+		logging.debug('AUTHORIZED: %s' % result.content)
+		
+		self.set_cookie(result.content)
+		self.redirect('/')
+		
+
+class MessagesHandler(RequestHandler):
+	def get(self):
+
+		HEADERS['Authorization'] = getOAuthHeaders(
+			token = self.request.get('oauth_token'), 
+			token_secret = self.request.get('oauth_token_secret'))
+		MSGS_URL = '%smessages%s' % (DEFAULT_API_PREFIX, DEFAULT_API_SUFFIX)
+		
+		result = urlfetch.fetch(
+			url=MSGS_URL,
+			method=urlfetch.GET,
+			headers=HEADERS,
+		)
+		
+		data = simplejson.loads(result.content)
+		
+		#self.response.out.write(data)
+		
+		users = {}
+		ref_msgs = {}
+		for r in data['references']:
+			if r['type'] == 'user':
+				users[r['id']] = r
+				users[r['name']] = r
+			elif r['type'] == 'message':
+				ref_msgs[r['id']] = r
+				
+		for m in data['messages']:
+			m['mugshot_url'] = users[m['sender_id']]['mugshot_url']
+			m['full_name'] = users[m['sender_id']]['full_name']
+			m['like_list'] = []
+			if m['liked_by']['count'] > 0:
+				for liked_by_name in m['liked_by']['names']:
+					try:
+						m['like_list'].append((liked_by_name['full_name'], users[liked_by_name['permalink']]['mugshot_url']))
+					except:
+						pass
+			if m['replied_to_id']:
+				in_reply_to = users[ref_msgs[m['replied_to_id']]['sender_id']]
+				m['in_reply_to'] = (in_reply_to['full_name'], in_reply_to['mugshot_url'])
+		
+		template_values = {
+			'messages' : data['messages'],
+		}
+		path = os.path.join(os.path.dirname(__file__), "templates/messagelist.html")
+		self.response.out.write(template.render(path, template_values))
+
+
+# ------------------------------------------------------------------------------
+# self runner -- gae cached main() function
+# ------------------------------------------------------------------------------
+
+def main():
+	logging.getLogger().setLevel(logging.DEBUG)
+	application = WSGIApplication([
+		('/', MainHandler),
+		('/messages', MessagesHandler),
+	   ], debug=True)
+	
+	wsgiref.handlers.CGIHandler().run(application)
+
+if __name__ == '__main__':
+	main()