diff --git a/rtv/content.py b/rtv/content.py index 4e5b296..d7cc17f 100644 --- a/rtv/content.py +++ b/rtv/content.py @@ -109,6 +109,7 @@ class BaseContent(object): data['url'] = ('selfpost' if is_selfpost(sub.url) else sub.url) data['likes'] = sub.likes data['gold'] = sub.gilded > 0 + data['nsfw'] = sub.over_18 return data diff --git a/rtv/submission.py b/rtv/submission.py index dc33f0e..4e466eb 100644 --- a/rtv/submission.py +++ b/rtv/submission.py @@ -267,4 +267,8 @@ class SubmissionPage(BasePage): text, attr = GOLD, (curses.A_BOLD | Color.YELLOW) win.addnstr(text, n_cols - win.getyx()[1], attr) + if data['nsfw']: + text, attr = 'NSFW', (curses.A_BOLD | Color.RED) + win.addnstr(text, n_cols - win.getyx()[1], attr) + win.border() \ No newline at end of file diff --git a/rtv/subreddit.py b/rtv/subreddit.py index bc3dae8..d1f7ff3 100644 --- a/rtv/subreddit.py +++ b/rtv/subreddit.py @@ -194,6 +194,10 @@ class SubredditPage(BasePage): text, attr = GOLD, (curses.A_BOLD | Color.YELLOW) win.addnstr(text, n_cols - win.getyx()[1], attr) + if data['nsfw']: + text, attr = 'NSFW', (curses.A_BOLD | Color.RED) + win.addnstr(text, n_cols - win.getyx()[1], attr) + row = n_title + offset + 2 if row in valid_rows: text = clean(u'{author}'.format(**data))