kumi/twitools
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Move follower functions to twitools

Browse source 
Create tables for storing followers and following in setup.py
This commit is contained in:
Klaus-Uwe Mitterer 2016-08-01 16:39:55 +02:00
parent 6ffa8e052d
commit b39b708270
2 changed files with 10 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
followertxt.py
View file

@ -3,16 +3,6 @@
import twitools, setuptools import twitools, setuptools
import os, time, tweepy import os, time, tweepy
def getFollowerIDs(two=twitools.twObject()):
''' Returns 5,000 follower IDs at most '''
return two.api.followers_ids(screen_name=twitools.twObject().whoami())
def getNamesByIDs(fids=getFollowerIDs(), two=twitools.twObject()):
for page in setuptools.paginate(fids, 100):
followers = two.api.lookup_users(user_ids=page)
for follower in followers:
yield follower.screen_name
def getOutDir(dirname="followers"): def getOutDir(dirname="followers"):
if not os.path.isdir(dirname): if not os.path.isdir(dirname):
os.mkdir(dirname) os.mkdir(dirname)
@ -23,7 +13,7 @@ def getOutFile(dirname="followers"):
def writeOutFile(outfile=getOutFile()): def writeOutFile(outfile=getOutFile()):
with open(getOutFile(), 'a') as f: with open(getOutFile(), 'a') as f:
for follower in getNamesByIDs(getFollowerIDs()): for follower in twitools.getNamesByIDs(twitools.getFollowerIDs()):
f.write(follower + "\n") f.write(follower + "\n")
if __name__ == "__main__": if __name__ == "__main__":

9
twitools/__init__.py
View file

@ -18,3 +18,12 @@ class twObject:
def whoami(self): def whoami(self):
return self.auth.get_username() return self.auth.get_username()
def getFollowerIDs(two=twObject()):
''' Returns 5,000 follower IDs at most '''
return two.api.followers_ids(screen_name=twObject().whoami())
def getNamesByIDs(fids=getFollowerIDs(), two=twObject()):
for page in setuptools.paginate(fids, 100):
followers = two.api.lookup_users(user_ids=page)
for follower in followers:
yield follower.screen_name