--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/tweetcast/server/tweetcast.py Thu Oct 06 11:56:48 2011 +0200
@@ -0,0 +1,85 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import anyjson
+from twisted.enterprise import adbapi
+from twisted.internet import reactor, task
+from twisted.internet.protocol import Protocol, Factory
+from autobahn.websocket import WebSocketServerFactory, WebSocketServerProtocol
+
+connectstring = "dbname='tweet_live' user='postgres' host='localhost' password='doiteshimashite'"
+columns = [ 'id', 'created_at', 'text', 'user_id', 'screen_name' ]
+lastid = 0L
+
+dbpool = adbapi.ConnectionPool("psycopg2",connectstring)
+
+class TweetCast:
+
+ def __init__(self):
+ self.lastid = 0L
+ self.serverfactory = TweetcastServerFactory(tweetcast=self)
+ dbpool.runQuery("SELECT MAX(tweet_tweet.id) FROM tweet_tweet").addCallback(self.callbackInit)
+
+ def callbackInit(self, result):
+ self.lastid = result[0][0]
+ task.LoopingCall(self.scheduleTweets).start(1)
+
+ def scheduleTweets(self):
+ dbpool.runQuery("SELECT tweet_tweet.id, tweet_tweet.created_at, text, user_id, screen_name FROM tweet_tweet JOIN tweet_user ON tweet_tweet.user_id = tweet_user.id WHERE tweet_tweet.id > %d ORDER BY tweet_tweet.id ASC"%self.lastid).addCallback(self.callbackTweets)
+
+ def callbackTweets(self, result):
+ if result:
+ self.lastid = result[len(result)-1][0]
+ data = [dict((columns[i], str(ligne[i])) for i in range(len(columns))) for ligne in result]
+ else:
+ data = None
+ self.serverfactory.broadcast(anyjson.serialize(data))
+
+class TweetcastServerProtocol(WebSocketServerProtocol):
+
+ def onOpen(self):
+ self.factory.register(self)
+
+ def connectionLost(self, reason):
+ WebSocketServerProtocol.connectionLost(self, reason)
+ self.factory.unregister(self)
+ def onMessage(self, msg, binary):
+ print "Got message: " + msg
+
+class TweetcastServerFactory(WebSocketServerFactory):
+
+ protocol = TweetcastServerProtocol
+
+ def __init__(self, tweetcast=None):
+ WebSocketServerFactory.__init__(self)
+ self.clients = []
+# self.tweetcast = tweetcast
+
+ def register(self, client):
+ if not client in self.clients:
+ print "registered client " + client.peerstr
+ self.clients.append(client)
+
+ def unregister(self, client):
+ if client in self.clients:
+ print "unregistered client " + client.peerstr
+ self.clients.remove(client)
+
+ def broadcast(self, msg):
+ # print "broadcasting message '%s' .." % msg
+ for c in self.clients:
+ print "send to " + c.peerstr
+ c.sendMessage(msg)
+
+class FlashPolicySocketProtocol(Protocol):
+ def dataReceived(self, data):
+ self.transport.write("<?xml version=\"1.0\"?><!DOCTYPE cross-domain-policy SYSTEM \"/xml/dtds/cross-domain-policy.dtd\"><cross-domain-policy><site-control permitted-cross-domain-policies=\"master-only\"/><allow-access-from domain=\"*\" to-ports=\"*\" /></cross-domain-policy>\0")
+
+class FlashPolicyFactory(Factory):
+ def __init__(self):
+ self.protocol = FlashPolicySocketProtocol;
+
+tc = TweetCast()
+reactor.listenTCP(843, FlashPolicyFactory())
+reactor.listenTCP(9000, tc.serverfactory)
+reactor.run()
\ No newline at end of file