Skip to main content

Python wrapper for RIPE Atlas API

Project description

RIPE Atlas Cousteau Build Status Code Health PYPI Version

A python wrapper around RIPE ATLAS API.

(Until version 0.9.* this wrapper supported v1 API. After version 0.10 and above v2 RIPE ATLAS API is only supported.)

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 ripe.atlas.cousteau

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_oneoff=True
)

(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 (add 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="GR",
    requested=3,
    type="country",
    action="add"
)

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 remove probes, you just have to change “action” key to “remove” and specify probes you want to remove. Keep in mind remove action supports only a list of probes and not the rest of the source types. For more info check the appropriate 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.

url_path = "/api/v2/anchors"
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, 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

Filter Probes/Measurements

This feature queries API for probes/measurements based on specified filters. Filters should be as specified in filter_api. It 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)

Represent Probes/Measurements Meta data in python

This will allow you to have a python object with attributes populated from probes/measurements meta data. Every time you create a new instance it will fetch meta data from API and return an object with selected attributes.

from ripe.atlas.cousteau import Probe, Measurement

probe = Probe(id=3)
print(probe.country_code)
print(probe.is_anchor)
print(probe.is_public)
print(probe.address_v4)
print(dir(probe)) # Full list of properties

measurement = Measurement(id=1000002)
print(measurement.protocol)
print(measurement.description)
print(measurement.is_oneoff)
print(measurement.is_public)
print(measurement.destination_address)
print(measurement.destination_asn)
print(measurement.type)
print(measurement.interval)
print(dir(measurement)) # Full list of properties

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.10.1 (released 2015-10-06)

New features:

  • Implement support for object return in the request generators for probe/measurement.

Changes:

  • Probe/Measurement python representation takes meta data from v2 API as well. Now everything should point to v2 API.

0.10 (released 2015-10-01)

New features:

  • add object representation of meta data for a probe or a measurement.

Changes:

  • Abandon v1 RIPE ATLAS API and use only v2.

Bug fix:

  • Fix bug that prevented users from specifying all possible source types when they tried to add more probes to existing measurements.

  • Cover case where a user specified really long list of probes/measurements in the ProbeRequest/MeasurementRequest that was causing ‘HTTP ERROR 414: Request-URI Too Long’. Additionally, now if API returns error raise an exception instead of stopping iteration.

0.9.2 (released 2015-09-21)

Changes:

  • Small refactor of Stream class and manually enforce websockets in SocketIO client

0.9.1 (released 2015-09-03)

Bug fix:

  • Fix bug related to binding result atlas stream.

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.10.1.tar.gz (48.2 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