Beispiel #1
0
def main():
    if ( len(sys.argv) < 2 ):
        usage()
        exit(1)
        
    config = loadConfig(sys.argv[1])
    initLogging(config)
    logger = logging.getLogger(__name__)
    logger.info("Execution du script avec le fichier de configuration %s",sys.argv[1])
    
    twit = TwitterManager(
                          config.get(SECTION_TWITTER, OPT_CONSUMER_KEY),
                          config.get(SECTION_TWITTER, OPT_CONSUMER_SECRET),
                          config.get(SECTION_TWITTER, OPT_OAUTH_TOKEN),
                          config.get(SECTION_TWITTER, OPT_OAUTH_TOKEN_SECRET))

    tweets = twit.getTweets(config.getint(SECTION_QUERY, OPT_MAX_COUNT),
                            config.getint(SECTION_QUERY, OPT_MAX_RANGE))
    
    logger.debug(json.dumps(tweets, indent=1))
    rdb = RethinkdbManager(config.get(SECTION_RETHINKDB, OPT_HOST),
                           config.getint(SECTION_RETHINKDB, OPT_PORT),
                           config.get(SECTION_RETHINKDB, OPT_DB),
                           config.get(SECTION_RETHINKDB, OPT_PWD),
                           config.getint(SECTION_RETHINKDB, OPT_TIMEOUT))
    rdb.insertTweets(tweets, config.get(SECTION_RETHINKDB, OPT_TABLE))
    rdb.disconnect()

    logger.info("Fin du script.")
    def testGetTweets(self):
        twit = TwitterManager(
                          self.config.get(nfe204Projet.SECTION_TWITTER, nfe204Projet.OPT_CONSUMER_KEY),
                          self.config.get(nfe204Projet.SECTION_TWITTER, nfe204Projet.OPT_CONSUMER_SECRET),
                          self.config.get(nfe204Projet.SECTION_TWITTER, nfe204Projet.OPT_OAUTH_TOKEN),
                          self.config.get(nfe204Projet.SECTION_TWITTER, nfe204Projet.OPT_OAUTH_TOKEN_SECRET))

        tweets = twit.getTweets(self.config.getint(nfe204Projet.SECTION_QUERY, nfe204Projet.OPT_MAX_COUNT),
                            self.config.getint(nfe204Projet.SECTION_QUERY, nfe204Projet.OPT_MAX_RANGE))
    
        self.logger.debug(json.dumps(tweets, indent=1))
        self.assertIsNotNone(tweets, 'Test en erreur')
    def testGetTweets(self):
        twit = TwitterManager(
            self.config.get(nfe204Projet.SECTION_TWITTER,
                            nfe204Projet.OPT_CONSUMER_KEY),
            self.config.get(nfe204Projet.SECTION_TWITTER,
                            nfe204Projet.OPT_CONSUMER_SECRET),
            self.config.get(nfe204Projet.SECTION_TWITTER,
                            nfe204Projet.OPT_OAUTH_TOKEN),
            self.config.get(nfe204Projet.SECTION_TWITTER,
                            nfe204Projet.OPT_OAUTH_TOKEN_SECRET))

        tweets = twit.getTweets(
            self.config.getint(nfe204Projet.SECTION_QUERY,
                               nfe204Projet.OPT_MAX_COUNT),
            self.config.getint(nfe204Projet.SECTION_QUERY,
                               nfe204Projet.OPT_MAX_RANGE))

        self.logger.debug(json.dumps(tweets, indent=1))
        self.assertIsNotNone(tweets, 'Test en erreur')