From 0e85a039d835af19263c97208a8c9f494a9940a4 Mon Sep 17 00:00:00 2001
From: "Phyks (Lucas Verney)" <phyks@phyks.me>
Date: Thu, 11 Feb 2016 15:17:32 +0100
Subject: [PATCH] Add a function to query publisher policy from Dissemin API
 with a DOI.

---
 libbmc/doi.py | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/libbmc/doi.py b/libbmc/doi.py
index 3fee128..f3a4b8d 100644
--- a/libbmc/doi.py
+++ b/libbmc/doi.py
@@ -19,6 +19,9 @@ REGEX = re.compile(r"\b(10[.][0-9]{4,}(?:[.][0-9]+)*/(?:(?![\"&\'])\S)+)\b",
 # Base dx.doi.org URL for redirections
 DX_URL = "http://dx.doi.org/{doi}"
 
+# Base DISSEMIN API
+DISSEMIN_API = "http://beta.dissem.in/api/"
+
 
 def is_valid(doi):
     """
@@ -130,9 +133,8 @@ def get_oa_version(doi):
     >>> get_oa_version('10.1209/0295-5075/111/40005')
     'http://arxiv.org/abs/1506.06690'
     """
-    # If DOI is a link, truncate it
     try:
-        r = requests.get("http://beta.dissem.in/api/%s" % (doi,))
+        r = requests.get("%s%s" % (DISSEMIN_API, doi))
         r.raise_for_status()
         result = r.json()
         assert(result["status"] == "ok")
@@ -141,6 +143,33 @@ def get_oa_version(doi):
         return None
 
 
+def get_oa_policy(doi):
+    """
+    Get OA policy for a given DOI.
+
+    .. note::
+
+        Uses beta.dissem.in API.
+
+    :param doi: A canonical DOI.
+    :returns: The OpenAccess policy for the associated publications, or \
+            ``None`` if unknown.
+
+    >>> tmp = get_oa_policy('10.1209/0295-5075/111/40005'); (tmp["published"], tmp["preprint"], tmp["postprint"], tmp["romeo_id"])
+    ('can', 'can', 'can', '1896')
+    """
+    try:
+        r = requests.get("%s%s" % (DISSEMIN_API, doi))
+        r.raise_for_status()
+        result = r.json()
+        assert(result["status"] == "ok")
+        return ([i
+                 for i in result["paper"]["publications"]
+                 if i["doi"] == doi][0])["policy"]
+    except (AssertionError, ValueError, KeyError, RequestException):
+        return None
+
+
 def get_linked_version(doi):
     """
     Get the original link behind the DOI.