b4tman/pddnsc
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects 1 Releases Activity

refactor vscale

Browse Source
This commit is contained in:
Dmitry Belyaev 2024-02-21 10:40:29 +03:00
parent c0d1d4b38d
commit ef4aaf61b8
Signed by: b4tman
GPG Key ID: 41A00BF15EA7E5F3
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
pddnsc/outputs/vscale.py
View File

@ -8,13 +8,20 @@ from pddnsc.base import BaseOutputProvider
class VscaleDomains(BaseOutputProvider): class VscaleDomains(BaseOutputProvider):
def post_init(self): def post_init(self):
self.__token = self.get_api_token() token = self.get_api_token()
if not self.__token: if not token:
raise KeyError("no api token, use env VSCALE_API_TOKEN") raise KeyError("no api token, use env VSCALE_API_TOKEN")
target = self.config["target"] self.__headers = {"X-Token": token}
self.domain = self.config["domain"] self.api_base = self.config.get("api_base", "https://api.vscale.io/v1/")
self.target = f"{target}.{self.domain}"
self.ttl = self.config.get("ttl", 300) self.ttl = self.config.get("ttl", 300)
self.domain = self.config["domain"]
target = self.config["target"]
self.target = f"{target}.{self.domain}"
self.save_ipv4 = self.config.get("ipv4", False)
self.save_ipv6 = self.config.get("ipv6", False)
if "ipv4" not in self.config and "ipv6" not in self.config:
self.save_ipv4 = self.save_ipv6 = True
def get_api_token(self) -> str: def get_api_token(self) -> str:
token_env = self.config.get("api_token_env", "VSCALE_API_TOKEN") token_env = self.config.get("api_token_env", "VSCALE_API_TOKEN")
@ -90,23 +97,19 @@ class VscaleDomains(BaseOutputProvider):
) )
async def set_addrs_imp(self, source_provider, addr_v4, addr_v6): async def set_addrs_imp(self, source_provider, addr_v4, addr_v6):
save_ipv4 = self.config.get("ipv4", False)
save_ipv6 = self.config.get("ipv6", False)
if "ipv4" not in self.config and "ipv6" not in self.config:
save_ipv4 = save_ipv6 = True
save_addrs = [] save_addrs = []
if addr_v4 and save_ipv4: if addr_v4 and self.save_ipv4:
save_addrs.append(("A", addr_v4)) save_addrs.append(("A", addr_v4))
if addr_v6 and save_ipv6: if addr_v6 and self.save_ipv6:
save_addrs.append(("AAAA", addr_v6)) save_addrs.append(("AAAA", addr_v6))
transport = self.best_transport(addr_v4, addr_v6) transport = self.best_transport(addr_v4, addr_v6)
api_base = self.config.get("api_base", "https://api.vscale.io/v1/")
headers = {"X-Token": self.__token}
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=transport, base_url=api_base, headers=headers, http2=True transport=transport,
base_url=self.api_base,
headers=self.__headers,
http2=True,
) as client: ) as client:
domain_id = await self.find_domain_id(client) domain_id = await self.find_domain_id(client)
for record_type, value in save_addrs: for record_type, value in save_addrs: