lib/util.py

   1 import os, sys
   2 import platform
   3 from datetime import datetime
   4 is_verbose = True
   5
   6 # Takes a timestamp and puts out a string with the approxomation of the age
   7 def age(from_date, since_date = None, target_tz=None, include_seconds=False):
   8   if from_date is None:
   9     return "Unknown"
  10
  11   from_date = datetime.fromtimestamp(from_date)
  12   if since_date is None:
  13     since_date = datetime.now(target_tz)
  14
  15   distance_in_time = since_date - from_date
  16   distance_in_seconds = int(round(abs(distance_in_time.days * 86400 + distance_in_time.seconds)))
  17   distance_in_minutes = int(round(distance_in_seconds/60))
  18
  19
  20   if distance_in_minutes <= 1:
  21     if include_seconds:
  22       for remainder in [5, 10, 20]:
  23         if distance_in_seconds < remainder:
  24           return "less than %s seconds ago" % remainder
  25         if distance_in_seconds < 40:
  26           return "half a minute"
  27         elif distance_in_seconds < 60:
  28           return "less than a minute ago"
  29         else:
  30           return "1 minute ago"
  31       else:
  32         if distance_in_minutes == 0:
  33           return "less than a minute ago"
  34         else:
  35           return "1 minute ago"
  36   elif distance_in_minutes < 45:
  37     return "%s minutes ago" % distance_in_minutes
  38   elif distance_in_minutes < 90:
  39     return "about 1 hour ago"
  40   elif distance_in_minutes < 1440:
  41     return "about %d hours ago" % (round(distance_in_minutes / 60.0))
  42   elif distance_in_minutes < 2880:
  43     return "1 day ago"
  44   elif distance_in_minutes < 43220:
  45     return "%d days ago" % (round(distance_in_minutes / 1440))
  46   elif distance_in_minutes < 86400:
  47     return "about 1 month ago"
  48   elif distance_in_minutes < 525600:
  49     return "%d months ago" % (round(distance_in_minutes / 43200))
  50   elif distance_in_minutes < 1051200:
  51     return "about 1 year ago"
  52   else:
  53     return "over %d years ago" % (round(distance_in_minutes / 525600))
  54
  55 def set_verbosity(b):
  56     global is_verbose
  57     is_verbose = b
  58
  59 def print_error(*args):
  60     if not is_verbose: return
  61     args = [str(item) for item in args]
  62     sys.stderr.write(" ".join(args) + "\n")
  63     sys.stderr.flush()
  64
  65 def print_msg(*args):
  66     # Stringify args
  67     args = [str(item) for item in args]
  68     sys.stdout.write(" ".join(args) + "\n")
  69     sys.stdout.flush()
  70
  71
  72 def user_dir():
  73     if "HOME" in os.environ:
  74         return os.path.join(os.environ["HOME"], ".electrum")
  75     elif "LOCALAPPDATA" in os.environ:
  76         return os.path.join(os.environ["LOCALAPPDATA"], "Electrum")
  77     elif "APPDATA" in os.environ:
  78         return os.path.join(os.environ["APPDATA"], "Electrum")
  79     else:
  80         #raise BaseException("No home directory found in environment variables.")
  81         return
  82
  83 def appdata_dir():
  84     """Find the path to the application data directory; add an electrum folder and return path."""
  85     if platform.system() == "Windows":
  86         return os.path.join(os.environ["APPDATA"], "Electrum")
  87     elif platform.system() == "Linux":
  88         return os.path.join(sys.prefix, "share", "electrum")
  89     elif (platform.system() == "Darwin" or
  90           platform.system() == "DragonFly" or
  91           platform.system() == "NetBSD"):
  92         return "/Library/Application Support/Electrum"
  93     else:
  94         raise Exception("Unknown system")
  95
  96
  97 def get_resource_path(*args):
  98     return os.path.join(".", *args)
  99
 100
 101 def local_data_dir():
 102     """Return path to the data folder."""
 103     assert sys.argv
 104     prefix_path = os.path.dirname(sys.argv[0])
 105     local_data = os.path.join(prefix_path, "data")
 106     return local_data
 107
 108
 109 def format_satoshis(x, is_diff=False, num_zeros = 0):
 110     from decimal import Decimal
 111     s = Decimal(x)
 112     sign, digits, exp = s.as_tuple()
 113     digits = map(str, digits)
 114     while len(digits) < 9:
 115         digits.insert(0,'0')
 116     digits.insert(-8,'.')
 117     s = ''.join(digits).rstrip('0')
 118     if sign:
 119         s = '-' + s
 120     elif is_diff:
 121         s = "+" + s
 122
 123     p = s.find('.')
 124     s += "0"*( 1 + num_zeros - ( len(s) - p ))
 125     s += " "*( 9 - ( len(s) - p ))
 126     s = " "*( 5 - ( p )) + s
 127     return s