new: Small query API

pull/12/head
Raphaël Vinot 2018-04-10 21:27:35 +02:00
parent 5670db9d1d
commit 0764f066fd
3 changed files with 80 additions and 0 deletions

0
api/__init__.py Normal file
View File

76
api/api.py Normal file
View File

@ -0,0 +1,76 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from typing import TypeVar
from datetime import datetime, date
from enum import Enum
from dateutil.parser import parse
import logging
from redis import StrictRedis
from bgpranking.libs.helpers import get_socket_path
from bgpranking.libs.exceptions import InvalidDateFormat
Dates = TypeVar('Dates', datetime, date, str)
class IPVersion(Enum):
v4 = 'v4'
v6 = 'v6'
class BGPRanking():
def __init__(self, loglevel: int=logging.DEBUG):
self.__init_logger(loglevel)
self.storage = StrictRedis(unix_socket_path=get_socket_path('storage'), decode_responses=True)
self.ranking = StrictRedis(unix_socket_path=get_socket_path('storage'), db=1, decode_responses=True)
self.asn_meta = StrictRedis(unix_socket_path=get_socket_path('storage'), db=2, decode_responses=True)
def __init_logger(self, loglevel: int):
self.logger = logging.getLogger(f'{self.__class__.__name__}')
self.logger.setLevel(loglevel)
def __normalize_date(self, date: Dates):
if isinstance(date, datetime):
return date.date().isoformat()
elif isinstance(date, date):
return date.isoformat()
elif isinstance(date, str):
try:
return parse(date).date().isoformat()
except ValueError:
raise InvalidDateFormat('Unable to parse the date. Should be YYYY-MM-DD.')
def asns_global_ranking(self, date: Dates= date.today(),
ipversion: IPVersion=IPVersion.v4, limit: int=100):
'''Aggregated ranking of all the ASNs known in the system, weighted by source.'''
d = self.__normalize_date(date)
key = f'{d}|asns|{ipversion.value}'
return self.ranking.zrevrange(key, start=0, end=limit, withscores=True)
def asn_details(self, asn: int, date: Dates= date.today(), ipversion: IPVersion=IPVersion.v4):
'''Aggregated ranking of all the prefixes anounced by the given ASN, weighted by source.'''
d = self.__normalize_date(date)
key = f'{d}|{asn}|{ipversion.value}'
return self.ranking.zrevrange(key, start=0, end=-1, withscores=True)
def asn_rank(self, asn: int, date: Dates= date.today(), ipversion: IPVersion=IPVersion.v4):
'''Get the rank of a single ASN, weighted by source.'''
d = self.__normalize_date(date)
key = f'{d}|asns|{ipversion.value}'
return self.ranking.zrank(key, asn)
def asn_rank_by_source(self, asn: int, source: str, date: Dates= date.today(), ipversion: IPVersion=IPVersion.v4):
'''Get the rank of a single ASN, not weighted by source.'''
d = self.__normalize_date(date)
key = f'{d}|{source}|{asn}|rank{ipversion.value}'
return self.ranking.get(key)
def asn_details_by_source(self, source: str, asn: int, date: Dates= date.today(),
ipversion: IPVersion=IPVersion.v4):
'''Get the rank of all the prefixes announced by an ASN, not weighted by source.'''
d = self.__normalize_date(date)
key = f'{d}|{source}|{asn}|rank{ipversion.value}|prefixes'
return self.ranking.zrevrange(key, start=0, end=-1, withscores=True)

View File

@ -20,3 +20,7 @@ class CreateDirectoryException(BGPRankingException):
class MissingEnv(BGPRankingException):
pass
class InvalidDateFormat(BGPRankingException):
pass