コード例 #1
0
ファイル: selection.py プロジェクト: thomasvs/dad
    def get(self, cb=None, *cbArgs, **cbKWArgs):
        """
        @returns: a deferred firing a generator of L{CouchTrackModel} objects.
        """
        d = defer.Deferred()

        self.debug("get")
        last = [time.time()]
        start = last[0]

        self.debug("filtering on artist_mids %r", self.document.artist_mids)

        from dadcouch.model import artist

        tracks = yield self.database._internal.viewDocs(
            "view-tracks-by-artist",
            self.database.modelFactory(artist.ItemTracksByArtist),
            keys=self.document.artist_mids,
        )

        trackList = list(tracks)
        self.debug("got %r tracks in %.3f seconds", len(trackList), time.time() - last[0])
        last[0] = time.time()

        artists = []
        gens = []

        # FIXME: reduce this somewhere
        for trackRow in trackList:
            mid = yield trackRow.getMid()
            if mid in artists:
                continue
            artists.append(mid)

            tracks = yield trackRow.getTracks()
            tracks = list(tracks)
            random.shuffle(tracks)
            gens.append((t for t in tracks))
            random.shuffle(gens)

        defer.returnValue(iterators.tmerge(*gens))
コード例 #2
0
ファイル: test_common_iterators.py プロジェクト: thomasvs/dad
    def testTMerge(self):
        i1 = (i for i in [0, 1, 2])
        i2 = (i for i in ["a", "b"])

        l = list(iterators.tmerge(i1, i2))
        self.assertEquals(l, [0, "a", 1, "b", 2])