addresswatcher.py

import datetime

import gevent
import requests
import base58

class AddressWatcher(gevent.Greenlet):

    addresses = {}

    def __init__(self, testnet=True):
        gevent.Greenlet.__init__(self)

        self.transfer_tx_callback = None
        self.testnet = testnet
        if testnet:
            self.url_base = "https://api-test.wavesplatform.com/v0"
            self.asset_id = "CgUrFtinLXEbJwJVjwwcppk4Vpz1nMmR3H5cQaDcUcfe"
        else:
            self.url_base = "https://api.wavesplatform.com/v0"
            self.asset_id = "9R3iLi4qGLVWKc16Tg98gmRvgg1usGEYd7SgC1W5D6HB"

    def _run(self):
        print("running AddressWatcher...")
        dt = datetime.datetime.utcnow()
        js_datestring = dt.strftime("%Y-%m-%dT%H:%M:%SZ")
        after = None
        last = True
        while 1:
            # poll for more transactions
            url = self.url_base + "/transactions/transfer"
            params = {"assetId": self.asset_id, "timeStart": js_datestring, "sort": "asc"}
            if after:
                params["after"] = after
            r = requests.get(url, params=params)
            if r.status_code == 200:
                body = r.json()
                for tx in body["data"]:
                    tx = tx["data"]
                    if tx["recipient"] in self.addresses:
                        if tx["attachment"]:
                            tx["attachment"] = base58.b58decode(tx["attachment"]).decode("utf-8")
                        tokens = self.addresses[tx["recipient"]]
                        if self.transfer_tx_callback:
                            self.transfer_tx_callback(tokens, tx)
                if "lastCursor" in body:
                    after = body["lastCursor"]
                if "isLastPage" in body:
                    last = body["isLastPage"]
            else:
                #TODO log error
                print(r)
            # sleep
            gevent.sleep(5)

    def watch(self, address, token):
        if not address in self.addresses:
            self.addresses[address] = tokens = [] 
        else:
            tokens = self.addresses[address]
        if token not in tokens:
            tokens.append(token)

    def watched(self):
        return self.addresses

    def transfer_tx(self, txid):
        url = self.url_base + "/transactions/transfer/" + txid
        r = requests.get(url)
        if r.status_code == 200:
            body = r.json()
            return body["data"]
        return None