fix: ipam interface should check on virtualmachine nil state

ninech · Oct 29, 2024 · bf4a644 · bf4a644
1 parent 89d6252
commit bf4a644
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/lib/netbox_client_ruby/api/ipam/ip_address.rb b/lib/netbox_client_ruby/api/ipam/ip_address.rb
@@ -33,12 +33,8 @@ def interface
         interface_data = data['interface']
 
         return nil unless interface_data
-
-        if interface_data.key?('virtual_machine')
-          Virtualization::Interface.new interface_data['id']
-        else
-          DCIM::Interface.new interface_data['id']
-        end
+        return Virtualization::Interface.new interface_data['id'] unless interface_data.dig('virtual_machine').nil?
+        return DCIM::Interface.new interface_data['id']
       end
     end
   end