More of a user's pages are visible

2018-09-10 13:07:41 -04:00
parent ae9f407dfd
commit dae2d0a27d
2 changed files with 27 additions and 13 deletions
--- a/rtv/page.py
+++ b/rtv/page.py
@@ -389,8 +389,12 @@ class Page(object):

        sub_name = self.content.name
        sub_name = sub_name.replace('/r/front', 'Front Page')
-        sub_name = sub_name.replace('/u/me', 'My Submissions')
-        sub_name = sub_name.replace('/u/saved', 'My Saved Submissions')
+
+        parts = sub_name.split('/')
+        if parts[1] == 'u':
+            noun = 'My' if parts[2] == 'me' else parts[2] + "'s"
+            user_room = {3: 'overview', 4: parts[-1]}[len(parts)].capitalize()
+            sub_name = "{} {}".format(noun, user_room)

        query = self.content.query
        if query: