上篇博客中已经介绍过Django的安装和基本使用，如果大家还有不明白请参考我的博客或者留言
Python [4] Django的安装和基础运行环境简介
http://467754239.blog.51cto.com/4878013/1613612
这篇博客和大家聊聊Python结合Django实现IT资产管理
基础环境：
系统版本：CentOS 6.4 64bit Python版本：2.6 Django版本：1.6.5 ip地址：192.168.1.210
一、安装Django环境
# rpm -ivh http://dl.fedoraproject.org/pub/epel/6/x86_64/epel-release-6-8.noarch.rpm # sed -i ‘s@^#@@‘ /etc/yum.repos.d/epel.repo # sed -i ‘s@mirrorlist@#mirrorlist@‘ /etc/yum.repos.d/epel.repo # yum -y install python-pip # pip install ‘django==1.6.5‘
二、创建工程和应用
[root@localhost ~]# django-admin.py startproject Simplecmdb [root@localhost ~]# cd Simplecmdb/ [root@localhost Simplecmdb]# django-admin.py startapp hostinfo [root@localhost Simplecmdb]# tree ./ ./ ├── hostinfo │ ├── admin.py │ ├── __init__.py │ ├── models.py │ ├── tests.py │ └── views.py ├── manage.py └── Simplecmdb ├── __init__.py ├── settings.py ├── urls.py └── wsgi.py 2 directories, 10 files
三、配置应用
1、修改工程配置文件
[root@localhost Simplecmdb]# vim Simplecmdb/settings.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‘,#注释此行（默认禁止第三方curl工具使用） ‘django.contrib.auth.middleware.AuthenticationMiddleware‘, ‘django.contrib.messages.middleware.MessageMiddleware‘, ‘django.middleware.clickjacking.XFrameOptionsMiddleware‘, ) LANGUAGE_CODE = ‘zh-cn‘ #修改此行 TIME_ZONE = ‘Asia/Shanghai‘ #修改此行
2、添加应用的url访问
[root@localhost Simplecmdb]# cat Simplecmdb/urls.py from django.conf.urls import patterns, include, url from django.contrib import admin admin.autodiscover() urlpatterns = patterns(‘‘, # Examples: # url(r‘^$‘, ‘Simplecmdb.views.home‘, name=‘home‘), # url(r‘^blog/‘, include(‘blog.urls‘)), url(r‘^admin/‘, include(admin.site.urls)), url(r‘^hostinfo$‘,‘hostinfo.views.index‘), #添加此行 )
3、定义搜集主机信息的数据模型（也就是字段名和数据类型）
[root@localhost Simplecmdb]# cat hostinfo/models.py from django.db import models # Create your models here. class Host(models.Model): hostname = models.CharField(max_length=50) ip = models.IPAddressField() osversion = models.CharField(max_length=50) memory = models.CharField(max_length=50) disk = models.CharField(max_length=50) vendor_id = models.CharField(max_length=50) model_name = models.CharField(max_length=50) cpu_core = models.CharField(max_length=50) product = models.CharField(max_length=50) Manufacturer = models.CharField(max_length=50) sn = models.CharField(max_length=50)
4、初始化模型数据库并生成数据库文件
[root@localhost Simplecmdb]# python manage.py syncdb Creating tables ... Creating table django_admin_log Creating table auth_permission Creating table auth_group_permissions Creating table auth_group Creating table auth_user_groups Creating table auth_user_user_permissions Creating table auth_user Creating table django_content_type Creating table django_session Creating table hostinfo_host You just installed Django‘s auth system, which means you don‘t have any superusers defined. Would you like to create one now? (yes/no): yes Username (leave blank to use ‘root‘): root Email address: zhengyansheng@nihao.com Password: #后台admin的登陆密码 Password (again): Superuser created successfully. Installing custom SQL ... Installing indexes ... Installed 0 object(s) from 0 fixture(s)
5、注册后台admin并显示注册信息
[root@localhost Simplecmdb]# vim hostinfo/admin.py #注册数据库 from django.contrib import admin from hostinfo.models import Host # Register your models here. class HostAdmin(admin.ModelAdmin): list_display = [ ‘hostname‘, ‘ip‘, ‘osversion‘, ‘memory‘, ‘disk‘, ‘vendor_id‘, ‘model_name‘, ‘cpu_core‘, ‘product‘, ‘Manufacturer‘, ‘sn‘] admin.site.register(Host,HostAdmin)
6、定义用户的响应请求
[root@localhost Simplecmdb]# vim hostinfo/views.py from django.shortcuts import render from django.http import HttpResponse from hostinfo.models import Host # Create your views here. def index(req): print req if req.method == ‘POST‘: hostname = req.POST.get(‘hostname‘) ip = req.POST.get(‘ip‘) osversion = req.POST.get(‘osversion‘) memory = req.POST.get(‘memory‘) disk = req.POST.get(‘disk‘) vendor_id = req.POST.get(‘vendor_id‘) model_name = req.POST.get(‘model_name‘) cpu_core = req.POST.get(‘cpu_core‘) product = req.POST.get(‘product‘) Manufacturer = req.POST.get(‘Manufacturer‘) sn = req.POST.get(‘sn‘) host = Host() host.hostname = hostname host.ip = ip host.osversion = osversion host.memory = memory host.disk = disk host.vendor_id = vendor_id host.model_name = model_name host.cpu_core = cpu_core host.product = product host.Manufacturer = Manufacturer host.sn = sn host.save() return HttpResponse(‘ok‘) else: return HttpResponse(‘no data‘)
三、运行django的工程服务
[root@localhost ~]# cd /root/Simplecmdb/ [root@localhost Simplecmdb]# python manage.py runserver 0.0.0.0:80 Validating models... 0 errors found March 02, 2015 - 16:11:12 Django version 1.6.5, using settings ‘Simplecmdb.settings‘ Starting development server at http://0.0.0.0:80/ Quit the server with CONTROL-C.
四、浏览器访问页面
1、访问主页面
2、访问hostinfo页面
因为我们没有传送任何数据，所以返回我们定义的字符串"no data"
3、访问后台admin页面
注册的账号是root，密码为123456
五、以下是用Python实现对Linux主机的信息采集
搜集的信息主要包括：主机名、ip、系统版本、硬盘、内存、cpu的个数、cpu的厂商、序列号和生产商
[root@localhost ~]# cat post_hostinfo.py
#!/usr/bin/env python
#coding:utf8
#author:Allentuns
#time:2015-02-14
from subprocess import Popen,PIPE
import urllib,urllib2
import pickle
import json
import re
###[hostname message]#####
def get_HostnameInfo(file):
with open(file,‘r‘) as fd:
data = fd.read().split(‘\n‘)
for line in data:
if line.startswith(‘HOSTNAME‘):
hostname = line.split(‘=‘)[1]
break
return hostname
#####[ipaddr message]#####
def get_Ipaddr():
P = Popen([‘ifconfig‘],stdout=PIPE)
data = P.stdout.read()
list = []
str = ‘‘
option = False
lines = data.split(‘\n‘)
for line in lines:
if not line.startswith(‘ ‘):
list.append(str)
str = line
else:
str += line
while True:
if ‘‘ in list:
list.remove(‘‘)
else:
break
r_devname = re.compile(‘(eth\d*|lo)‘)
r_mac = re.compile(‘HWaddr\s([A-F0-9:]{17})‘)
r_ip = re.compile(‘addr:([\d.]{7,15})‘)
for line in list:
devname = r_devname.findall(line)
mac = r_mac.findall(line)
ip = r_ip.findall(line)
if mac:
return ip[0]
#####[osversion message]#####
def get_OsVerion(file):
with open(file) as fd:
lines = fd.readlines()
os_version = lines[0][:-8]
return os_version
#####[memory message]#####
def get_MemoryInfo(file):
with open(file) as fd:
data_list = fd.read().split(‘\n‘)
MemTotal_line = data_list[0]
Memory_K = MemTotal_line.split()[1]
Memory_G = float(Memory_K)/1000/1000
Memory_G2 = ‘%.2f‘ % Memory_G
memory = Memory_G2 + ‘G‘
return memory
#####[disk message]#####
def get_DiskInfo():
p = Popen([‘fdisk‘,‘-l‘],stdout=PIPE,stderr=PIPE)
stdout,stderr = p.communicate()
diskdata = stdout
disk_initial_size = 0
re_disk_type = re.compile(r‘Disk /dev/[shd]{1}.*:\s+[\d.\s\w]*,\s+([\d]+).*‘)
disk_size_bytes = re_disk_type.findall(diskdata)
for size in disk_size_bytes:
disk_initial_size += int(size)
disk_size_total_bytes = ‘%.2f‘ % (float(disk_initial_size)/1000/1000/1000)
disk_size_total_G = disk_size_total_bytes + ‘G‘
disk = disk_size_total_G
return disk
#####[cpu message]#####
def get_CpuInfo():
p = Popen([‘cat‘,‘/proc/cpuinfo‘],stdout=PIPE,stderr=PIPE)
stdout, stderr = p.communicate()
cpudata = stdout.strip()
cpu_dict = {}
re_cpu_cores = re.compile(r‘processor\s+:\s+([\d])‘)
re_vendor_id = re.compile(r‘vendor_id\s+:\s([\w]+)‘)
re_model_name = re.compile(r‘model name\s+:\s+(.*)‘)
res_cpu_cores = re_cpu_cores.findall(cpudata)
cpu_dict[‘Cpu_Cores‘] = int(res_cpu_cores[-1]) + 1
res_vendor_id = re_vendor_id.findall(cpudata)
cpu_dict[‘Vendor_Id‘] = res_vendor_id[-1]
res_model_name = re_model_name.findall(cpudata)
cpu_dict[‘Model_Name‘] = res_model_name[-1]
return cpu_dict
#####[Demi message]#####
def get_dmidecode():
P = Popen([‘dmidecode‘],stdout=PIPE)
data = P.stdout.read()
lines = data.split(‘\n\n‘)
dmidecode_line = lines[2]
line = [i.strip() for i in dmidecode_line.split(‘\n‘) if i]
Manufacturer = line[2].split(‘: ‘)[-1]
product = line[3].split(‘: ‘)[-1]
sn = line[5].split(‘: ‘)[-1]
return Manufacturer,product,sn
if __name__ == ‘__main__‘:
#####[get data]#####
hostname = get_HostnameInfo(‘/etc/sysconfig/network‘)
ip = get_Ipaddr()
osversion = get_OsVerion(‘/etc/issue‘)
memory = get_MemoryInfo(‘/proc/meminfo‘)
disk = get_DiskInfo()
Vendor_Id = get_CpuInfo()[‘Vendor_Id‘]
Model_Name = get_CpuInfo()[‘Model_Name‘]
Cpu_Cores = get_CpuInfo()[‘Cpu_Cores‘]
Manufacturer,product,sn = get_dmidecode()
#####[get dict]#####
hostinfo = {
‘hostname‘:hostname,
‘ip‘:ip,
‘osversion‘:osversion,
‘memory‘:memory,
‘disk‘:disk,
‘vendor_id‘:Vendor_Id,
‘model_name‘:Model_Name,
‘cpu_core‘:Cpu_Cores,
‘product‘:product,
‘Manufacturer‘:Manufacturer,
‘sn‘:sn,
}
data = urllib.urlencode(hostinfo)
req = urllib2.urlopen(‘http://192.168.1.210:80/hostinfo‘,data)执行此脚本
[root@localhost ~]# python post_hostinfo.py
然后刷新后台admin的页面
以上配置还有问题，就是不能保证主机的唯一性，如果反复执行上述脚本会得到以下结果
这个问题可以这样解决，用户执行脚本然后传递参数到服务器，服务器可以做个判断，如果传递的参数已经存在，那么我们不保存此信息，如果是更新，那么我们重新赋值并保存到数据库中
[root@localhost Simplecmdb]# cat hostinfo/views.py from django.shortcuts import render from django.http import HttpResponse from hostinfo.models import Host # Create your views here. def index(req): if req.method == ‘POST‘: hostname = req.POST.get(‘hostname‘) ip = req.POST.get(‘ip‘) osversion = req.POST.get(‘osversion‘) memory = req.POST.get(‘memory‘) disk = req.POST.get(‘disk‘) vendor_id = req.POST.get(‘vendor_id‘) model_name = req.POST.get(‘model_name‘) cpu_core = req.POST.get(‘cpu_core‘) product = req.POST.get(‘product‘) Manufacturer = req.POST.get(‘Manufacturer‘) sn = req.POST.get(‘sn‘) try: #修改这一部分 host = Host.objects.get(hostname=hostname) #判断是否存在，存在报异常 except: host = Host() host.hostname = hostname host.ip = ip host.osversion = osversion host.memory = memory host.disk = disk host.vendor_id = vendor_id host.model_name = model_name host.cpu_core = cpu_core host.product = product host.Manufacturer = Manufacturer host.sn = sn host.save() return HttpResponse(‘ok‘) else: return HttpResponse(‘no data‘)
最终显示的效果如下：
本文出自 “郑彦生” 博客，请务必保留此出处http://467754239.blog.51cto.com/4878013/1616551
原文地址：http://467754239.blog.51cto.com/4878013/1616551
