diff options
author | Kali Kaneko <kali@leap.se> | 2015-08-05 18:54:58 -0700 |
---|---|---|
committer | Ivan Alejandro <ivanalejandro0@gmail.com> | 2015-08-07 19:02:24 -0300 |
commit | 657005c72abdd478d1f4809e9984d401cbbbb845 (patch) | |
tree | e9572880f02624bab24f2e01cead83153d815abf /src/leap | |
parent | 877e92d8698510b243b32d45e9c46f25e6f04615 (diff) |
[feat] add util function fo find location when frozen
Diffstat (limited to 'src/leap')
-rw-r--r-- | src/leap/bitmask/util/__init__.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/leap/bitmask/util/__init__.py b/src/leap/bitmask/util/__init__.py index e8eddd64..9853803a 100644 --- a/src/leap/bitmask/util/__init__.py +++ b/src/leap/bitmask/util/__init__.py @@ -20,6 +20,7 @@ Some small and handy functions. import datetime import itertools import os +import sys from leap.bitmask.config import flags from leap.common.config import get_path_prefix as common_get_path_prefix @@ -154,3 +155,14 @@ def flags_to_dict(): values = dict((i, getattr(flags, i)) for i in items) return values + +def here(module=None): + if getattr(sys, 'frozen', False): + # we are running in a |PyInstaller| bundle + return sys._MEIPASS + else: + dirname = os.path.dirname + if module: + return dirname(module.__file__) + else: + return dirname(__file__) |