lib/util.py

   1 import os, sys
   2 import platform
   3
   4 is_verbose = True
   5
   6 def set_verbosity(b):
   7     global is_verbose
   8     is_verbose = b
   9
  10 def print_error(*args):
  11     if not is_verbose: return
  12     print_msg(*args)
  13
  14 def print_msg(*args):
  15     # Stringify args
  16     args = [str(item) for item in args]
  17     sys.stderr.write(" ".join(args) + "\n")
  18     sys.stderr.flush()
  19
  20
  21 def user_dir():
  22     if "HOME" in os.environ:
  23         return os.path.join(os.environ["HOME"], ".electrum")
  24     elif "LOCALAPPDATA" in os.environ:
  25         return os.path.join(os.environ["LOCALAPPDATA"], "Electrum")
  26     elif "APPDATA" in os.environ:
  27         return os.path.join(os.environ["APPDATA"], "Electrum")
  28     else:
  29         #raise BaseException("No home directory found in environment variables.")
  30         return
  31
  32 def appdata_dir():
  33     """Find the path to the application data directory; add an electrum folder and return path."""
  34     if platform.system() == "Windows":
  35         return os.path.join(os.environ["APPDATA"], "Electrum")
  36     elif platform.system() == "Linux":
  37         return os.path.join(sys.prefix, "share", "electrum")
  38     elif (platform.system() == "Darwin" or
  39           platform.system() == "DragonFly"):
  40         return "/Library/Application Support/Electrum"
  41     else:
  42         raise Exception("Unknown system")
  43
  44
  45 def get_resource_path(*args):
  46     return os.path.join(".", *args)
  47
  48
  49 def local_data_dir():
  50     """Return path to the data folder."""
  51     assert sys.argv
  52     prefix_path = os.path.dirname(sys.argv[0])
  53     local_data = os.path.join(prefix_path, "data")
  54     return local_data
  55
  56
  57 def format_satoshis(x, is_diff=False, num_zeros = 0):
  58     from decimal import Decimal
  59     s = Decimal(x)
  60     sign, digits, exp = s.as_tuple()
  61     digits = map(str, digits)
  62     while len(digits) < 9:
  63         digits.insert(0,'0')
  64     digits.insert(-8,'.')
  65     s = ''.join(digits).rstrip('0')
  66     if sign:
  67         s = '-' + s
  68     elif is_diff:
  69         s = "+" + s
  70
  71     p = s.find('.')
  72     s += "0"*( 1 + num_zeros - ( len(s) - p ))
  73     s += " "*( 9 - ( len(s) - p ))
  74     s = " "*( 5 - ( p )) + s
  75     return s