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     print_msg(*args)
  62
  63 def print_msg(*args):
  64     # Stringify args
  65     args = [str(item) for item in args]
  66     sys.stderr.write(" ".join(args) + "\n")
  67     sys.stderr.flush()
  68
  69
  70 def user_dir():
  71     if "HOME" in os.environ:
  72         return os.path.join(os.environ["HOME"], ".electrum")
  73     elif "LOCALAPPDATA" in os.environ:
  74         return os.path.join(os.environ["LOCALAPPDATA"], "Electrum")
  75     elif "APPDATA" in os.environ:
  76         return os.path.join(os.environ["APPDATA"], "Electrum")
  77     else:
  78         #raise BaseException("No home directory found in environment variables.")
  79         return
  80
  81 def appdata_dir():
  82     """Find the path to the application data directory; add an electrum folder and return path."""
  83     if platform.system() == "Windows":
  84         return os.path.join(os.environ["APPDATA"], "Electrum")
  85     elif platform.system() == "Linux":
  86         return os.path.join(sys.prefix, "share", "electrum")
  87     elif (platform.system() == "Darwin" or
  88           platform.system() == "DragonFly"):
  89         return "/Library/Application Support/Electrum"
  90     else:
  91         raise Exception("Unknown system")
  92
  93
  94 def get_resource_path(*args):
  95     return os.path.join(".", *args)
  96
  97
  98 def local_data_dir():
  99     """Return path to the data folder."""
 100     assert sys.argv
 101     prefix_path = os.path.dirname(sys.argv[0])
 102     local_data = os.path.join(prefix_path, "data")
 103     return local_data
 104
 105
 106 def format_satoshis(x, is_diff=False, num_zeros = 0):
 107     from decimal import Decimal
 108     s = Decimal(x)
 109     sign, digits, exp = s.as_tuple()
 110     digits = map(str, digits)
 111     while len(digits) < 9:
 112         digits.insert(0,'0')
 113     digits.insert(-8,'.')
 114     s = ''.join(digits).rstrip('0')
 115     if sign:
 116         s = '-' + s
 117     elif is_diff:
 118         s = "+" + s
 119
 120     p = s.find('.')
 121     s += "0"*( 1 + num_zeros - ( len(s) - p ))
 122     s += " "*( 9 - ( len(s) - p ))
 123     s = " "*( 5 - ( p )) + s
 124     return s