From 36e8dcc191f8e8403f5fa96581f11b89773af62e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dan=20Michael=20O=2E=20Hegg=C3=B8?= <danmichaelo@gmail.com>
Date: Sun, 2 Sep 2012 16:33:31 +0200
Subject: [PATCH] Add shorthand Page.resolve_redirect()
so you can do something like
>>> page = site.pages[pagename].resolve_redirect()
without knowing if 'pagename' is a redirect page or not
---
page.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/page.py b/page.py
index b2c6218..6a3e5e4 100644
--- a/page.py
+++ b/page.py
@@ -58,6 +58,14 @@ class Page(object):
else:
return None
+ def resolve_redirect(self):
+ """ Returns the redirect target page, or the current page if it's not a redirect page."""
+ target_page = self.redirects_to()
+ if target_page == None:
+ return self
+ else:
+ return target_page
+
def __repr__(self):
return "<Page object '%s' for %s>" % (self.name.encode('utf-8'), self.site)
--
GitLab