Encapsulate Link header behaviour

1 files changed, 26 insertions, 0 deletions

diff --git a/pygithub3/core/link.py b/pygithub3/core/link.py
new file mode 100644
index 0000000..1d6be2c
--- /dev/null
+++ b/pygithub3/core/link.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from urlparse import urlparse, parse_qs
+
+from .third_libs.link_header import parse_link_value
+
+class Link(str):
+
+    class Url(str):
+
+        @property
+        def query(self):
+            return urlparse(self).query
+
+        @property
+        def params(self):
+            return dict([
+                (param, values.pop())
+                for param, values in parse_qs(self.query).items()])
+
+    def __init__(self, object_):
+        super(Link, self).__init__(object_)
+        parsed = parse_link_value(self)
+        for url in parsed:
+            setattr(self, parsed[url]['rel'], Link.Url(url))