Allow dataverse creation/deletion

Author: rliebz

dataverse/connection.py

@@ -2,14 +2,13 @@
 import requests
 
 from dataverse import Dataverse
-from exceptions import UnauthorizedError, ConnectionError
+import exceptions
 from utils import get_elements
 
 
 class Connection(object):
 
     def __init__(self, host, token):
-        # Connection Properties
         self.token = token
         self.host = host
         self.sd_uri = "https://{host}/dvn/api/data-deposit/v1.1/swordv2/service-document".format(host=self.host)
@@ -25,12 +24,41 @@ def connect(self):
         resp = requests.get(self.sd_uri, auth=self.auth)
 
         if resp.status_code == 403:
-            raise UnauthorizedError('The credentials provided are invalid.')
+            raise exceptions.UnauthorizedError('The credentials provided are invalid.')
         elif resp.status_code != 200:
-            raise ConnectionError('Could not connect to the Dataverse')
+            raise exceptions.ConnectionError('Could not connect to the Dataverse')
 
         self.service_document = etree.XML(resp.content)
-        
+
+    def create_dataverse(self, alias, name, email, parent=':root'):
+        resp = requests.post(
+            'https://{0}/api/dataverses/{1}'.format(self.host, parent),
+            json={
+                'alias': alias,
+                'name': name,
+                'dataverseContacts': [{'contactEmail': email}],
+            },
+            params={'key': self.token},
+        )
+
+        if resp.status_code == 404:
+            raise exceptions.DataverseNotFoundError('Dataverse {0} was not found.'.format(parent))
+        elif resp.status_code != 201:
+            raise exceptions.OperationFailedError('{0} Dataverse could not be created.'.format(name))
+
+    def delete_dataverse(self, alias):
+        resp = requests.delete(
+            'https://{0}/api/dataverses/{1}'.format(self.host, alias),
+            params={'key': self.token},
+        )
+
+        if resp.status_code == 401:
+            raise exceptions.UnauthorizedError('Delete Dataverse unauthorized.')
+        elif resp.status_code == 404:
+            raise exceptions.DataverseNotFoundError('Dataverse {0} was not found.'.format(alias))
+        elif resp.status_code != 200:
+            raise exceptions.OperationFailedError('Dataverse {0} could not be deleted.'.format(alias))
+
     def get_dataverses(self, refresh=False):
         if refresh:
             self.connect()

dataverse/exceptions.py

@@ -45,4 +45,9 @@ class UnpublishedDataverseError(DataverseError):
 
 class VersionJsonNotFoundError(DataverseError):
     """Raised when requested json data for a version is not found"""
+    pass
+
+
+class DataverseNotFoundError(DataverseError):
+    """Raised when a Dataverse cannot be found"""
     pass