diff options
Diffstat (limited to 'gitlab')
| -rw-r--r-- | gitlab/__init__.py | 1 | ||||
| -rw-r--r-- | gitlab/exceptions.py | 4 | ||||
| -rw-r--r-- | gitlab/v4/objects.py | 77 |
3 files changed, 82 insertions, 0 deletions
diff --git a/gitlab/__init__.py b/gitlab/__init__.py index 2159492..4cf81ea 100644 --- a/gitlab/__init__.py +++ b/gitlab/__init__.py @@ -105,6 +105,7 @@ class Gitlab(object): self.broadcastmessages = objects.BroadcastMessageManager(self) self.deploykeys = objects.DeployKeyManager(self) + self.geonodes = objects.GeoNodeManager(self) self.gitlabciymls = objects.GitlabciymlManager(self) self.gitignores = objects.GitignoreManager(self) self.groups = objects.GroupManager(self) diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py index 514d742..4aec7fc 100644 --- a/gitlab/exceptions.py +++ b/gitlab/exceptions.py @@ -217,6 +217,10 @@ class GitlabRenderError(GitlabOperationError): pass +class GitlabRepairError(GitlabOperationError): + pass + + def on_http_error(error): """Manage GitlabHttpError exceptions. diff --git a/gitlab/v4/objects.py b/gitlab/v4/objects.py index d6ae6c5..8feb09b 100644 --- a/gitlab/v4/objects.py +++ b/gitlab/v4/objects.py @@ -3444,3 +3444,80 @@ class TodoManager(ListMixin, DeleteMixin, RESTManager): return int(result) except ValueError: return 0 + + +class GeoNode(SaveMixin, ObjectDeleteMixin, RESTObject): + @cli.register_custom_action('GeoNode') + @exc.on_http_error(exc.GitlabRepairError) + def repair(self, **kwargs): + """Repair the OAuth authentication of the geo node. + + Args: + **kwargs: Extra options to send to the server (e.g. sudo) + + Raises: + GitlabAuthenticationError: If authentication is not correct + GitlabRepairError: If the server failed to perform the request + """ + path = '/geo_nodes/%s/repair' % self.get_id() + server_data = self.manager.gitlab.http_post(path, **kwargs) + self._update_attrs(server_data) + + @cli.register_custom_action('GeoNode') + @exc.on_http_error(exc.GitlabGetError) + def status(self, **kwargs): + """Get the status of the geo node. + + Args: + **kwargs: Extra options to send to the server (e.g. sudo) + + Raises: + GitlabAuthenticationError: If authentication is not correct + GitlabGetError: If the server failed to perform the request + + Returns: + dict: The status of the geo node + """ + path = '/geo_nodes/%s/status' % self.get_id() + return self.manager.gitlab.http_get(path, **kwargs) + + +class GeoNodeManager(RetrieveMixin, UpdateMixin, DeleteMixin, RESTManager): + _path = '/geo_nodes' + _obj_cls = GeoNode + _update_attrs = (tuple(), ('enabled', 'url', 'files_max_capacity', + 'repos_max_capacity')) + + @cli.register_custom_action('GeoNodeManager') + @exc.on_http_error(exc.GitlabGetError) + def status(self, **kwargs): + """Get the status of all the geo nodes. + + Args: + **kwargs: Extra options to send to the server (e.g. sudo) + + Raises: + GitlabAuthenticationError: If authentication is not correct + GitlabGetError: If the server failed to perform the request + + Returns: + list: The status of all the geo nodes + """ + return self.gitlab.http_list('/geo_nodes/status', **kwargs) + + @cli.register_custom_action('GeoNodeManager') + @exc.on_http_error(exc.GitlabGetError) + def current_failures(self, **kwargs): + """Get the list of failures on the current geo node. + + Args: + **kwargs: Extra options to send to the server (e.g. sudo) + + Raises: + GitlabAuthenticationError: If authentication is not correct + GitlabGetError: If the server failed to perform the request + + Returns: + list: The list of failures + """ + return self.gitlab.http_list('/geo_nodes/current/failures', **kwargs) |