file reorganization with top-level module

electrum/contacts.py

@@ -0,0 +1,135 @@
+# Electrum - Lightweight Bitcoin Client
+# Copyright (c) 2015 Thomas Voegtlin
+#
+# Permission is hereby granted, free of charge, to any person
+# obtaining a copy of this software and associated documentation files
+# (the "Software"), to deal in the Software without restriction,
+# including without limitation the rights to use, copy, modify, merge,
+# publish, distribute, sublicense, and/or sell copies of the Software,
+# and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+import re
+import dns
+from dns.exception import DNSException
+import json
+import traceback
+import sys
+
+from . import bitcoin
+from . import dnssec
+from .util import export_meta, import_meta, print_error, to_string
+
+
+class Contacts(dict):
+
+    def __init__(self, storage):
+        self.storage = storage
+        d = self.storage.get('contacts', {})
+        try:
+            self.update(d)
+        except:
+            return
+        # backward compatibility
+        for k, v in self.items():
+            _type, n = v
+            if _type == 'address' and bitcoin.is_address(n):
+                self.pop(k)
+                self[n] = ('address', k)
+
+    def save(self):
+        self.storage.put('contacts', dict(self))
+
+    def import_file(self, path):
+        import_meta(path, self._validate, self.load_meta)
+
+    def load_meta(self, data):
+        self.update(data)
+        self.save()
+
+    def export_file(self, filename):
+        export_meta(self, filename)
+
+    def __setitem__(self, key, value):
+        dict.__setitem__(self, key, value)
+        self.save()
+
+    def pop(self, key):
+        if key in self.keys():
+            dict.pop(self, key)
+            self.save()
+
+    def resolve(self, k):
+        if bitcoin.is_address(k):
+            return {
+                'address': k,
+                'type': 'address'
+            }
+        if k in self.keys():
+            _type, addr = self[k]
+            if _type == 'address':
+                return {
+                    'address': addr,
+                    'type': 'contact'
+                }
+        out = self.resolve_openalias(k)
+        if out:
+            address, name, validated = out
+            return {
+                'address': address,
+                'name': name,
+                'type': 'openalias',
+                'validated': validated
+            }
+        raise Exception("Invalid Bitcoin address or alias", k)
+
+    def resolve_openalias(self, url):
+        # support email-style addresses, per the OA standard
+        url = url.replace('@', '.')
+        try:
+            records, validated = dnssec.query(url, dns.rdatatype.TXT)
+        except DNSException as e:
+            print_error('Error resolving openalias: ', str(e))
+            return None
+        prefix = 'btc'
+        for record in records:
+            string = to_string(record.strings[0], 'utf8')
+            if string.startswith('oa1:' + prefix):
+                address = self.find_regex(string, r'recipient_address=([A-Za-z0-9]+)')
+                name = self.find_regex(string, r'recipient_name=([^;]+)')
+                if not name:
+                    name = address
+                if not address:
+                    continue
+                return address, name, validated
+
+    def find_regex(self, haystack, needle):
+        regex = re.compile(needle)
+        try:
+            return regex.search(haystack).groups()[0]
+        except AttributeError:
+            return None
+            
+    def _validate(self, data):
+        for k, v in list(data.items()):
+            if k == 'contacts':
+                return self._validate(v)
+            if not bitcoin.is_address(k):
+                data.pop(k)
+            else:
+                _type, _ = v
+                if _type != 'address':
+                    data.pop(k)
+        return data
+