def story_get_by_offset(
request,
feed_unionid: T.feed_unionid.object,
offset: T.int.min(0).optional,
detail: StoryDetailSchema,
set_readed: T.bool.default(False),
) -> StorySchema:
"""Story detail"""
check_unionid(request, feed_unionid)
try:
story = UnionStory.get_by_feed_offset(feed_unionid,
offset,
detail=detail)
except StoryNotFoundError:
return Response({"message": "does not exist"}, status=400)
if set_readed:
try:
UnionFeed.set_story_offset(feed_unionid, offset + 1)
except FeedStoryOffsetError as ex:
return Response({'message': str(ex)}, status=400)
except ConcurrentUpdateError as ex:
LOG.error(f'ConcurrentUpdateError: story set_readed {ex}',
exc_info=ex)
image_token = ImageToken(
referrer=story.link,
feed=feed_unionid.feed_id,
offset=offset,
).encode(secret=CONFIG.image_token_secret)
ret = story.to_dict()
ret.update(image_token=image_token)
return ret
def story_query_batch(
request,
storys: T.list(
T.dict(
feed_id=T.feed_unionid.object,
offset=T.int.min(0),
limit=T.int.min(1).max(10).default(1),
)),
detail: StoryDetailSchema,
) -> StoryResultSchema:
feed_union_ids = [x['feed_id'] for x in storys]
check_unionid(request, feed_union_ids)
story_keys = []
for item in storys:
feed_id = item['feed_id'].feed_id
offset = item['offset']
for i in range(item['limit']):
story_keys.append((feed_id, offset + i))
storys = UnionStory.batch_get_by_feed_offset(story_keys=story_keys,
user_id=request.user.id,
detail=detail)
storys = [x.to_dict() for x in storys]
return dict(
total=len(storys),
size=len(storys),
storys=storys,
)
def story_set_favorited(
request,
feed_unionid: T.feed_unionid.object,
offset: T.int.min(0).optional,
is_favorited: T.bool.default(True),
) -> StorySchema:
check_unionid(request, feed_unionid)
story = UnionStory.set_favorited_by_feed_offset(feed_unionid, offset, is_favorited=is_favorited)
return story.to_dict()
def story_query_watched(
request,
detail: StoryDetailSchema,
) -> StoryResultSchema:
"""Query watched storys"""
storys = UnionStory.query_watched(user_id=request.user.id, detail=detail)
storys = [x.to_dict() for x in storys]
return dict(
total=len(storys),
size=len(storys),
storys=storys,
)
def story_get_by_offset(
request,
feed_unionid: T.feed_unionid.object,
offset: T.int.min(0).optional,
detail: StoryDetailSchema,
) -> StorySchema:
"""Story detail"""
check_unionid(request, feed_unionid)
try:
story = UnionStory.get_by_feed_offset(feed_unionid, offset, detail=detail)
except StoryNotFoundError:
return Response({"message": "does not exist"}, status=400)
return story.to_dict()
def story_fetch_fulltext(
request,
feed_id: T.feed_unionid.object,
offset: T.int.min(0),
) -> T.dict(
feed_id=T.feed_unionid,
offset=T.int.min(0),
response_status=T.int,
response_status_name=T.str,
use_proxy=T.bool.optional,
accept=T_ACCEPT.optional,
story=StorySchema.optional,
):
feed_unionid = feed_id
check_unionid(request, feed_unionid)
user_id, feed_id = feed_unionid
content = dict(feed_id=feed_id, offset=offset)
expire_at = int(time.time() + 60)
use_proxy = None
accept = None
try:
result = scheduler.ask('harbor_rss.sync_story_fulltext',
content,
expire_at=expire_at)
except _TIMEOUT_ERRORS as ex:
LOG.error(f'Ask harbor_rss.sync_story_fulltext timeout: {ex}')
response_status = FeedResponseStatus.CONNECTION_TIMEOUT
else:
response_status = result['response_status']
use_proxy = result['use_proxy']
accept = result['accept']
story = None
if accept != FulltextAcceptStrategy.REJECT.value:
story = UnionStory.get_by_feed_offset(feed_unionid,
offset,
detail=True)
story = story.to_dict()
response_status_name = FeedResponseStatus.name_of(response_status)
return dict(
feed_id=feed_unionid,
offset=offset,
response_status=response_status,
response_status_name=response_status_name,
use_proxy=use_proxy,
accept=accept,
story=story,
)
def story_query_recent(
request,
feed_ids: T.list(T.feed_unionid.object).optional,
days: T.int.min(1).max(30).default(14),
detail: StoryDetailSchema,
) -> StoryResultSchema:
check_unionid(request, feed_ids)
storys = UnionStory.query_recent_by_user(user_id=request.user.id,
feed_unionids=feed_ids,
days=days,
detail=detail)
storys = [x.to_dict() for x in storys]
return dict(
total=len(storys),
size=len(storys),
storys=storys,
)
def story_get_by_offset(
request,
feed_unionid: T.feed_unionid.object,
offset: T.int.min(0).optional,
detail: StoryDetailSchema,
) -> StorySchema:
"""Story detail"""
check_unionid(request, feed_unionid)
try:
story = UnionStory.get_by_feed_offset(feed_unionid, offset, detail=detail)
except StoryNotFoundError:
return Response({"message": "does not exist"}, status=400)
image_token = ImageToken(referrer=story.link)\
.encode(secret=CONFIG.image_token_secret)
ret = story.to_dict()
ret.update(image_token=image_token)
return ret
def story_query_by_feed(
request,
feed_id: T.feed_unionid.object,
offset: T.int.min(0).optional,
size: T.int.min(1).max(100).default(10),
detail: StoryDetailSchema,
) -> StoryResultSchema:
"""Story list"""
check_unionid(request, feed_id)
try:
total, offset, storys = UnionStory.query_by_feed(
feed_unionid=feed_id, offset=offset, size=size, detail=detail)
except FeedNotFoundError:
return Response({"message": "feed does not exist"}, status=400)
storys = [x.to_dict() for x in storys]
return dict(
total=total,
offset=offset,
size=len(storys),
storys=storys,
)