127 lines
4.7 KiB
Python
Executable file
127 lines
4.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import argparse, dbtools, setuptools, time, twitools
|
|
|
|
def getTweets(db=dbtools.dbHelper(), user=twitools.twObject().whoami(), two=twitools.twObject()):
|
|
query = "from:" + user
|
|
savepoint = db.getLatestTweet() + 1
|
|
last = savepoint
|
|
|
|
timeline = two.search(query, savepoint)
|
|
|
|
tw_counter = 0
|
|
|
|
for status in timeline:
|
|
timestamp = status.created_at.strftime('%Y-%m-%d %H:%M:%S') + " +0000"
|
|
text = setuptools.unescapeText(status.text)
|
|
|
|
try:
|
|
db.executeQuery("INSERT INTO tweets(tweet_id,timestamp,text) VALUES(" + str(status.id) + ",'" + timestamp + "','" + text + "')")
|
|
except:
|
|
pass
|
|
db.commit()
|
|
|
|
last = status.id
|
|
tw_counter = tw_counter + 1
|
|
|
|
return tw_counter, last, savepoint
|
|
|
|
def getMessages(db=dbtools.dbHelper(), two=twitools.twObject()):
|
|
mcount = 0
|
|
savepoint = db.getLatestMessage() + 1
|
|
new_messages = two.api.direct_messages(since_id=savepoint, count=200, full_text=True)
|
|
new_out_messages = two.api.sent_direct_messages(since_id=savepoint, count=200, full_text=True)
|
|
|
|
for m in new_messages:
|
|
try:
|
|
db.executeQuery("INSERT INTO messages VALUES(%s, '%s', %s, %s, '%s')" % (m.id, setuptools.unescapeText(m.text), m.sender_id, m.recipient_id, m.created_at))
|
|
mcount += 1
|
|
except:
|
|
pass
|
|
|
|
for m in new_out_messages:
|
|
try:
|
|
db.executeQuery("INSERT INTO messages VALUES(%s, '%s', %s, %s, '%s')" % (m.id, setuptools.unescapeText(m.text), m.sender_id, m.recipient_id, m.created_at))
|
|
mcount += 1
|
|
except:
|
|
pass
|
|
|
|
db.commit()
|
|
|
|
return mcount, savepoint or 0, db.getLatestMessage()
|
|
|
|
def getFollowers(db=dbtools.dbHelper(), two=twitools.twObject(), firstrun=False):
|
|
""" Get handles of users we are following. :param db: Database object to be used. :param two: Twitter object to be used. :param firstrun: Must be set to True if the function is executed for the first time. Defaults to False. :return: Returns the number of gained and lost followings in a list (gained, lost). """
|
|
|
|
current = list(db.getFollowers())
|
|
new = list(twitools.getNamesByIDs(twitools.getFollowerIDs()))
|
|
gained = 0
|
|
lost = 0
|
|
|
|
if (len(current) == 0 or len(new) == 0) and not firstrun:
|
|
print("Something went wrong.")
|
|
return 0,0
|
|
|
|
for follower in new:
|
|
if follower not in current:
|
|
db.executeQuery("INSERT INTO followers VALUES('%s', %i, 0)" % (follower, int(time.time())))
|
|
print("New follower: %s" % follower)
|
|
gained += 1
|
|
|
|
for follower in current:
|
|
if follower not in new:
|
|
db.executeQuery("UPDATE followers SET `until` = %i WHERE `id` = '%s' AND `until` = 0" % (int(time.time()), follower))
|
|
print("Lost follower: %s" % follower)
|
|
lost += 1
|
|
|
|
db.commit()
|
|
|
|
return gained, lost
|
|
|
|
def getFollowing(db=dbtools.dbHelper(), two=twitools.twObject(), firstrun=False):
|
|
"""
|
|
Get handles of users we are following.
|
|
|
|
:param db: Database object to be used.
|
|
:param two: Twitter object to be used.
|
|
:param firstrun: Must be set to True if the function is executed for the first time. Defaults to False.
|
|
:return: Returns the number of gained and lost followings in a list (gained, lost).
|
|
"""
|
|
current = list(db.getFollowing())
|
|
new = list(twitools.getNamesByIDs(twitools.getFollowingIDs()))
|
|
gained = 0
|
|
lost = 0
|
|
|
|
if (len(current) == 0 or len(new) == 0) and not firstrun:
|
|
print("Something went wrong.")
|
|
return 0,0
|
|
|
|
for following in new:
|
|
if following not in current:
|
|
db.executeQuery("INSERT INTO following VALUES('%s', %i, 0)" % (following, int(time.time())))
|
|
print("You started following: %s" % following)
|
|
gained += 1
|
|
|
|
for following in current:
|
|
if following not in new:
|
|
db.executeQuery("UPDATE following SET `until` = %i WHERE `id` = '%s' AND `until` = 0" % (int(time.time()), following))
|
|
print("You no longer follow: %s" % following)
|
|
lost += 1
|
|
|
|
db.commit()
|
|
|
|
return gained, lost
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-f", "--first", help="first run: ignore empty databases", action="store_true")
|
|
args = parser.parse_args()
|
|
db = dbtools.dbHelper()
|
|
count, last, first = getTweets(db)
|
|
print("Stored %i tweets." % count)
|
|
count, last, first = getMessages(db)
|
|
print("Stored %i messages." % count)
|
|
gained, lost = getFollowers(db, firstrun=args.first)
|
|
print("Gained %i followers, lost %i." % (gained, lost))
|
|
gained, lost = getFollowing(db, firstrun=args.first)
|
|
print("Started following %i, stopped following %i." % (gained, lost))
|