def test_seedids_oper(self): SeedidsOper.insert_seeds(FAKE_IDS) assert len(SeedidsOper.get_seed_ids()) == 2 assert SeedidsOper.get_seed_by_id(FAKE_ID) is not None SeedidsOper.set_seed_crawled(FAKE_ID, 1) assert len(SeedidsOper.get_seed_ids()) == 1
def crawl_follower_fans(uid): seed = SeedidsOper.get_seed_by_id(uid) if seed.other_crawled == 0: rs = get_fans_or_followers_ids(uid, 1, 1) rs.extend(get_fans_or_followers_ids(uid, 2, 1)) datas = set(rs) # If data already exits, just skip it if datas: SeedidsOper.insert_seeds(datas) SeedidsOper.set_seed_other_crawled(uid)