系统安装

Author: chenziyaos

installos/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

installos/admin.py

@@ -0,0 +1,35 @@
+import cobbler
+import xmlrpclib
+class CobblerAPI(object):
+    def __init__(self,url,user,password):
+        self.cobbler_user= user
+        self.cobbler_pass = password
+        self.cobbler_url = url
+
+    def add_system(self,hostname,ip_add,mac_add,profile):
+        '''
+        Add Cobbler System Infomation
+        '''
+        ret = {
+            "result": True,
+            "comment": [],
+        }
+
+        remote = xmlrpclib.Server(self.cobbler_url)
+        token = remote.login(self.cobbler_user,self.cobbler_pass)
+        system_id = remote.new_system(token)
+        remote.modify_system(system_id,"name",hostname,token)
+        remote.modify_system(system_id,"hostname",hostname,token)
+        remote.modify_system(system_id,'modify_interface', {
+            "macaddress-eth0" : mac_add,
+            "ipaddress-eth0" : ip_add,
+            "dnsname-eth0" : hostname,
+        }, token)
+        remote.modify_system(system_id,"profile",profile,token)
+        remote.save_system(system_id, token)
+        try:
+            remote.sync(token)
+        except Exception as e:
+            ret['result'] = False
+            ret['comment'].append(str(e))
+        return ret

installos/cobbler_api.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+from django import forms
+from installos.models import SystemInstall
+
+class SystemInstallForm(forms.ModelForm):
+    class Meta:
+        model = SystemInstall
+        exclude = ('install_date',)
+        widgets = {
+          'ip': forms.TextInput(attrs={'class': 'form-control'}),
+          'hostname': forms.TextInput(attrs={'class': 'form-control'}),
+          'macaddress': forms.TextInput(attrs={'class': 'form-control'}),
+          'system_version': forms.TextInput(attrs={'class': 'form-control'}),
+        }

installos/form.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='InstallRecord',
+            fields=[
+                ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
+                ('install_date', models.CharField(max_length=20, verbose_name='\u5b89\u88c5\u65f6\u95f4')),
+                ('ip', models.CharField(max_length=20, verbose_name='\u5b89\u88c5IP')),
+                ('system_version', models.CharField(max_length=20, verbose_name=b'\xe5\xae\x89\xe8\xa3\x85\xe6\x93\x8d\xe4\xbd\x9c\xe7\xb3\xbb\xe7\xbb\x9f\xe7\x89\x88\xe6\x9c\xac')),
+            ],
+            options={
+                'verbose_name': '\u88c5\u673a\u8bb0\u5f55\u7ba1\u7406',
+            },
+        ),
+        migrations.CreateModel(
+            name='SystemInstall',
+            fields=[
+                ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
+                ('ip', models.CharField(max_length=20, verbose_name='\u5f85\u88c5\u673aIP')),
+                ('hostname', models.CharField(max_length=30, verbose_name='\u4e3b\u673a\u540d')),
+                ('macaddress', models.CharField(max_length=50, verbose_name='MAC\u5730\u5740')),
+                ('system_version', models.CharField(max_length=20, verbose_name='\u64cd\u4f5c\u7cfb\u7edf')),
+                ('install_date', models.DateTimeField(auto_now_add=True, verbose_name='\u5b89\u88c5\u65f6\u95f4')),
+            ],
+            options={
+                'verbose_name': '\u88c5\u673a\u5217\u8868',
+                'verbose_name_plural': '\u88c5\u673a\u5217\u8868\u7ba1\u7406',
+            },
+        ),
+    ]

installos/migrations/0001_initial.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+from django.db import models
+
+class SystemInstall(models.Model):
+    """
+    System Install Manage
+    """
+    ip = models.CharField(max_length=20, verbose_name=u'待装机IP')
+    hostname = models.CharField(max_length=30, verbose_name=u'主机名')
+    macaddress = models.CharField(max_length=50, verbose_name=u'MAC地址')
+    system_version = models.CharField(max_length=20, verbose_name=u'操作系统')
+    install_date = models.DateTimeField(auto_now_add=True, verbose_name=u'安装时间')
+
+    def __unicode__(self):
+        return u'%s -- %s' %(self.ip, self.install_date)
+
+    class Meta:
+        verbose_name = u'装机列表'
+        verbose_name_plural = u'装机列表管理'
+
+class InstallRecord(models.Model):
+    """
+    Server Install Recored
+    """
+    install_date = models.CharField(max_length=20, verbose_name=u'安装时间')
+    ip = models.CharField(max_length=20, verbose_name=u'安装IP')
+    system_version = models.CharField(max_length=20, verbose_name = '安装操作系统版本')
+
+    def __unicode__(self):
+        return u'%s - %s' %(self.ip, self.system_version)
+
+    class Meta:
+        verbose_name = u'装机记录'
+        verbose_name = u'装机记录管理'

installos/migrations/__init__.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

installos/models.py

@@ -0,0 +1,142 @@
+# -*- coding: utf-8 -*-
+from django.http import HttpResponseRedirect
+from django.template import RequestContext
+from django.shortcuts import render_to_response, get_object_or_404
+from django.core.urlresolvers import reverse
+from django.contrib.auth.decorators import login_required
+from django.core.paginator import Paginator
+from installos.models import *
+from asset.models import HostList
+from installos.form import SystemInstallForm
+from installos.cobbler_api import CobblerAPI
+from ycm.models import *
+from ycm.mysql import db_operate
+from Ycm import settings
+
+def system_install_managed(request,id=None):
+    """
+    Management host to be installed
+    """
+    user = request.user
+    if id:
+        system_install = get_object_or_404(SystemInstall, pk=id)
+        action = 'edit'
+        page_name = '编辑安装信息'
+    else:
+        system_install = SystemInstall()
+        action = 'add'
+        page_name = '添加安装信息'
+    if request.method == 'POST':
+        operate = request.POST.get('operate')
+        form = SystemInstallForm(request.POST,instance=system_install)
+        if form.is_valid():
+            if action == 'add':
+                form.save()
+                return HttpResponseRedirect(reverse('install_list'))
+            if operate:
+                if operate == 'update':
+                    form.save()
+                    db = db_operate()
+                    sql = 'select ip from installed_systeminstall where id = %s' % (id)
+                    ret = db.mysql_command(settings.YCM_MYSQL,sql)
+                    Message.objects.create(type='idc', action='install', action_ip=ret, content='主机信息已更新(macadd、system_version)，准备装机')
+                    return HttpResponseRedirect(reverse('install_list'))
+                else:
+                    pass
+    else:
+        form = SystemInstallForm(instance=system_install)
+
+    return render_to_response('install_manage.html',
+           {"form": form,
+            "page_name": page_name,
+            "action": action,
+           },context_instance=RequestContext(request))
+
+def system_install_list(request):
+    """
+    List all waiting for the host operating system is installed
+    """
+    user = request.user
+
+    #获取待装机的信息,从数据库中查询是否存在，未存在的插入到列表
+    result = HostList.objects.filter(status='待装机')
+    install_list = []
+    for i in range(len(result)):
+        ip = str(result[i]).split()[0]
+        hostname = str(result[i]).split()[2]
+        ret = SystemInstall.objects.filter(ip=ip)
+        if ret:
+            message = ip + ' already in database'
+        else:
+            data = {'ip': ip, 'hostname': hostname}
+            install_list.append(data)
+
+    #列表数据插入数据库
+    for i in range(len(install_list)):
+        p = SystemInstall(ip=install_list[i]['ip'],hostname=install_list[i]['hostname'])
+        p.save()
+
+    all_system_list = SystemInstall.objects.all()
+    paginator = Paginator(all_system_list,10)
+
+    try:
+        page = int(request.GET.get('page','1'))
+    except ValueError:
+        page = 1
+
+    try:
+        all_system_list = paginator.page(page)
+    except :
+        all_system_list = paginator.page(paginator.num_pages)
+
+    return render_to_response('install_list.html', {'all_system_list': all_system_list, 'page': page, 'paginator':paginator},context_instance=RequestContext(request))
+
+
+def system_install(request):
+    """
+    安装系统
+    """
+
+    cobbler = CobblerAPI(url=settings.Cobbler_API['url'],username=settings.Cobbler_API['user'],password=settings.Cobbler_API['password'])
+    if request.method == 'GET':
+       ip = request.GET.get('ip')
+       hostname = request.GET.get('host')
+       mac_add = request.GET.get('mac')
+       profile = request.GET.get('ver')
+       ret = cobbler.add_system(hostname=hostname,ip_add=ip,mac_add=mac_add,profile=profile)
+       if ret['result']:
+           data = SystemInstall.objects.filter(ip=ip)
+           install_date = str(data[0]).split('--')[1].strip()
+           InstallRecord.objects.create(ip=ip,system_version=profile,install_date=install_date)
+           HostList.objects.filter(ip=ip).update(status='已使用')     #主机信息加入cobbler system，主机列表的状态变更为已使用状态，不再是待装机状态！
+           SystemInstall.objects.filter(ip=ip).delete()               #安装后，装机列表此IP信息删除，转让到安装记录里供审计
+           Message.objects.create(type='idc', action='install', action_ip=ip, content='主机信息添加至cobbler，进入安装模式')
+       return HttpResponseRedirect(reverse('install_list'),{},context_instance=RequestContext(request))
+
+
+def system_install_record(request):
+    """
+    List all operating system installation records
+    """
+
+    user = request.user
+
+    record = InstallRecord.objects.all()
+    paginator = Paginator(record,10)
+
+    try:
+        page = int(request.GET.get('page','1'))
+    except ValueError:
+        page = 1
+
+    try:
+        record = paginator.page(page)
+    except :
+        record = paginator.page(paginator.num_pages)
+
+    return render_to_response('install_record_list.html',
+                                {'record': record,
+                                 'page': page,
+                                 'paginator':paginator},
+                                 context_instance=RequestContext(request)
+                              )