Skip to main content

Python wrapper for RIPE Atlas API

Project description

RIPE Atlas Cousteau Build Status Code Health

A python wrapper around RIPE ATLAS API.

Installation

You can install by either cloning the repo and run the following inside the repo:

$ python setup.py install

or via pip using:

$ pip install https://github.com/RIPE-NCC/ripe-atlas-cousteau/zipball/latest

Usage

Creating a Measurement

Creating two new RIPE Atlas UDMs is as easy as:

from datetime import datetime
from ripe.atlas.cousteau import (
  Ping,
  Traceroute,
  AtlasSource,
  AtlasCreateRequest
)

ATLAS_API_KEY = ""

ping = Ping(af=4, target="www.google.gr", description="testing new wrapper")

traceroute = Traceroute(
    af=4,
    target="www.ripe.net",
    description="testing",
    protocol="ICMP",
)

source = AtlasSource(type="area", value="WW", requested=5)

atlas_request = AtlasCreateRequest(
    start_time=datetime.utcnow(),
    key=ATLAS_API_KEY,
    measurements=[ping, traceroute],
    sources=[source]
)

(is_success, response) = atlas_request.create()

Keep in mind that this library is trying to comply with what is stated in the documentation pages. This means that if you try to create a request that is missing a field stated as required, the library won’t go ahead and do the HTTP query. On the contrary, it will raise an exception with some info in it.

Changing Measurement Sources

Similarly if you want to change (remove in the following example) probes for an existing measurement you can do:

from ripe.atlas.cousteau import AtlasChangeSource, AtlasChangeRequest

ATLAS_MODIFY_API_KEY = ""

source = AtlasChangeSource(
    value="1,2,3",
    requested=3,
    action="remove"
)

atlas_request = AtlasChangeRequest(
    key=ATLAS_MODIFY_API_KEY,
    msm_id=1000001,
    sources=[source]
)

(is_success, response) = atlas_request.create()

Same applies if you want to add a list of probes, you just have to change “action” key to “add” as stated on the docs.

Stopping Measurement

You can stop a measurement with:

from ripe.atlas.cousteau import AtlasStopRequest

ATLAS_STOP_API_KEY = ""

atlas_request = AtlasStopRequest(msm_id=1000001, key=ATLAS_STOP_API_KEY)

(is_success, response) = atlas_request.create()

In order to be able to successfully create most of the above you need to create an API key.

Make Any API Get Requests

If you know the url path you can make any request easily towards ATLAS API.

msm_id = 2016892
url_path = (
    "api/v1/measurement/{0}/?fields=participant_count&"
    "format=json"
).format(msm_id)
request = AtlasRequest(**{"url_path": url_path})
result = namedtuple('Result', 'success response')
(is_success, response) = request.get()
if not is_success:
    return False

return result.response["participant_count"]

Fetch Results

You can fetch results for any measurements using cousteau. In the following example we are getting all results for measurement ID 2016892 and for probe IDs 1,2,3,4 between 2015-05-19 and 2015-05-20. Times can be python datetime objects, Unix timestamps or string representations of dates.

from ripe.atlas.cousteau import AtlasResultsRequest

kwargs = {
    "msm_id": 2016892,
    "start": datetime(2015, 05, 19)
    "stop": datetime(2015, 05, 20)
    "probe_ids": [1,2,3,4]
}

is_success, results = AtlasResultsRequest(**kwargs).create()

if is_success:
    print(results)

Fetch real time results

Besides fetching results from main API it is possible to get results though streaming API.

from ripe.atlas.cousteau import AtlasStream

def on_result_response(*args):
    """
    Function that will be called every time we receive a new result.
    Args is a tuple, so you should use args[0] to access the real message.
    """
    print args[0]

atlas_stream = AtlasStream()
atlas_stream.connect()
# Measurement results
stream_type = "result"
# Bind function we want to run with every result message received
atlas_stream.bind_stream(stream_type=stream_type, on_result_response)
# Subscribe to new stream for 1001 measurement results
stream_parameters = {"msm": 1001}
atlas_stream.start_stream(stream_type=stream_type, **stream_parameters)

# Probe's connection status results
stream_type = "probestatus"
atlas_stream.bind_stream(stream_type=stream_type, on_result_response)
stream_parameters = {"enrichProbes": True}
atlas_stream.start_stream(stream_type=stream_type, **stream_parameters)

# Timeout all subscriptions after 5 secs. Leave seconds empty for no timeout.
# Make sure you have this line after you start *all* your streams
atlas_stream.timeout(seconds=5)
# Shut down everything
atlas_stream.disconnect()

The available stream parameters for every stream type are described in the streaming results docs

Fetch Probes/Measurements Meta data

This is a helpful feature that hides all the complexity of traversing the API using the next url each time there are more objects. It returns a python generator that you can use to access each object.

Fetches all probes from NL with asn_v4 3333 and with tag NAT

from ripe.atlas.cousteau import ProbeRequest

filters = {"tags": "NAT", "country_code": "NL", "asn_v4": "3333"}
probes = ProbeRequest(**filters)

for probe in probes:
    print(probe["id"])

# Print total count of found probes
print(probes.total_count)

Fetches all specified measurements.

from ripe.atlas.cousteau import MeasurementRequest

filters = {"status": 1}
measurements = MeasurementRequest(**filters)

for msm in measurements:
    print(msm["msm_id"])

# Print total count of found measurements
print(measurements.total_count)

Colophon

But why Cousteau? The RIPE Atlas team decided to name all of its modules after explorers, and this is not an exception :)

Releases History

0.9 (released 2015-09-01)

New features:

  • add support for fetching real time results using RIPE Atlas stream server.

  • this version and on will be available on PYPI.

0.8 (released 2015-08-31)

New features:

  • add support for NTP measurements.

0.7 (released 2015-06-03)

New features:

  • add support for fetching probes/measurements meta data using python generators.

0.6 (released 2014-06-17)

New features:

  • add support for querying results based on start/end time, msm_id and probe id.

Changes:

  • add http agent according to package version to all requests.

0.5 (released 2014-05-22)

Changes:

  • change package structure to comply with the new structure of atlas packages

  • add continuous integration support

  • add tests

  • enable travis

  • enable code health checks

  • add required files for uploading to github

0.4 (released 2014-03-31)

New features:

  • add support for stopping a measurement.

0.3 (released 2014-02-25)

New features:

  • add simple support for HTTP GET queries.

0.2 (released 2014-02-03)

New features:

  • add support for adding/removing probes API request.

Changes:

  • use AtlasCreateRequest instead of AtlasRequest for creating a new measurement.

0.1 (released 2014-01-21)

  • Initial release.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ripe.atlas.cousteau-0.9.tar.gz (42.0 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page