added erc20 and sourcecode functionality, added some tests and examples

Author: corpetty

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "/usr/bin/python3.6"
+}

changelog.md

@@ -25,7 +25,7 @@ def get_balance_multiple(self):
         return req['result']
 
     def get_transaction_page(self, page=1, offset=10000, sort='asc',
-                             internal=False) -> list:
+                             internal=False, erc20=False) -> list:
         """
         Get a page of transactions, each transaction
         returns list of dict with keys:
@@ -52,12 +52,20 @@ def get_transaction_page(self, page=1, offset=10000, sort='asc',
             'asc' -> ascending order
             'desc' -> descending order
 
-        internal options:
-            True  -> Gets the internal transactions of a smart contract
+        internal options: (currently marked at Beta for etherscan.io)
+            True  -> Gets the internal transactions of the address
+            False -> (default) get normal external transactions
+
+        erc20 options: (currently marked at Beta for etherscan.io)
+            True  -> Gets the erc20 token transcations of the address
             False -> (default) get normal external transactions
+
+        NOTE: not sure if this works for contract addresses, requires testing
         """
         if internal:
             self.url_dict[self.ACTION] = 'txlistinternal'
+        elif erc20:
+            self.url_dict[self.ACTION] = 'tokentx'
         else:
             self.url_dict[self.ACTION] = 'txlist'
         self.url_dict[self.PAGE] = str(page)

etherscan/accounts.py

@@ -11,3 +11,9 @@ def get_abi(self):
         self.build_url()
         req = self.connect()
         return req['result']
+
+    def get_sourcecode(self):
+        self.url_dict[self.ACTION] = 'getsourcecode'
+        self.build_url()
+        req = self.connect()
+        return req['result']

etherscan/contracts.py

@@ -0,0 +1,11 @@
+from etherscan.accounts import Account
+import json
+
+with open('../../api_key.json', mode='r') as key_file:
+    key = json.loads(key_file.read())['key']
+
+address = '0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a'
+
+api = Account(address=address, api_key=key)
+transactions = api.get_transaction_page(page=1, offset=10000, sort='des', erc20=True)
+print(transactions)

examples/accounts/Accounts Examples Notebook.ipynb

@@ -0,0 +1,12 @@
+from etherscan.contracts import Contract
+import json
+
+with open('../../api_key.json', mode='r') as key_file:
+    key = json.loads(key_file.read())['key']
+
+address = '0xfb6916095ca1df60bb79ce92ce3ea74c37c5d359'
+
+api = Contract(address=address, api_key=key)
+sourcecode = api.get_sourcecode()
+# TODO: make this return something pretty
+print(sourcecode[0]['SourceCode'])

examples/accounts/get_transaction_page_erc20.py

@@ -0,0 +1,25 @@
+import unittest
+
+from etherscan.accounts import Account
+
+SINGLE_BALANCE = '40807168566070000000000'
+SINGLE_ACCOUNT = '0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a'
+MULTI_ACCOUNT = ['0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a',
+           '0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a']
+MULTI_BALANCE = [
+    {'account': '0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a', 
+     'balance': '40807168566070000000000'},
+    {'account': '0xddbd2b932c763ba5b1b7ae3b362eac3e8d40121a', 
+    'balance': '40807168566070000000000'}
+]
+API_KEY = 'YourAPIkey'
+
+class AccountsTestCase(unittest.TestCase):
+
+    def test_get_balance(self):
+        api = Account(address=SINGLE_ACCOUNT, api_key=API_KEY)
+        self.assertEqual(api.get_balance(), SINGLE_BALANCE)
+
+    def test_get_balance_multi(self):
+        api = Account(address=MULTI_ACCOUNT, api_key=API_KEY)
+        self.assertEqual(api.get_balance_multiple(), MULTI_BALANCE)