weckbot/getter.py

68 lines
1.7 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import HTMLParser, tweepy, os, sqlite3, datetime, dateutil.parser, time
2015-03-19 17:16:17 +00:00
user = "User Name (without @)"
cke = "Consumer Key"
cse = "Consumer Secret"
ato = "Access Token"
ase = "Access Secret"
database_filename = "Database.db"
search = "@" + user
auth = tweepy.OAuthHandler(cke, cse)
auth.set_access_token(ato, ase)
api = tweepy.API(auth)
sql_conn = sqlite3.connect(database_filename)
cur = sql_conn.cursor()
try:
cur.execute("SELECT max(tweet_id) FROM tweets")
savepoint = int(cur.fetchone()[0])
except:
savepoint = ""
print "No savepoint found. Trying to get as many results as possible."
while True:
timeline = tweepy.Cursor(api.search, q=search, since_id=savepoint).items()
for status in timeline:
2015-03-19 17:16:17 +00:00
text = status.text.encode("UTF-8")
sender = status.user.screen_name.encode("UTF-8")
twid = int(status.id)
2015-03-19 17:16:17 +00:00
if "storn" in text.lower():
try:
api.update_status("@%s Okay, dein Wecktweet wurde storniert." % sender, twid)
cur.execute("DELETE FROM tweets WHERE sender = %s;" % sender)
sql_conn.commit()
continue
except:
continue
try:
2015-03-19 17:16:17 +00:00
date = dateutil.parser.parse(text,dayfirst=True,fuzzy=True)
except ValueError, e:
print e
try:
api.update_status("@%s Sorry, ich verstehe deinen Tweet nicht... :(" % sender, twid)
except Exception, f:
print f
continue
2015-03-19 17:16:17 +00:00
if date < datetime.datetime.now():
if date.date() == datetime.datetime.now().date():
date += datetime.timedelta(days = 1)
else:
api.update_status("@%s Das war doch schon...?" % sender, twid)
continue
cur.execute("INSERT INTO tweets VALUES(%i,'%s','%s','',0)" % (twid,date.strftime("%Y-%m-%dT%H:%M:%S"),sender))
2015-03-18 21:50:03 +00:00
sql_conn.commit()
savepoint = twid
time.sleep(60)