from noc.core.mongo.connection import connect
connect()
from noc.sa.models.managedobject import ManagedObject,ManagedObjectProfile
from noc.sa.models.managedobject import AdministrativeDomain
from noc.main.models.pool import Pool
from noc.sa.models.authprofile import AuthProfile
from noc.inv.models.networksegment import NetworkSegment
from noc.sa.models.profile import Profile
from noc.vc.models.vcdomain import *
import nmap
import time

is_managed= True
administrative_domain="default"
profile="Generic.Host"
object_profile="default"
description=""
segment="ALL"
auth_profile="Loggin"
scheme="1"
address=""
port="23"
user=""
password=""
super_password=""
remote_path=""
trap_source_ip=""
trap_community=""
snmp_ro="public"
snmp_rw="private"
vc_domain="default"
vrf=""
termination_group=""
service_terminator=""
shape=""
config_filter_rule=""
config_diff_filter_rule=""
config_validation_rule=""
max_scripts="20"
tags=""
pool="default"
container=""
trap_source_type="m"
syslog_source_type="m"
time_pattern=""
x=""
y=""
default_zoom=""


nm = nmap.PortScanner()
nm.scan(hosts='10.112.0.0-253', arguments='-sn -n')
hosts_list = [(x, nm[x]['status']['state']) for x in nm.all_hosts()]
for host, status in hosts_list:
     print (host,status)
     if ManagedObject.objects.filter(address=host):
         continue
     m = ManagedObject(name=host,is_managed=is_managed,administrative_domain=AdministrativeDomain.objects.get(name=administrative_domain),\
                    profile=Profile.objects.get(name=profile),description=description,\
                    object_profile = ManagedObjectProfile.objects.get(name=object_profile), segment = NetworkSegment.objects.get(name = segment),\
                    auth_profile=AuthProfile.objects.get(name=auth_profile),scheme = 1, address = host,\
                    trap_source_ip=host,trap_community=trap_community,vc_domain = VCDomain.objects.get(name=vc_domain),shape=shape,\
                    trap_source_type=trap_source_type,syslog_source_type=syslog_source_type,\
                    pool = Pool.objects.get(name=pool))
     m.save()