Esempio n. 1
0
def main():
    conf = get_conf()
    twitter = Twitter(conf)

    parser = argparse.ArgumentParser()

    parser.add_argument('-q', '--query', type = str,
        help = "Gets all tweets and retweets for a specifc search query (realtime)")
    parser.add_argument('-t', '--timeline', action = "store_true",
        help = "Gets all tweets for the authenticated user")
    parser.add_argument('-u', '--user', type = str,
        help = "Get a timeline with a username")
    parser.add_argument('-s', '--search', type = str,
        help = "Get results for a search query (not realtime)")
    parser.add_argument('-l', '--lookup',
        help = "Get all the details for a single tweet")
    parser.add_argument('-rts', '--retweetstats',
        help = "Returns user info of up to 100 people that retweeted the tweet specified")

    args = parser.parse_args()

    if args.query:
        twitter.query(args.query)
    elif args.timeline:
        twitter.run_timeline()
    elif args.user:
        twitter.user_timeline()
    elif args.search:
        twitter.search(args.search)
    elif args.lookup:
        twitter.lookup(args.lookup)
    elif args.retweetstats:
        sid = args.retweetstats
        print("Collecting retweeters for %s" % sid)
        users = twitter.retweetstats(sid)
        path = "%s-retweeters.json" % sid

        with open(path, "w") as f:
            f.write(json.dumps(users))

        print("Written retweeters to %s" % path)
    else:
        parser.print_help()