X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=lib%2Futil.py;h=0415b1ce2b693e39cef81197d7f7b2196927e848;hb=3cbe11a42473af52e7c5e002c36aaaf32646f627;hp=33cfaef410e384f7fe81010d432ff34c468da700;hpb=72358fbc20c36e4e268d247bded4a97826e4e966;p=electrum-nvc.git diff --git a/lib/util.py b/lib/util.py index 33cfaef..0415b1c 100644 --- a/lib/util.py +++ b/lib/util.py @@ -1,43 +1,71 @@ -import os, sys +import os, sys, re, json import platform +import shutil +from datetime import datetime +is_verbose = False + + +class MyEncoder(json.JSONEncoder): + def default(self, obj): + from transaction import Transaction + if isinstance(obj, Transaction): + return obj.as_dict() + return super(MyEncoder, self).default(obj) -is_verbose = True def set_verbosity(b): global is_verbose is_verbose = b + def print_error(*args): if not is_verbose: return - print_msg(*args) + print_stderr(*args) -def print_msg(*args): - # Stringify args +def print_stderr(*args): args = [str(item) for item in args] sys.stderr.write(" ".join(args) + "\n") sys.stderr.flush() +def print_msg(*args): + # Stringify args + args = [str(item) for item in args] + sys.stdout.write(" ".join(args) + "\n") + sys.stdout.flush() + +def print_json(obj): + try: + s = json.dumps(obj, sort_keys = True, indent = 4, cls=MyEncoder) + except TypeError: + s = repr(obj) + sys.stdout.write(s + "\n") + sys.stdout.flush() def user_dir(): if "HOME" in os.environ: - return os.path.join(os.environ["HOME"], ".electrum") - elif "LOCALAPPDATA" in os.environ: - return os.path.join(os.environ["LOCALAPPDATA"], "Electrum") + return os.path.join(os.environ["HOME"], ".electrum-nvc") elif "APPDATA" in os.environ: - return os.path.join(os.environ["APPDATA"], "Electrum") + return os.path.join(os.environ["APPDATA"], "Electrum-NVC") + elif "LOCALAPPDATA" in os.environ: + return os.path.join(os.environ["LOCALAPPDATA"], "Electrum-NVC") + elif 'ANDROID_DATA' in os.environ: + return "/sdcard/electrum/" else: - #raise BaseException("No home directory found in environment variables.") - return + #raise Exception("No home directory found in environment variables.") + return def appdata_dir(): """Find the path to the application data directory; add an electrum folder and return path.""" if platform.system() == "Windows": - return os.path.join(os.environ["APPDATA"], "Electrum") + return os.path.join(os.environ["APPDATA"], "Electrum-NVC") elif platform.system() == "Linux": - return os.path.join(sys.prefix, "share", "electrum") + return os.path.join(sys.prefix, "share", "electrum-nvc") elif (platform.system() == "Darwin" or - platform.system() == "DragonFly"): - return "/Library/Application Support/Electrum" + platform.system() == "DragonFly" or + platform.system() == "OpenBSD" or + platform.system() == "FreeBSD" or + platform.system() == "NetBSD"): + return "/Library/Application Support/Electrum-NVC" else: raise Exception("Unknown system") @@ -54,22 +82,132 @@ def local_data_dir(): return local_data -def format_satoshis(x, is_diff=False, num_zeros = 0): +def format_satoshis(x, is_diff=False, num_zeros = 0, decimal_point = 6, whitespaces=False): from decimal import Decimal s = Decimal(x) sign, digits, exp = s.as_tuple() digits = map(str, digits) - while len(digits) < 9: + while len(digits) < decimal_point + 1: digits.insert(0,'0') - digits.insert(-8,'.') + digits.insert(-decimal_point,'.') s = ''.join(digits).rstrip('0') - if sign: + if sign: s = '-' + s elif is_diff: s = "+" + s p = s.find('.') s += "0"*( 1 + num_zeros - ( len(s) - p )) - s += " "*( 9 - ( len(s) - p )) - s = " "*( 5 - ( p )) + s + if whitespaces: + s += " "*( 1 + decimal_point - ( len(s) - p )) + s = " "*( 13 - decimal_point - ( p )) + s return s + + +# Takes a timestamp and returns a string with the approximation of the age +def age(from_date, since_date = None, target_tz=None, include_seconds=False): + if from_date is None: + return "Unknown" + + from_date = datetime.fromtimestamp(from_date) + if since_date is None: + since_date = datetime.now(target_tz) + + distance_in_time = since_date - from_date + distance_in_seconds = int(round(abs(distance_in_time.days * 86400 + distance_in_time.seconds))) + distance_in_minutes = int(round(distance_in_seconds/60)) + + if distance_in_minutes <= 1: + if include_seconds: + for remainder in [5, 10, 20]: + if distance_in_seconds < remainder: + return "less than %s seconds ago" % remainder + if distance_in_seconds < 40: + return "half a minute ago" + elif distance_in_seconds < 60: + return "less than a minute ago" + else: + return "1 minute ago" + else: + if distance_in_minutes == 0: + return "less than a minute ago" + else: + return "1 minute ago" + elif distance_in_minutes < 45: + return "%s minutes ago" % distance_in_minutes + elif distance_in_minutes < 90: + return "about 1 hour ago" + elif distance_in_minutes < 1440: + return "about %d hours ago" % (round(distance_in_minutes / 60.0)) + elif distance_in_minutes < 2880: + return "1 day ago" + elif distance_in_minutes < 43220: + return "%d days ago" % (round(distance_in_minutes / 1440)) + elif distance_in_minutes < 86400: + return "about 1 month ago" + elif distance_in_minutes < 525600: + return "%d months ago" % (round(distance_in_minutes / 43200)) + elif distance_in_minutes < 1051200: + return "about 1 year ago" + else: + return "over %d years ago" % (round(distance_in_minutes / 525600)) + + +# URL decode +#_ud = re.compile('%([0-9a-hA-H]{2})', re.MULTILINE) +#urldecode = lambda x: _ud.sub(lambda m: chr(int(m.group(1), 16)), x) + +def parse_URI(uri): + import urlparse + import bitcoin + from decimal import Decimal + + if ':' not in uri: + assert bitcoin.is_address(uri) + return uri, None, None, None, None + + u = urlparse.urlparse(uri) + assert u.scheme == 'bitcoin' + + address = u.path + valid_address = bitcoin.is_address(address) + + pq = urlparse.parse_qs(u.query) + + for k, v in pq.items(): + if len(v)!=1: + raise Exception('Duplicate Key', k) + + amount = label = message = request_url = '' + if 'amount' in pq: + am = pq['amount'][0] + m = re.match('([0-9\.]+)X([0-9])', am) + if m: + k = int(m.group(2)) - 8 + amount = Decimal(m.group(1)) * pow( Decimal(10) , k) + else: + amount = Decimal(am) * 1000000 + if 'message' in pq: + message = pq['message'][0] + if 'label' in pq: + label = pq['label'][0] + if 'r' in pq: + request_url = pq['r'][0] + + if request_url != '': + return address, amount, label, message, request_url + + assert valid_address + + return address, amount, label, message, request_url + + +# Python bug (http://bugs.python.org/issue1927) causes raw_input +# to be redirected improperly between stdin/stderr on Unix systems +def raw_input(prompt=None): + if prompt: + sys.stdout.write(prompt) + return builtin_raw_input() +import __builtin__ +builtin_raw_input = __builtin__.raw_input +__builtin__.raw_input = raw_input