コード例 #1
0
ファイル: stats_test.py プロジェクト: mhawthorne/antonym
    def test_get(self):
        request, response = new_mock_request_response(self.m)
        self.m.StubOutWithMock(ArtifactSourceAccessor, "find_artifact_counts")
        self.m.StubOutWithMock(ArtifactSourceAccessor, "find_artifact_counts_newer")
        self.m.StubOutWithMock(memcache, "get_stats")
        self.m.StubOutWithMock(ConfigurationAccessor, "get_or_create")
        
        ArtifactSourceAccessor.find_artifact_counts().AndReturn({})
        ArtifactSourceAccessor.find_artifact_counts_newer(IgnoreArg()).AndReturn({})
        memcache.get_stats().AndReturn({})
        
        # token = "oauth_token_secret=hi&oauth_token=hi"
        # config = MockEntity(key_name="hi", twitter_oauth_enabled=True, twitter_access_token=token, twitter_read_only=False)
        # ConfigurationAccessor.get_or_create().AndReturn(config)

        api = new_mock_twitter_api(self.m)
        api.GetUserTimeline(count=IgnoreArg()).AndReturn(())
        api.GetDirectMessages(since=IgnoreArg()).AndReturn(())
        api.GetReplies(since=IgnoreArg()).AndReturn(())
                    
        handler = StatsHandler()
        handler.initialize(request, response)
        
        self.m.ReplayAll()
        handler.get()
        self.m.VerifyAll()
コード例 #2
0
ファイル: stats.py プロジェクト: mhawthorne/antonym
    def get(self, **kw):
        helper = RequestHelper(self)
        result = {}

        today = datetime.combine(datetime.now(), time(0, 0, 0))
        result['timestamp'] = str(datetime.now())
        result['today'] = str(today)

        warnings = []
        result['warnings'] = warnings

        # source/artifact counts
        result[
            'source_artifact_counts'] = ArtifactSourceAccessor.find_artifact_counts(
            )

        # new artifacts
        result[
            'new_artifacts'] = ArtifactSourceAccessor.find_artifact_counts_newer(
                today)

        # newer_arts = ArtifactAccessor.find_newer(today, refresh=True)

        # new_art_stats = []
        # result['new_artifacts'] = new_art_stats

        # for art in newer_arts:
        #     new_art_stats.append(dict(guid=art.guid, source_name=art.source_name))

        # memcache stats
        result['memcache'] = memcache.get_stats()

        try:
            twactor = TwitterActor()

            # outgoing messages
            result['statuses_out'] = [
                describe_status_with_timestamp(s)
                for s in twactor.latest_statuses(5)
            ]

            # incoming messages
            direct_stats = []
            mention_stats = []
            result['directs'] = direct_stats
            result['mentions'] = mention_stats
            directs, mentions = twactor.messages(today)
            directs.reverse()
            mentions.reverse()
            direct_stats.extend(
                [describe_status_with_timestamp(s) for s in directs])
            mention_stats.extend(
                [describe_status_with_timestamp(s) for s in mentions])
        except TwitterException, ex:
            warnings.append(str(ex))
コード例 #3
0
ファイル: stats.py プロジェクト: mhawthorne/antonym
 def get(self, **kw):
     helper = RequestHelper(self)
     result = {}
     
     today = datetime.combine(datetime.now(), time(0, 0, 0))
     result['timestamp'] = str(datetime.now())
     result['today'] = str(today)
     
     warnings = []
     result['warnings'] = warnings
     
     # source/artifact counts
     result['source_artifact_counts'] = ArtifactSourceAccessor.find_artifact_counts()
     
     # new artifacts
     result['new_artifacts'] = ArtifactSourceAccessor.find_artifact_counts_newer(today)
     
     # newer_arts = ArtifactAccessor.find_newer(today, refresh=True)
     
     # new_art_stats = []
     # result['new_artifacts'] = new_art_stats
             
     # for art in newer_arts:
     #     new_art_stats.append(dict(guid=art.guid, source_name=art.source_name))
     
     # memcache stats
     result['memcache'] = memcache.get_stats()
     
     try:
         twactor = TwitterActor()
         
         # outgoing messages
         result['statuses_out'] = [describe_status_with_timestamp(s) for s in twactor.latest_statuses(5)]
         
         # incoming messages
         direct_stats = []
         mention_stats = []
         result['directs'] = direct_stats
         result['mentions'] = mention_stats
         directs, mentions = twactor.messages(today)
         directs.reverse()
         mentions.reverse()
         direct_stats.extend([describe_status_with_timestamp(s) for s in directs])
         mention_stats.extend([describe_status_with_timestamp(s) for s in mentions])
     except TwitterException, ex:
         warnings.append(str(ex))