forked from routedbits/dn42-peers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathroutedbits.py
58 lines (43 loc) · 1.45 KB
/
routedbits.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#
#
#
from requests import Session
class RoutedBitsNotFound(Exception):
def __init__(self):
super().__init__('Not Found')
class RoutedBitsTooManyArguments(Exception):
def __init__(self):
super().__init__('Too many arguments')
class RoutedBits(object):
BASE = 'https://dn42.routedbits.com'
def __init__(self):
self._session = Session()
def _request(self, method, path, params=None, data=None):
url = f'{self.BASE}{path}'
resp = self._session.request(method, url, params=params, json=data)
if resp.status_code == 404:
raise RoutedBitsNotFound()
resp.raise_for_status()
return resp
def nodes(self, minimal=False, sort_by='city'):
path = '/nodes.json'
resp = self._request('GET', path).json()['regions']
nodes = resp
if minimal:
nodes = []
for region in resp:
nodes.extend(region['sites'])
nodes = sorted(nodes, key=lambda node: node[sort_by])
return nodes
def node(self, hostname=None, name=None):
if hostname and name:
raise RoutedBitsTooManyArguments()
nodes = self.nodes(minimal=True)
for node in nodes:
if hostname:
if node['hostname'] == hostname:
return node
if name:
if node['name'] == name:
return node
raise RoutedBitsNotFound()