def setUp(self): self.session = self.dbSessionMaker() self.config = {"startingBalance":100, "startingPrice":1, "historyLength":10} self.friendBizAPI = friendBizAPI(self.dbSessionMaker, self.config) self.fakeTwitter = fakeTwitterAPI() self.command = botCommands(self.fakeTwitter, self.friendBizAPI)
self.command._dispatch(event.command, event.params, event.sender) else: print("Not command") print(twitterData['text'] + "\n" + twitterData['user']['screen_name'] + ' : ' + twitterData['created_at']) return True def on_error(self, status): print(status) if __name__ == '__main__': config = {"startingBalance":100, "startingPrice":1, "historyLength":10, "botname": "friendbiz"} platform = "prod" auth = OAuthHandler(configuration.friendBizConfig.twitterAPICredentials['consumer_key'], configuration.friendBizConfig.twitterAPICredentials['consumer_secret']) auth.set_access_token(configuration.friendBizConfig.twitterAPICredentials['access_token'], configuration.friendBizConfig.twitterAPICredentials['access_token_secret']) twitterAPI = API(auth) engine = create_engine(configuration.friendBizConfig.dbConnectionString[platform], pool_recycle=3600) dbSessionMaker = sessionmaker(bind=engine) friendBizAPI = friendBizAPI(dbSessionMaker, config) l = StdOutListener(config, twitterAPI, friendBizAPI) stream = Stream(auth, l) while True: try: stream.userstream() except ReadTimeoutError as e: print(e)