initial commit

daemon.py

@@ -0,0 +1,65 @@
+import socket, time, urllib, urllib2, random, json, struct
+from datetime import datetime
+
+def get_datetime(tweet):
+    return datetime.strptime(tweet['created_at'][:-6],
+        '%a, %d %b %Y %H:%M:%S')
+
+class DalekTwitter(object):
+    api_url = 'http://search.twitter.com/search.json'
+    def __init__(self, options, *a, **kw):
+        self.options = options
+        self.last_tid = self.read_last_tid()
+        self.ua = 'adenozynotrojfosforan'
+    def read_last_tid(self):
+        try:
+            with open(self.options.last_id_file, 'r') as f:
+                return int(f.readline())
+        except (ValueError, IOError):
+            return 0
+    def write_last_tid(self, tid):
+        if tid:
+            with open(self.options.last_id_file, 'w') as f:
+                f.write(str(tid))
+    def rua(self):
+        l = list(self.ua)
+        random.shuffle(l)
+        self.ua = ''.join(l)
+    def poll_twitter(self):
+        args = urllib.urlencode({
+            'q': self.options.twitter_query,
+            'since_id': self.last_tid
+        })
+        self.rua()
+        req = urllib2.Request(self.api_url + '?' + args, None,
+            { 'User-Agent': self.ua })
+        r = urllib2.urlopen(req)
+        return json.load(r)
+    def dalek_write(self, m):
+        um = m.encode('utf-8')
+        s = socket.socket()
+        s.connect((self.options.host, self.options.port))
+        s.send('m' + struct.pack('!I', len(um)) + um)
+    def format_tweet(self, tweet):
+        return '\n%s %s: %s\n' % \
+            (get_datetime(tweet).strftime('%d/%m/%y %H:%M'), 
+                tweet['from_user'], tweet['text'])
+    def start(self):
+        try:
+            while True:
+                print 'refreshing'
+                reply = self.poll_twitter()
+                self.last_tid = reply['max_id']
+                print 'last id:', self.last_tid
+                self.write_last_tid(self.last_tid)
+                for tweet in reply['results']:
+                    will_write = self.format_tweet(tweet)
+                    print 'Will send:\n', will_write
+                    self.dalek_write(will_write)
+                time.sleep(self.options.twitter_refresh)
+        except KeyboardInterrupt:
+            self.write_last_tid(self.last_tid)
+
+if __name__ == '__main__':
+    import options
+    DalekTwitter(options).start()

options.py

@@ -0,0 +1,10 @@
+host = '10.8.0.205'
+port = 31337
+
+twitter_refresh = 5 #in seconds
+user_agent = 'proliant@hackerspace.pl'
+
+#refer to https://dev.twitter.com/docs/using-search
+twitter_query = '@hackerspacepl OR #hackerspacepl'
+
+last_id_file = 'lastid'