python-Django收集主机信息

1.创建工程simplecmdb

django-admin.py startproject simplecmdb

2.创建应用

cd simplecmdb

python manage.py startapp hostinfo

3.编辑配置文件

vim simplecmdb/setting.py

INSTALLED_APPS = (
    'django.contrib.admin',
    'django.contrib.auth',
    'django.contrib.contenttypes',
    'django.contrib.sessions',
    'django.contrib.messages',
    'django.contrib.staticfiles',
    'hostinfo'
#添加应用
)

MIDDLEWARE_CLASSES = (
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
#   'django.middleware.csrf.CsrfViewMiddleware',
#注释csrf方便使用第三方工具curl
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
)

4.添加model

from django.db import models

# Create your models here.

class Host(models.Model):
    hostname = models.CharField(max_length=50)
    ip = models.IPAddressField()
    vendor = models.CharField(max_length=50)
    product = models.CharField(max_length=50)
    sn = models.CharField(max_length=50)
    cpu_model = models.CharField(max_length=50)
    cpu_num = models.IntegerField()
    memory = models.CharField(max_length=50)
    osver = models.CharField(max_length=50)

5.初始化数据库

python manage.py sqlall hostinfo

python manage,py syncdb

6.编辑url.py，添加url

url(r'^hostinfo/collect/','hostinfo.views.collect')

7.编辑views.py

from django.shortcuts import render
from django.http import HttpResponse
from hostinfo.models import Host

# Create your views here.

def collect(req):
    if req.POST:
        hostname = req.POST.get('hostname')
        ip = req.POST.get('ip')
        product = req.POST.get('product')
        sn = req.POST.get('sn')
        vendor = req.POST.get('vendor')
        cpu_model = req.POST.get('cpu_model')
        cpu_num = req.POST.get('cpu_num')
        memory = req.POST.get('memory')
        osver = req.POST.get('osver')

        host = Host()
        host.hostname = hostname
        host.ip = ip
        host.product = product
        host.sn = sn
        host.vendor = vendor
        host.cpu_model = cpu_model
        host.cpu_num = cpu_num
        host.memory = memory
        host.osver = osver

        host.save()

        return HttpResponse('data have save into DB')
    else:
        return HttpResponse('there is no data from POST method')

8.注册model到admin界面

from django.contrib import admin
from hostinfo.models import Host

# Register your models here.

class HostAdmin(admin.ModelAdmin):
    list_display = [
            'hostname',
            'ip',
            'product',
            'vendor',
            'sn',
            'cpu_model',
            'cpu_num',
            'memory',
            'osver']

admin.site.register(Host,HostAdmin)

9.启动服务

python manage.py runserver 0.0.0.0:8000

10.访问

curl -d hostname='node02' -d ip='192.168.1.2' -d product='BL 380' -d sn='XX213' -d vendor='HP' -d cpu_model='Intel' -d cpu_num=1 -d memory='250G' -d osver='md 05c' http://192.168.1.120:8000/hostinfo/collect/

#!/usr/bin/env python

import urllib,urllib2
from subprocess import Popen,PIPE

def getIfconfig():
    p = Popen(['ifconfig'],stdout=PIPE)
    data = p.stdout.read()
    return data

def getDmi():
    p = Popen(['dmidecode'],stdout=PIPE)
    data = p.stdout.read()
    return data

def parseData(data):
    parsed_data = []
    new_line = ''
    data = [i for i in data.split('
') if i]
    for line in data:
        if line[0].strip():
             parsed_data.append(new_line)
            new_line = line +'
'
        else:        
            new_line += line+'
'
    parsed_data.append(new_line)
    return [i for i in parsed_data if i]

def parseIfconfig(parsed_data):
    parsed_data = [i for i in parsed_data if not i.startswith('lo')]
    for lines in parsed_data:
        line_list = lines.split('
')
        devname = line_list[0].split()[0]
        macaddr = line_list[0].split()[-1]
        ipaddr = line_list[1].split()[1].split(':')[1]
        break
    dic['ip'] =ipaddr
    return dic

def parseDmi(parsed_data):
    dic = {}
    parsed_data = [i for i in parsed_data if i.startswith('System Information')]
    parsed_data = [i for i in parsed_data[0].split('
')[1:] if i]
    dmi_dic =  dict([i.strip().split(':') for i in parsed_data])
    dic['vendor'] = dmi_dic['Manufacturer'].strip()
    dic['product'] = dmi_dic['Product Name'].strip()
    dic['sn'] = dmi_dic['Serial Number'].strip()[:15]
    return dic

def getHostname(f):
    with open(f) as fd:
        for line in fd:
            if line.startswith('HOSTNAME'):
                hostname = line.split('=')[1].strip()
                break
    return {'hostname':hostname}

def getOsver(f):
    with open(f) as fd:
        for line in fd:
            osver = line.strip()
            break
    return {'osver':osver}

def getCpu(f):
    num = 0
    with open(f) as fd:
        for line in fd:
            if line.startswith('processor'):
                num +=1
            if line.startswith('model name'):
                cpu_model = line.split(':')[1].split()
                cpu_model = cpu_model[0]+' '+cpu_model[-1]
    return {'cpu_num':num,'cpu_model':cpu_model}

def getMemory(f):
    with open(f) as fd:
        for line in fd:
            if line.startswith('MemTotal'):
                mem = int(line.split()[1].strip())
                break
    mem = "%d" % int(mem/1024.0)+'M'
    return {'memory':mem}
                

if __name__ == '__main__':
    dic = {}
    data_ip = getIfconfig()
    parsed_data_ip = parseData(data_ip)
    ip = parseIfconfig(parsed_data_ip)
    data_dmi = getDmi()
    parsed_data_dmi = parseData(data_dmi)
    dmi = parseDmi(parsed_data_dmi)
    hostname = getHostname('/etc/sysconfig/network')
    osver = getOsver('/etc/issue')
    cpu = getCpu('/proc/cpuinfo')
    mem = getMemory('/proc/meminfo')
    dic.update(ip)
    dic.update(dmi)
    dic.update(hostname)
    dic.update(osver)
    dic.update(cpu)
    dic.update(mem)
    data = urllib.urlencode(dic)
    req = urllib2.urlopen('http://192.168.1.120:8000/hostinfo/collect/',data)
    print req.read()