Example #1
0
    def delete_vertex(self, id):
        this_object = self._validated_vertex_id(id)

        if this_object is None:
            raise ResourceNotFoundException()
        else:
            self._do_next(self.g.V(this_object).drop())
Example #2
0
    def get_inbound(self, id, search_depth=DEFAULT_SEARCH_DEPTH):
        this_object = self._validated_vertex_id(id)

        if this_object is None:
            raise ResourceNotFoundException(f"Unable to resolve Object {id}")
        else:
            return _format_graph_results(
                self._depth_search(this_object, __.inE(), search_depth))
Example #3
0
def get_glue_job_status(job_name, run_id):
    glue_client = _get_glue_client()

    try:
        response = glue_client.get_job_run(JobName=job_name, RunId=run_id)

        if response is not None:
            return _extract_glue_job_status(response['JobRun'])
    except glue_client.exceptions.EntityNotFoundException:
        raise ResourceNotFoundException("Unable to resolve Job Name or Run ID")
Example #4
0
def get_running_export_jobs(job_name):
    glue_client = _get_glue_client()

    response = glue_client.get_job_runs(JobName=job_name)

    if response is None:
        raise ResourceNotFoundException(
            f"Unable to resolve Job Name {job_name}")
    else:
        running = []
        if 'JobRuns' in response:
            for j in response['JobRuns']:
                status = j['JobRunState']

                if not any(x in status
                           for x in ['STARTING', 'RUNNING', 'STOPPING']):
                    pass
                else:
                    running.append(_extract_glue_job_status(j))

        return running
Example #5
0
    def setup_graph_method(self, *args, **kwargs):
        # lazy load the graph traverser
        if self._url is not None and self.g is None:
            n = Neptune()
            try:
                self.g = n.graphTraversal(
                    neptune_endpoint=self._url,
                    neptune_port=self._port,
                    retry_limit=self._connection_retries
                    if self._connection_retries is not None else
                    params.DEFAULT_RETRY_COUNT)
            except Exception as e:
                log.error(e)
                raise ResourceNotFoundException(
                    f"Unable to connect to Gremlin Endpoint {self._url}:{self._port}"
                )

            print(f"Connected Gremlin Endpoint {self._url}:{self._port}")

        # return the decorated function
        return f(self, *args, **kwargs)