external api testnet support

Author: andrewrothman

blockchain_utils.py

@@ -5,6 +5,21 @@
 from config import BTAPIKEY
 from rpcclient import gettxout
 from cacher import *
+import config
+
+BLOCKCHAININFO_API_URL = "https://blockchain.info/"
+BLOCKR_API_URL = "http://btc.blockr.io/api/v1"
+BLOCKTRAIL_API_URL = "https://api.blocktrail.com/v1/btc/"
+BLOCKCYPHER_API_URL = "https://api.blockcypher.com/v1/btc/main/"
+BITGO_API_URL = "https://www.bitgo.com/api/v1/"
+BLOCKONOMICS_API_URL = "https://www.blockonomics.co/api"
+
+# note: "blockchain.info" does not support testnet
+# note: "blockonomics" does not appear to support testnet
+TESTNET_BLOCKR_API_URL = "http://tbtc.blockr.io/api/v1"
+TESTNET_BLOCKTRAIL_API_URL = "https://api.blocktrail.com/v1/tbtc/"
+TESTNET_BLOCKCYPHER_API_URL = "https://api.blockcypher.com/v1/btc/test3/"
+TESTNET_BITGO_API_URL = "https://test.bitgo.com/api/v1/"
 
 try:
   expTime=config.BTCBAL_CACHE
@@ -15,7 +30,8 @@ def bc_getutxo(address, ramount, page=1, retval=None, avail=0):
   if retval==None:
     retval=[]
   try:
-    r = requests.get('https://api.blocktrail.com/v1/btc/address/'+address+'/unspent-outputs?api_key='+str(BTAPIKEY)+'&limit=200&page='+str(page))
+    apiUrl = BLOCKTRAIL_API_URL if not config.TESTNET else TESTNET_BLOCKTRAIL_API_URL
+    r = requests.get(apiUrlr + '/address/'+address+'/unspent-outputs?api_key='+str(BTAPIKEY)+'&limit=200&page='+str(page))
     if r.status_code == 200:
       response = r.json()
       unspents = response['data']
@@ -40,7 +56,8 @@ def bc_getutxo(address, ramount, page=1, retval=None, avail=0):
 
 def bc_getutxo_blockcypher(address, ramount):
   try:
-    r = requests.get('https://api.blockcypher.com/v1/btc/main/addrs/'+address+'?unspentOnly=true')
+    apiUrl = BLOCKCYPHER_API_URL if not config.TESTNET else TESTNET_BLOCKCYPHER_API_URL
+    r = requests.get(apiUrl + '/addrs/'+address+'?unspentOnly=true')
 
     if r.status_code == 200:
       unspents = r.json()['txrefs']
@@ -65,7 +82,8 @@ def bc_getutxo_blockcypher(address, ramount):
 
 def bc_getutxo_blockr(address, ramount):
   try:
-    r = requests.get('http://btc.blockr.io/api/v1/address/unspent/'+address+'?unconfirmed=1')
+    apiUrl = BLOCKR_API_URL if not config.TESTNET else TESTNET_BLOCKR_API_URL
+    r = requests.get(apiUrl + '/address/unspent/'+address+'?unconfirmed=1')
 
     if r.status_code == 200:
       #Process and format response from blockr.io
@@ -92,8 +110,12 @@ def bc_getutxo_blockr(address, ramount):
 
 
 def bc_getpubkey(address):
+  # note: only supports mainnet
   try:
-    r = requests.get('https://blockchain.info/q/pubkeyaddr/'+address)
+    if config.TESTNET:
+      return "error: tried using blockchain.info api with testnet enabled"
+    
+    r = requests.get(BLOCKCHAININFO_API_URL + '/q/pubkeyaddr/'+address)
 
     if r.status_code == 200:
       return str(r.text)
@@ -117,7 +139,8 @@ def bc_getbalance(address):
 
 def bc_getbalance_bitgo(address):
   try:
-    r= requests.get('https://www.bitgo.com/api/v1/address/'+address)
+    apiUrl = BITGO_API_URL if not config.TESTNET else TESTNET_BITGO_API_URL
+    r= requests.get(apiUrl + '/address/'+address)
     if r.status_code == 200:
       balance = int(r.json()['balance'])
       return {"bal":balance , "error": None}
@@ -128,7 +151,8 @@ def bc_getbalance_bitgo(address):
 
 def bc_getbalance_blockcypher(address):
   try:
-    r= requests.get('https://api.blockcypher.com/v1/btc/main/addrs/'+address+'/balance')
+    apiUrl = BLOCKCYPHER_API_URL if not config.TESTNET else TESTNET_BLOCKCYPHER_API_URL
+    r= requests.get(apiUrl + '/addrs/'+address+'/balance')
     if r.status_code == 200:
       balance = int(r.json()['balance'])
       return {"bal":balance , "error": None}
@@ -139,7 +163,8 @@ def bc_getbalance_blockcypher(address):
 
 def bc_getbalance_blockr(address):
   try:
-    r= requests.get('http://btc.blockr.io/api/v1/address/balance/'+address)
+    apiUrl = BLOCKR_API_URL if not config.TESTNET else TESTNET_BLOCKR_API_URL
+    r= requests.get(apiUrl + '/address/balance/'+address)
     if r.status_code == 200:
       balance = int(r.json()['data']['balance']*1e8)
       return {"bal":balance , "error": None}
@@ -222,7 +247,10 @@ def bc_getbulkbalance_blockonomics(addresses):
       formatted=formatted+" "+address
 
   try:
-    r = requests.post('https://www.blockonomics.co/api/balance',json.dumps({"addr":formatted}))
+    if config.TESTNET:
+      return "error: tried using blockonomics api with testnet enabled"
+    
+    r = requests.post(apiUrl + '/balance',json.dumps({"addr":formatted}))
     if r.status_code == 200:
       balances = r.json()['response']
       retval = {}
@@ -243,7 +271,8 @@ def bc_getbulkbalance_blockr(addresses):
       formatted=formatted+","+address
 
   try:
-    r= requests.get('http://btc.blockr.io/api/v1/address/balance/'+formatted)
+    apiUrl = BLOCKR_API_URL if not config.TESTNET else TESTNET_BLOCKR_API_URL
+    r= requests.get(apiUrl + '/address/balance/'+formatted)
     if r.status_code == 200:
       balances = r.json()['data']
       retval = {}
@@ -263,7 +292,10 @@ def bc_getbulkbalance_blockchain(addresses):
     else:
       formatted=formatted+"|"+address
   try:
-    r= requests.get('https://blockchain.info/balance?active='+formatted)
+    if config.TESTNET:
+      return "error: tried using blockchain.info api with testnet enabled"
+    
+    r= requests.get(BLOCKCHAININFO_API_URL + '/balance?active='+formatted)
     if r.status_code == 200:
       balances = r.json()
       retval = {}

config.py.example

@@ -9,3 +9,5 @@ REDIS_DB=0
 #How long, in seconds, to cache BTC balance info for new addresses, Default 10min (600)
 BTCBAL_CACHE=600
 
+# Set to True to switch to processing testnet 
+TESTNET = False

omniEngine.py

@@ -3,11 +3,11 @@
 import sys
 from datetime import datetime
 from cacher import *
+import config
 
 USER=getpass.getuser()
 lockFile='/tmp/omniEngine.lock.'+str(USER)
 now=datetime.now()
-testnet=0
 sys.argv.pop(0)
 
 if os.path.isfile(lockFile):
@@ -176,7 +176,7 @@
       #check any active crowdsales and update json if the endtime has passed (based on block time)
       expireCrowdsales(block_data['result']['time'], Protocol)
       #exodus address generates dev msc, sync our balance to match the generated balanace
-      if testnet:
+      if config.TESTNET:
         syncAddress('mpexoDuSkGGqvqrkrjiFng38QPkJQVFyqv', Protocol)
         #upadate temp orderbook
         #updateorderblob()