From f1617f6dd9a7cc89948af973375bba89c11b4e1f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alo=C3=AFs=20Micard?= <a.micard@vold.lu>
Date: Wed, 19 Feb 2025 11:48:43 +0100
Subject: [PATCH] Handle edge cases

---
 app/__main__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/__main__.py b/app/__main__.py
index ab0cd73..ea2a249 100644
--- a/app/__main__.py
+++ b/app/__main__.py
@@ -26,10 +26,11 @@ def fetch_netbox_host_overrides(nb_api: pynetbox.api) -> dict:
             continue
 
         host = nb_ip_address.dns_name.split('.')[0]
+        domain = '.'.join(nb_ip_address.dns_name.split('.')[1:])
 
         nb_host_overrides[nb_ip_address.dns_name] = {
             'host': host,
-            'domain': nb_ip_address.dns_name.replace(f'{host}.', ''),
+            'domain': domain,
             'ip': [
                 nb_ip_address.address.split('/')[0],
             ],