You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
import requests
|
|
import requests.exceptions
|
|
import json
|
|
import time
|
|
|
|
class ApiKeyAuth( requests.auth.AuthBase ):
|
|
def __init__( self, api_username, api_key ):
|
|
self.api_username = api_username
|
|
self.api_key = api_key
|
|
|
|
def __call__( self, r ):
|
|
r.headers[ 'Authorization' ] = "ApiKey %s:%s" % ( self.api_username, self.api_key )
|
|
r.headers[ 'Content-Type' ] = "application/json"
|
|
r.headers[ 'Accept-Encoding' ] = "application/json"
|
|
return r
|
|
|
|
class APIv2( object ):
|
|
def __init__( self, api_username, api_key, base = "https://fulcrm.org", api_path = "/api/v2" ):
|
|
self.api_username = api_username
|
|
self.api_key = api_key
|
|
self.api_path = api_path
|
|
self.base = base
|
|
self.auth = ApiKeyAuth( self.api_username, self.api_key )
|
|
|
|
def normalize_url( self, url, no_d = False, count = None ):
|
|
if not url.startswith( self.base ):
|
|
if not url.startswith( self.api_path ):
|
|
if not url.startswith( "/" ):
|
|
url = "/" + url
|
|
url = self.api_path + url
|
|
url = self.base + url
|
|
|
|
if not no_d:
|
|
if "?" in url:
|
|
url = url + "&expand=d&count=" + str( count )
|
|
else:
|
|
url = url + "?expand=d&count=" + str( count )
|
|
|
|
if count is not None:
|
|
if "?" in url:
|
|
url = url + "&count=" + str( count )
|
|
else:
|
|
url = url + "?count=" + str( count )
|
|
|
|
|
|
return url
|
|
|
|
def make_request( self, url, data, method, _countdown = 5 ):
|
|
url = self.normalize_url( url )
|
|
|
|
method = { 'POST': requests.post,
|
|
'PATCH': requests.patch,
|
|
'GET': requests.get,
|
|
'DELETE': requests.delete
|
|
}.get( method, requests.get )
|
|
|
|
try:
|
|
if data:
|
|
request = method( url, data = json.dumps( data ), auth = self.auth )
|
|
else:
|
|
request = method( url, auth = self.auth )
|
|
except requests.exceptions.ConnectionError:
|
|
if _countdown > 0:
|
|
time.sleep( 2 ** ( 5 - _countdown ) )
|
|
return self.make_request( url, data, method, _countdown - 1 )
|
|
else:
|
|
raise
|
|
if request.ok:
|
|
try:
|
|
return request.json()
|
|
except:
|
|
return None
|
|
else:
|
|
raise ValueError( request.text )
|
|
|
|
def get_one( self, url, no_d = False ):
|
|
return self.make_request( url, None, 'GET' )
|
|
|
|
def get_many( self, url, count = 10, no_d = False ):
|
|
next = None
|
|
url = self.normalize_url( url, no_d = no_d, count = count )
|
|
|
|
while True:
|
|
result = self.make_request( url, None, 'GET' )
|
|
if 'results' in result:
|
|
for object in result[ 'results' ]:
|
|
yield object
|
|
if 'next' in result:
|
|
next = result.get( 'next', None )
|
|
if next:
|
|
url = next
|
|
else:
|
|
return
|