droplet.py: Override set_common_state to actually use auth_token

adisbladis · Jul 13, 2020 · ae4fdaa · ae4fdaa
1 parent ebc8f88
commit ae4fdaa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nixops_digitalocean/backends/droplet.py b/nixops_digitalocean/backends/droplet.py
@@ -64,7 +64,7 @@ def get_type(cls) -> str:
 
     def __init__(self, name: str, config: ResourceEval):
         super().__init__(name, config)
-        self.auth_token = self.config.droplet.authToken
+        self.auth_token = self.config.droplet.authToken.strip()
         self.region = self.config.droplet.region
         self.size = self.config.droplet.size
         self.enable_ipv6 = self.config.droplet.enableIpv6
@@ -165,6 +165,10 @@ def create_after(self, resources, defn) -> Set:
             if isinstance(r, ssh_keypair.SSHKeyPairState)
         }
 
+    def set_common_state(self, defn: DropletDefinition) -> None:
+        super().set_common_state(defn)
+        self.auth_token = defn.auth_token
+
     def get_auth_token(self) -> Optional[str]:
         return os.environ.get("DIGITAL_OCEAN_AUTH_TOKEN", self.auth_token)