Update device.py

DragonBuild · Aug 26, 2021 · 79342db · 79342db
1 parent 00479bc
commit 79342db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dragontools/device.py b/src/dragontools/device.py
@@ -96,7 +96,7 @@ def setup_key_auth(self):
 class DeviceManager(object):
 
     def __init__(self):
-        with open(f'{os.environ["DRAGONBUILD"]}/internal/state.yml') as state:
+        with open(f'{os.environ["DRAGONDIR"]}/internal/state.yml') as state:
             dragon_state = yaml.safe_load(state)
 
         self.dragon_state = dragon_state
@@ -108,7 +108,7 @@ def __init__(self):
         self.current = self.devices[dragon_state['device']['current']]
 
     def savestate(self):
-        with open(f'{os.environ["DRAGONBUILD"]}/internal/state.yml', 'w') as state:
+        with open(f'{os.environ["DRAGONDIR"]}/internal/state.yml', 'w') as state:
             yaml.dump(self.dragon_state, state)
 
     def add_device(self, device: Device):