- environment = os.environ['NODE_NAME']
- assert environment
-
- # API URL
- energy_recorder_uri = os.environ["ENERGY_RECORDER_API_URL"]
- assert energy_recorder_uri
+ assert env.get('NODE_NAME')
+ assert env.get('ENERGY_RECORDER_API_URL')
+ environment = env.get('NODE_NAME')
+ energy_recorder_uri = env.get(
+ 'ENERGY_RECORDER_API_URL')