Improve dunikey path loading

This commit is contained in:
poka 2020-12-02 07:36:19 +01:00
parent 62e776be7e
commit f39be0e03b
3 changed files with 19 additions and 3 deletions

View File

@ -19,6 +19,12 @@ pod = os.getenv('POD')
if not dunikey or not pod:
sys.stderr.write("Please fill the path of your private key (PubSec), and a Cesium ES address in .env file\n")
sys.exit(1)
if not os.path.isfile(dunikey):
HOME = os.getenv("HOME")
dunikey = HOME + os.getenv('DUNIKEY')
if not os.path.isfile(dunikey):
sys.stderr.write("File {0} doesn't exist.\n".format(dunikey))
sys.exit(1)
# Parse arguments
parser = argparse.ArgumentParser()

View File

@ -9,18 +9,22 @@ from lib.cesium import ReadFromCesium, SendToCesium, DeleteFromCesium
VERSION = "0.1.1"
# Get variables environment
HOME = os.getenv("HOME")
if not os.path.isfile('.env'):
copyfile(".env.template", ".env")
dotenv_path = join(dirname(__file__), '.env')
load_dotenv(dotenv_path)
dunikey = HOME + os.getenv('DUNIKEY')
dunikey = os.getenv('DUNIKEY')
pod = os.getenv('POD')
if not dunikey or not pod:
sys.stderr.write("Please fill the path of your private key (PubSec), and a Cesium ES address in .env file\n")
sys.exit(1)
if not os.path.isfile(dunikey):
HOME = os.getenv("HOME")
dunikey = HOME + os.getenv('DUNIKEY')
if not os.path.isfile(dunikey):
sys.stderr.write("File {0} doesn't exist.\n".format(dunikey))
sys.exit(1)
# Parse arguments
parser = argparse.ArgumentParser()

View File

@ -17,6 +17,12 @@ pod = os.getenv('POD')
if not dunikey or not pod:
sys.stderr.write("Please fill the path of your private key (PubSec), and a Cesium ES address in .env file\n")
sys.exit(1)
if not os.path.isfile(dunikey):
HOME = os.getenv("HOME")
dunikey = HOME + os.getenv('DUNIKEY')
if not os.path.isfile(dunikey):
sys.stderr.write("File {0} doesn't exist.\n".format(dunikey))
sys.exit(1)
# Parse arguments