def test_take_outdated(self):
self.assertFalse(Feed.get_by_pk(self._feed.id).dt_checked)
outdated1 = Feed.take_outdated_feeds()
self.assertEqual(len(outdated1), 1)
self.assertEqual(outdated1[0]['feed_id'], self._feed.id)
self.assertTrue(Feed.get_by_pk(self._feed.id).dt_checked)
outdated2 = Feed.take_outdated_feeds()
self.assertEqual(len(outdated2), 0)
def _feed_merge_duplicate(found: list):
for feed_ids in found:
primary_id, *duplicates = feed_ids
with transaction.atomic():
primary = Feed.get_by_pk(primary_id)
primary_info = f'#{primary.id} url={primary.url!r}'
for feed_id in duplicates:
other = Feed.get_by_pk(feed_id)
other_info = f'#{other.id} url={other.url!r}'
LOG.info('merge duplicate feed %s into %s', other_info, primary_info)
FeedUrlMap(source=other.url, target=primary.url).save()
primary.merge(other)
def do_sync_story_fulltext(
ctx: ActorContext,
feed_id: T.int,
offset: T.int,
) -> T.dict(
feed_id=T.int,
offset=T.int.min(0),
use_proxy=T.bool,
url=T.url,
response_status=T.int,
accept=T_ACCEPT,
):
with log_django_context_metric('harbor_rss.sync_story_fulltext:read'):
feed = Feed.get_by_pk(feed_id, detail='+use_proxy')
story = STORY_SERVICE.get_by_offset(feed_id, offset, detail=True)
assert story, f'story#{feed_id},{offset} not found'
story_content_info = StoryContentInfo(story.content)
num_sub_sentences = len(split_sentences(story_content_info.text))
ret = dict(
feed_id=feed_id,
offset=offset,
url=story.link,
use_proxy=feed.use_proxy,
accept=FulltextAcceptStrategy.REJECT.value,
)
try:
result = ctx.ask(
'worker_rss.fetch_story',
dict(
url=story.link,
use_proxy=feed.use_proxy,
feed_id=feed_id,
offset=offset,
num_sub_sentences=num_sub_sentences,
))
except _TIMEOUT_ERRORS as ex:
LOG.error(f'Ask worker_rss.fetch_story timeout: {ex}')
ret.update(response_status=FeedResponseStatus.CONNECTION_TIMEOUT)
return ret
else:
ret.update(
response_status=result['response_status'],
use_proxy=result['use_proxy'],
)
if not result['content']:
return ret
with log_django_context_metric('harbor_rss.sync_story_fulltext:write'):
accept = _update_story(
story=story,
story_content_info=story_content_info,
content=result['content'],
summary=None, # not need update summary
url=result['url'],
sentence_count=result['sentence_count'],
)
ret.update(accept=accept.value)
return ret
def do_update_feed(
ctx: ActorContext,
feed_id: T.int,
feed: FeedSchema,
is_refresh: T.bool.default(False).desc('Deprecated'),
):
with transaction.atomic():
feed_dict = feed
storys = feed_dict.pop('storys')
feed = Feed.get_by_pk(feed_id)
is_feed_url_changed = feed.url != feed_dict['url']
if is_feed_url_changed:
target_feed = Feed.get_first_by_url(feed_dict['url'])
if target_feed:
LOG.info(f'merge feed#{feed.id} url={feed.url} into '
f'feed#{target_feed.id} url={target_feed.url}')
target_feed.merge(feed)
return
for k, v in feed_dict.items():
if v != '' and v is not None:
setattr(feed, k, v)
now = timezone.now()
now_sub_30d = now - timezone.timedelta(days=30)
if not feed.dt_updated:
feed.dt_updated = now
feed.dt_checked = feed.dt_synced = now
feed.status = FeedStatus.READY
feed.save()
for s in storys:
if not s['dt_updated']:
s['dt_updated'] = now
if not s['dt_published']:
# set dt_published to now - 30d to avoid these storys
# take over mushroom page, i.e. Story.query_recent_by_user
s['dt_published'] = now_sub_30d
modified_storys = Story.bulk_save_by_feed(feed.id, storys)
LOG.info(
'feed#%s save storys total=%s num_modified=%s',
feed.id, len(storys), len(modified_storys)
)
feed.refresh_from_db()
if modified_storys:
feed.unfreeze()
need_fetch_story = _is_feed_need_fetch_storys(feed)
for story in modified_storys:
if not story.link:
continue
if need_fetch_story and (not is_fulltext_story(feed, story)):
ctx.tell('worker_rss.fetch_story', dict(
url=story.link,
story_id=str(story.id)
))
else:
_detect_story_images(ctx, story)
def do_update_feed_info(
ctx: ActorContext,
feed_id: T.int,
feed: FeedInfoSchema,
):
with transaction.atomic():
feed_dict = feed
feed = Feed.get_by_pk(feed_id)
for k, v in feed_dict.items():
setattr(feed, k, v)
feed.dt_updated = timezone.now()
feed.save()
def update_feed_dryness(feeds=None):
feed_ids = _get_feed_ids(feeds)
LOG.info('total %s feeds', len(feed_ids))
for feed_id in tqdm.tqdm(feed_ids, ncols=80, ascii=True):
with transaction.atomic():
feed = Feed.get_by_pk(feed_id)
if feed.total_storys <= 0:
continue
cnt = feed.monthly_story_count
if not cnt:
Story.refresh_feed_monthly_story_count(feed_id)
feed.refresh_from_db()
feed.dryness = feed.monthly_story_count.dryness()
feed.save()
def delete_feed(key):
try:
key = int(key)
except ValueError:
pass # ignore
if isinstance(key, int):
feed = Feed.get_by_pk(key)
else:
feed = Feed.objects.filter(
Q(url__contains=key) | Q(title__contains=key)).first()
if not feed:
print(f'not found feed like {key}')
return
if click.confirm(f'delete {feed} ?'):
feed.delete()
def update_feed_dt_first_story_published(feeds=None):
feed_ids = _get_feed_ids(feeds)
LOG.info('total %s feeds', len(feed_ids))
for feed_id in tqdm.tqdm(feed_ids, ncols=80, ascii=True):
with transaction.atomic():
feed = Feed.get_by_pk(feed_id)
if feed.dt_first_story_published:
continue
if feed.total_storys <= 0:
continue
try:
story = Story.get_by_offset(feed_id, 0, detail=True)
except Story.DoesNotExist:
LOG.warning(f'story feed_id={feed_id} offset=0 not exists')
continue
feed.dt_first_story_published = story.dt_published
feed.save()
def assert_feed_total_storys(self, expect):
total_storys = Feed.get_by_pk(self.feed_id).total_storys
self.assertEqual(total_storys, expect)
def do_update_feed(
ctx: ActorContext,
feed_id: T.int,
feed: FeedSchema,
is_refresh: T.bool.default(False),
):
with transaction.atomic():
feed_dict = feed
storys = feed_dict.pop('storys')
feed = Feed.get_by_pk(feed_id)
is_feed_url_changed = feed.url != feed_dict['url']
if is_feed_url_changed:
target_feed = Feed.get_first_by_url(feed_dict['url'])
# FIXME: feed merge 无法正确处理订阅重定向问题。
# 对于这种情况，暂时保留旧的订阅，以后再彻底解决。
# if target_feed:
# LOG.info(f'merge feed#{feed.id} url={feed.url} into '
# f'feed#{target_feed.id} url={target_feed.url}')
# target_feed.merge(feed)
# return
if target_feed:
LOG.warning(
f'FIXME: redirect feed#{feed.id} url={feed.url!r} into '
f'feed#{target_feed.id} url={target_feed.url!r}')
feed_dict.pop('url')
# only update dt_updated if has storys or feed fields updated
is_feed_updated = bool(storys)
for k, v in feed_dict.items():
if k == 'dt_updated':
continue
if (v != '' and v is not None) or k in {'warnings'}:
old_v = getattr(feed, k, None)
if v != old_v:
is_feed_updated = True
setattr(feed, k, v)
now = timezone.now()
now_sub_30d = now - timezone.timedelta(days=30)
if is_feed_updated:
# set dt_updated to now, not trust rss date
feed.dt_updated = now
feed.dt_checked = feed.dt_synced = now
feed.reverse_url = reverse_url(feed.url)
feed.status = FeedStatus.READY
feed.save()
# save storys, bulk_save_by_feed has standalone transaction
for s in storys:
if not s['dt_updated']:
s['dt_updated'] = now
if not s['dt_published']:
# set dt_published to now - 30d to avoid these storys
# take over mushroom page, i.e. Story.query_recent_by_user
s['dt_published'] = now_sub_30d
modified_storys = STORY_SERVICE.bulk_save_by_feed(feed.id,
storys,
is_refresh=is_refresh)
LOG.info('feed#%s save storys total=%s num_modified=%s', feed.id,
len(storys), len(modified_storys))
feed = Feed.get_by_pk(feed_id)
is_freezed = feed.freeze_level is None or feed.freeze_level > 1
if modified_storys and is_freezed:
Feed.unfreeze_by_id(feed_id)
need_fetch_story = _is_feed_need_fetch_storys(feed, modified_storys)
for story in modified_storys:
if not story.link:
continue
if need_fetch_story and (not _is_fulltext_story(story)):
text = processor.story_html_to_text(story.content)
num_sub_sentences = len(split_sentences(text))
ctx.tell(
'worker_rss.fetch_story',
dict(
url=story.link,
use_proxy=feed.use_proxy,
feed_id=story.feed_id,
offset=story.offset,
num_sub_sentences=num_sub_sentences,
))
def do_update_feed(
ctx: ActorContext,
feed_id: T.int,
feed: FeedSchema,
is_refresh: T.bool.default(False),
):
with transaction.atomic():
feed_dict = feed
storys = feed_dict.pop('storys')
feed = Feed.get_by_pk(feed_id)
is_feed_url_changed = feed.url != feed_dict['url']
if is_feed_url_changed:
target_feed = Feed.get_first_by_url(feed_dict['url'])
if target_feed:
LOG.info(f'merge feed#{feed.id} url={feed.url} into '
f'feed#{target_feed.id} url={target_feed.url}')
target_feed.merge(feed)
return
# only update dt_updated if has storys or feed fields updated
is_feed_updated = bool(storys)
for k, v in feed_dict.items():
if k == 'dt_updated':
continue
if v != '' and v is not None:
old_v = getattr(feed, k, None)
if v != old_v:
is_feed_updated = True
setattr(feed, k, v)
now = timezone.now()
now_sub_30d = now - timezone.timedelta(days=30)
if is_feed_updated:
# set dt_updated to now, not trust rss date
feed.dt_updated = now
feed.dt_checked = feed.dt_synced = now
feed.status = FeedStatus.READY
feed.save()
for s in storys:
if not s['dt_updated']:
s['dt_updated'] = now
if not s['dt_published']:
# set dt_published to now - 30d to avoid these storys
# take over mushroom page, i.e. Story.query_recent_by_user
s['dt_published'] = now_sub_30d
modified_storys = Story.bulk_save_by_feed(feed.id, storys, is_refresh=is_refresh)
LOG.info(
'feed#%s save storys total=%s num_modified=%s',
feed.id, len(storys), len(modified_storys)
)
feed.refresh_from_db()
if modified_storys:
feed.unfreeze()
need_fetch_story = _is_feed_need_fetch_storys(feed, modified_storys)
for story in modified_storys:
if not story.link:
continue
if need_fetch_story and (not is_fulltext_story(story)):
text = processor.story_html_to_text(story.content)
num_sub_sentences = len(split_sentences(text))
ctx.tell('worker_rss.fetch_story', dict(
url=story.link,
use_proxy=feed.use_proxy,
story_id=str(story.id),
num_sub_sentences=num_sub_sentences,
))
else:
_detect_story_images(ctx, story)
