diff --git a/pydas/drivers.py b/pydas/drivers.py index 2e30f78..a902c5e 100644 --- a/pydas/drivers.py +++ b/pydas/drivers.py @@ -922,7 +922,7 @@ def create_link(self, token, folder_id, url, **kwargs): response = self.request('midas.link.create', parameters) return response - def generate_upload_token(self, token, item_id, filename, checksum=None): + def generate_upload_token(self, token, item_id, filename, **kwargs): """ Generate a token to use for upload. @@ -953,8 +953,10 @@ def generate_upload_token(self, token, item_id, filename, checksum=None): parameters['token'] = token parameters['itemid'] = item_id parameters['filename'] = filename - if checksum is not None: - parameters['checksum'] = checksum + optional_keys = ['checksum', 'create_additional_revision'] + for key in optional_keys: + if key in kwargs: + parameters[key] = kwargs[key] response = self.request('midas.upload.generatetoken', parameters) return response['token']