__init__.py: Improve tag checking (With the previous version of check wasn't working, don't know if it was dependant on old Netbox version)

nmap.py: Add extra function to improve DNS resolution since nmap is not always consistent

nbs/__init__.py

@@ -1,6 +1,6 @@
 import logging
-import requests
 
+import requests
 from pynetbox import api
 
 
@@ -47,7 +47,12 @@ class NetBoxScanner(object):
             return False
 
         if nbhost:
-            if (self.tag in nbhost.tags):
+            tag_update = False
+            for tag in nbhost.tags:
+                if (self.tag == tag.name):
+                    tag_update = True
+                    break
+            if tag_update:
                 if (host[1] != nbhost.description):
                     aux = nbhost.description
                     nbhost.description = host[1]
@@ -59,7 +64,7 @@ class NetBoxScanner(object):
                     logging.info(f'unchanged: {host[0]}/32 "{host[1]}"')
                     self.stats['unchanged'] += 1
             else:
-                logging.info(f'unchanged: {host[0]}/32 "{host[1]}"')
+                logging.info(f'no-tag(unchanged): {host[0]}/32 "{host[1]}"')
                 self.stats['unchanged'] += 1
         else:
             self.netbox.ipam.ip_addresses.create(

nbs/nmap.py

@@ -1,5 +1,8 @@
+import socket
+
 import nmap3
 
+
 class Nmap(object):
 
     def __init__(self, unknown, networks):
@@ -13,13 +16,26 @@ class Nmap(object):
         for item in self.networks:
             temp_scan_result = nmap.nmap_no_portscan(item.replace('\n', ''))
             self.scan_results = {**self.scan_results, **temp_scan_result}
+            self.scan_results.pop("stats")
+            self.scan_results.pop("runtime")
         return self.scan_results
 
+    def dns_resolution(self):
+        # Try to improve DNS resolution since NMAP is not consistent
+        for ip, v in self.scan_results.items():
+            try:
+                name, arpa, ip = socket.gethostbyaddr(ip)
+                try:
+                    v["hostname"][0]["name"]
+                except (TypeError, IndexError):
+                    v.update({"hostname": {"name": name, "type": 'PTR'}})
+            except socket.herror:
+                pass
+
     def run(self):
-        scan_result = self.scan()
+        self.scan()
-        scan_result.pop("stats")
+        self.dns_resolution()
-        scan_result.pop("runtime")
+        for k,v in self.scan().items():
-        for k,v in scan_result.items():
             try:
                 self.hosts.append((
                     k,
@@ -30,4 +46,3 @@ class Nmap(object):
                     k,
                     self.unknown
                 ))
-