🐛 fix: fix get_token in create_setter_by_str

zrr-lab · Apr 9, 2024 · 374f70a · 374f70a
1 parent ed5f686
commit 374f70a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dns_manager/utils.py b/dns_manager/utils.py
@@ -23,7 +23,12 @@ def create_setter_by_str(config: dict, dns_setter: str = "dnspod"):
     >>> assert isinstance(setter, DNSPodSetter)
 
     """
-    token = get_token("dnspod", get_token_config(), create=True)
+    token = get_token(
+        "dnspod",
+        get_token_config(),
+        create=True,
+        env_names=["TENCENTCLOUD_SECRET_ID", "TENCENTCLOUD_SECRET_KEY"],
+    )
     match dns_setter:
         case "dnspod":
             setter = DNSPodSetter(config, token)