#21 Create licenses

Signed-off-by: Uilian Ries <uilianries@gmail.com>

Author: uilianries

bintray/bintray.py

@@ -107,6 +107,45 @@ def download_content(self, subject, repo, remote_file_path, local_file_path):
 
     # Licenses
 
+    def get_org_proprietary_licenses(self, org):
+        """ Get a list of custom, proprietary licenses associated with an organization
+
+        :param org: Organization name
+        :return: Licenses list
+        """
+        url = "{}/orgs/{}/licenses".format(Bintray.BINTRAY_URL, org)
+        return self._requester.get(url)
+
+    def get_user_proprietary_licenses(self, user):
+        """ Get a list of custom, proprietary licenses associated with an user
+
+        :param user: User name
+        :return: Licenses list
+        """
+        url = "{}/users/{}/licenses".format(Bintray.BINTRAY_URL, user)
+        return self._requester.get(url)
+
+    def create_org_proprietary_license(self, org, license):
+        """ Create a license associated with an organization.
+            Caller must be an admin of the organization.
+
+        :param org: Organization name
+        :param license: JSON data with license information
+        :return: request answer
+        """
+        url = "{}/orgs/{}/licenses".format(Bintray.BINTRAY_URL, org)
+        return self._requester.post(url, json=license)
+
+    def create_user_proprietary_license(self, user, license):
+        """ Create a license associated with an user.
+
+        :param user: User name
+        :param license: JSON data with license information
+        :return: request answer
+        """
+        url = "{}/users/{}/licenses".format(Bintray.BINTRAY_URL, user)
+        return self._requester.post(url, json=license)
+
     def get_oss_licenses(self):
         """ Returns a list of all the OSS licenses.
 

bintray/requester.py

@@ -78,3 +78,29 @@ def put(self, url, params=None, data=None):
         if not response.ok:
             self._raise_error("Could not PUT", response)
         return self._add_status_code(response)
+
+    def post(self, url, json=None, params=None):
+        """ Forward POST method
+
+        :param url: URL address
+        :param params: URL parameters
+        :param json: Data to be posted
+        :return: Request response
+        """
+        response = requests.post(url, auth=self._get_authentication(), json=json, params=params)
+        if not response.ok:
+            self._raise_error("Could not POST", response)
+        return self._add_status_code(response)
+
+    def patch(self, url, json=None, params=None):
+        """ Forward PATCH method
+
+        :param url: URL address
+        :param params: URL parameters
+        :param json: Data to be patched
+        :return: Request response
+        """
+        response = requests.post(url, auth=self._get_authentication(), json=json, params=params)
+        if not response.ok:
+            self._raise_error("Could not PATCH", response)
+        return self._add_status_code(response)

tests/test_licenses.py

@@ -10,6 +10,28 @@ def test_get_oss_licenses():
     assert copyfree in licenses
 
 
+def test_get_org_proprietary_licenses():
+    bintray = Bintray()
+    error_message = ""
+    try:
+        bintray.get_org_proprietary_licenses(org="jfrog")
+    except Exception as error:
+        error_message = str(error)
+    assert "Could not GET (403): 403 Client Error: Forbidden for url: " \
+            "https://api.bintray.com/orgs/jfrog/licenses" == error_message
+
+
+def test_get_user_proprietary_licenses():
+    bintray = Bintray()
+    error_message = ""
+    try:
+        bintray.get_user_proprietary_licenses(user="uilianries")
+    except Exception as error:
+        error_message = str(error)
+    assert "Could not GET (400): 400 Client Error: Bad Request for url: " \
+            "https://api.bintray.com/users/uilianries/licenses" == error_message
+
+
 def test_bad_credentials_for_get_oss_licenses():
     bintray = Bintray("foobar", "85abc6aece02515e8bd87b9754a18af697527d88")
     error_message = ""
@@ -19,3 +41,25 @@ def test_bad_credentials_for_get_oss_licenses():
         error_message = str(error)
     assert "Could not GET (401): 401 Client Error: Unauthorized for url: " \
             "https://api.bintray.com/licenses/oss_licenses" == error_message
+
+
+def test_create_org_proprietary_licenses():
+    bintray = Bintray()
+    error_message = ""
+    try:
+        bintray.create_org_proprietary_license(org="jfrog", license=[{}])
+    except Exception as error:
+        error_message = str(error)
+    assert "Could not POST (403): 403 Client Error: Forbidden for url: " \
+            "https://api.bintray.com/orgs/jfrog/licenses" == error_message
+
+
+def test_create_user_proprietary_licenses():
+    bintray = Bintray()
+    error_message = ""
+    try:
+        bintray.create_user_proprietary_license(user="uilianries", license=[{}])
+    except Exception as error:
+        error_message = str(error)
+    assert "Could not POST (400): 400 Client Error: Bad Request for url: " \
+            "https://api.bintray.com/users/uilianries/licenses" == error_message