Skip to content

Commit fe72491

Browse files
kvinwangkvinwang
committed
ingress: Format code
1 parent 652f3dc commit fe72491

4 files changed

Lines changed: 30 additions & 46 deletions

File tree

‎custom-domain/dstack-ingress/scripts/dns_manager.py‎

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -36,40 +36,34 @@ def main():
3636

3737
if args.action == "set_cname":
3838
if not args.content:
39-
print("Error: --content is required for CNAME records",
40-
file=sys.stderr)
39+
print("Error: --content is required for CNAME records", file=sys.stderr)
4140
sys.exit(1)
4241

4342
success = provider.set_alias_record(args.domain, args.content)
4443
if not success:
45-
print(
46-
f"Failed to set alias record for {args.domain}", file=sys.stderr)
44+
print(f"Failed to set alias record for {args.domain}", file=sys.stderr)
4745
sys.exit(1)
4846
print(f"Successfully set alias record for {args.domain}")
4947

5048
elif args.action == "set_alias":
5149
if not args.content:
52-
print("Error: --content is required for alias records",
53-
file=sys.stderr)
50+
print("Error: --content is required for alias records", file=sys.stderr)
5451
sys.exit(1)
5552

5653
success = provider.set_alias_record(args.domain, args.content)
5754
if not success:
58-
print(
59-
f"Failed to set alias record for {args.domain}", file=sys.stderr)
55+
print(f"Failed to set alias record for {args.domain}", file=sys.stderr)
6056
sys.exit(1)
6157
print(f"Successfully set alias record for {args.domain}")
6258

6359
elif args.action == "set_txt":
6460
if not args.content:
65-
print("Error: --content is required for TXT records",
66-
file=sys.stderr)
61+
print("Error: --content is required for TXT records", file=sys.stderr)
6762
sys.exit(1)
6863

6964
success = provider.set_txt_record(args.domain, args.content)
7065
if not success:
71-
print(
72-
f"Failed to set TXT record for {args.domain}", file=sys.stderr)
66+
print(f"Failed to set TXT record for {args.domain}", file=sys.stderr)
7367
sys.exit(1)
7468
print(f"Successfully set TXT record for {args.domain}")
7569

@@ -81,12 +75,9 @@ def main():
8175
)
8276
sys.exit(1)
8377

84-
success = provider.set_caa_record(
85-
args.domain, args.caa_tag, args.caa_value
86-
)
78+
success = provider.set_caa_record(args.domain, args.caa_tag, args.caa_value)
8779
if not success:
88-
print(
89-
f"Failed to set CAA record for {args.domain}", file=sys.stderr)
80+
print(f"Failed to set CAA record for {args.domain}", file=sys.stderr)
9081
sys.exit(1)
9182
print(f"Successfully set CAA record for {args.domain}")
9283

‎custom-domain/dstack-ingress/scripts/dns_providers/base.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -217,9 +217,7 @@ def set_cname_record(
217217
)
218218
return self.create_dns_record(new_record)
219219

220-
def set_txt_record(
221-
self, name: str, content: str, ttl: int = 60
222-
) -> bool:
220+
def set_txt_record(self, name: str, content: str, ttl: int = 60) -> bool:
223221
"""Set a TXT record (delete existing and create new).
224222
225223
Args:

‎custom-domain/dstack-ingress/scripts/dns_providers/cloudflare.py‎

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,7 @@ def __init__(self):
2323
super().__init__()
2424
self.api_token = os.getenv("CLOUDFLARE_API_TOKEN")
2525
if not self.api_token:
26-
raise ValueError(
27-
"CLOUDFLARE_API_TOKEN environment variable is required")
26+
raise ValueError("CLOUDFLARE_API_TOKEN environment variable is required")
2827
self.base_url = "https://api.cloudflare.com/client/v4"
2928
self.headers = {
3029
"Authorization": f"Bearer {self.api_token}",
@@ -58,8 +57,7 @@ def setup_certbot_credentials(self) -> bool:
5857
return True
5958

6059
except Exception as e:
61-
print(
62-
f"Error setting up Cloudflare credentials: {e}", file=sys.stderr)
60+
print(f"Error setting up Cloudflare credentials: {e}", file=sys.stderr)
6361
return False
6462

6563
def _make_request(
@@ -169,8 +167,7 @@ def get_dns_records(
169167
"""Get DNS records for a domain."""
170168
zone_id = self._ensure_zone_id(name)
171169
if not zone_id:
172-
print(
173-
f"Error: Could not find zone for domain {name}", file=sys.stderr)
170+
print(f"Error: Could not find zone for domain {name}", file=sys.stderr)
174171
return []
175172

176173
params = f"zones/{zone_id}/dns_records?name={name}"
@@ -204,7 +201,8 @@ def create_dns_record(self, record: DNSRecord) -> bool:
204201
zone_id = self._ensure_zone_id(record.name)
205202
if not zone_id:
206203
print(
207-
f"Error: Could not find zone for domain {record.name}", file=sys.stderr)
204+
f"Error: Could not find zone for domain {record.name}", file=sys.stderr
205+
)
208206
return False
209207

210208
data = {
@@ -224,17 +222,15 @@ def create_dns_record(self, record: DNSRecord) -> bool:
224222
data["priority"] = record.priority
225223

226224
print(f"Adding {record.type.value} record for {record.name}")
227-
result = self._make_request(
228-
"POST", f"zones/{zone_id}/dns_records", data)
225+
result = self._make_request("POST", f"zones/{zone_id}/dns_records", data)
229226

230227
return result.get("success", False)
231228

232229
def delete_dns_record(self, record_id: str, domain: str) -> bool:
233230
"""Delete a DNS record."""
234231
zone_id = self._ensure_zone_id(domain)
235232
if not zone_id:
236-
print(
237-
f"Error: Could not find zone for domain {domain}", file=sys.stderr)
233+
print(f"Error: Could not find zone for domain {domain}", file=sys.stderr)
238234
return False
239235

240236
print(f"Deleting record ID: {record_id}")
@@ -249,7 +245,9 @@ def create_caa_record(self, caa_record: CAARecord) -> bool:
249245
zone_id = self._ensure_zone_id(caa_record.name)
250246
if not zone_id:
251247
print(
252-
f"Error: Could not find zone for domain {caa_record.name}", file=sys.stderr)
248+
f"Error: Could not find zone for domain {caa_record.name}",
249+
file=sys.stderr,
250+
)
253251
return False
254252

255253
clean_value = caa_record.value.strip('"')
@@ -268,7 +266,6 @@ def create_caa_record(self, caa_record: CAARecord) -> bool:
268266
print(
269267
f"Adding CAA record for {caa_record.name} with tag {caa_record.tag} and value {clean_value}"
270268
)
271-
result = self._make_request(
272-
"POST", f"zones/{zone_id}/dns_records", data)
269+
result = self._make_request("POST", f"zones/{zone_id}/dns_records", data)
273270

274271
return result.get("success", False)

‎custom-domain/dstack-ingress/scripts/dns_providers/linode.py‎

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,7 @@ def __init__(self):
2424
super().__init__()
2525
self.api_token = os.getenv("LINODE_API_TOKEN")
2626
if not self.api_token:
27-
raise ValueError(
28-
"LINODE_API_TOKEN environment variable is required")
27+
raise ValueError("LINODE_API_TOKEN environment variable is required")
2928
self.base_url = "https://api.linode.com/v4"
3029
self.headers = {
3130
"Authorization": f"Bearer {self.api_token}",
@@ -181,8 +180,7 @@ def get_dns_records(
181180
"""Get DNS records for a domain."""
182181
zone_id = self._ensure_zone_id(name)
183182
if not zone_id:
184-
print(
185-
f"Error: Could not find zone for domain {name}", file=sys.stderr)
183+
print(f"Error: Could not find zone for domain {name}", file=sys.stderr)
186184
return []
187185

188186
result = self._make_request("GET", f"domains/{zone_id}/records")
@@ -238,7 +236,8 @@ def create_dns_record(self, record: DNSRecord) -> bool:
238236
zone_id = self._ensure_zone_id(record.name)
239237
if not zone_id:
240238
print(
241-
f"Error: Could not find zone for domain {record.name}", file=sys.stderr)
239+
f"Error: Could not find zone for domain {record.name}", file=sys.stderr
240+
)
242241
return False
243242

244243
subdomain = self._get_subdomain(record.name, zone_id)
@@ -268,13 +267,11 @@ def delete_dns_record(self, record_id: str, domain: str) -> bool:
268267
"""Delete a DNS record."""
269268
zone_id = self._ensure_zone_id(domain)
270269
if not zone_id:
271-
print(
272-
f"Error: Could not find zone for domain {domain}", file=sys.stderr)
270+
print(f"Error: Could not find zone for domain {domain}", file=sys.stderr)
273271
return False
274272

275273
print(f"Deleting record ID: {record_id}")
276-
result = self._make_request(
277-
"DELETE", f"domains/{zone_id}/records/{record_id}")
274+
result = self._make_request("DELETE", f"domains/{zone_id}/records/{record_id}")
278275

279276
return result.get("success", False)
280277

@@ -297,8 +294,7 @@ def set_alias_record(
297294
print(f"✅ Resolved {domain} to IP: {ip_address}")
298295

299296
if not ip_address:
300-
raise socket.gaierror(
301-
"Could not resolve any variant of the domain")
297+
raise socket.gaierror("Could not resolve any variant of the domain")
302298

303299
# Delete any existing CNAME records for this name (clean transition)
304300
existing_cname_records = self.get_dns_records(name, RecordType.CNAME)
@@ -317,7 +313,9 @@ def create_caa_record(self, caa_record: CAARecord) -> bool:
317313
zone_id = self._ensure_zone_id(caa_record.name)
318314
if not zone_id:
319315
print(
320-
f"Error: Could not find zone for domain {caa_record.name}", file=sys.stderr)
316+
f"Error: Could not find zone for domain {caa_record.name}",
317+
file=sys.stderr,
318+
)
321319
return False
322320

323321
subdomain = self._get_subdomain(caa_record.name, zone_id)

0 commit comments

Comments
 (0)