diff --git a/rtv/content.py b/rtv/content.py index cbbeb9f..4092163 100644 --- a/rtv/content.py +++ b/rtv/content.py @@ -310,14 +310,7 @@ class SubredditContent(BaseContent): else: if '/' in name: - parse = name.split('/') - name = parse[0] - - if parse[1] == 'top': - display_type = 'top' - - elif parse[1] == 'new': - display_type = 'new' + name, display_type = name.split('/') try: with loader(): @@ -326,10 +319,10 @@ class SubredditContent(BaseContent): raise SubredditNameError(name) if display_type == 'top': - return cls('/r/'+sub.display_name, sub.get_top_from_all(limit=None), loader) + return cls('/r/'+sub.display_name + '/top', sub.get_top_from_all(limit=None), loader) elif display_type == 'new': - return cls('/r/'+sub.display_name, sub.get_new(limit=None), loader) + return cls('/r/'+sub.display_name + '/new', sub.get_new(limit=None), loader) else: return cls('/r/'+sub.display_name, sub.get_hot(limit=None), loader)