利用Python 程序实现Linux 网卡 bonding 实现

#!/usr/bin/env python

import os,sys,time,re,shutil
import socket
import fcntl
import struct
import traceback
import commands

#Get interface name
interface_path = '/etc/sysconfig/network-scripts/'
def LOG(info):
""" Log files ...."""
logfile = '/root/pxe_install.log'
files = open(logfile,'a')
try:
files.write('%s : %s \n'%(time.ctime(),info))
except IOError:
files.close()
files.close()

def get_interface ():
os.chdir(interface_path)
eth  = em = list()
for inter in os.listdir(interface_path):
if inter[0:-1] == 'ifcfg-em':
if inter == 'ifcfg-em1' or inter == 'ifcfg-em2':
em.append(inter)
elif inter[0:-1] == 'ifcfg-eth':
if inter == 'ifcfg-eth0' or inter == 'ifcfg-eth1':
eth.append(inter)
if eth:
LOG("Getting interface file name %s is Ok" %eth)
return eth
else:
LOG("Getting interface file name %s is Ok" %em)
return em

def main():
net_name = get_interface()
ipaddr = str()
for inter in net_name:
try:
shutil.move(inter,'/opt/' + inter+'.bak')
_interface_config(inter)
new_interface = inter.split('-')[-1]
if _configure_bond(new_interface):
_configure_bond(new_interface)
LOG("bond script init is Ok")
except Exception,e:
LOG(traceback.format_exc())

if _interface_modprobe():
_interface_modprobe()

if _rester_network():
_rester_network()

# Set interface eth* or em*
def _interface_config(interface):
"""
DEVICE=eth0
BOOTPROTO=static
NOBOOT=yes
NM_CONTROLLED=no
MASTER=bond0
SLAVE=yes
"""
fp = open(interface,'w')
new_interface = interface.split('-')[-1]
if interface == 'ifcfg-em1':
fp.write('DEVICE=%s \n'%new_interface)
fp.write('BOOTPROTO=static \n')
fp.write('ONBOOT=yes \n')
fp.write('NM_CONTROLLED=no \n')
fp.write('MASTER=bond0 \n')
fp.write('SLAVE=yes \n')
elif interface == 'ifcfg-em2':
fp.write('DEVICE=%s \n'%new_interface)
fp.write('BOOTPROTO=static \n')
fp.write('ONBOOT=yes \n')
fp.write('NM_CONTROLLED=no \n')
fp.write('MASTER=bond0 \n')
fp.write('SLAVE=yes \n')
elif interface == 'ifcfg-eth0':
fp.write('DEVICE=%s \n'%new_interface)
fp.write('BOOTPROTO=static \n')
fp.write('ONBOOT=yes \n')
fp.write('NM_CONTROLLED=no \n')
fp.write('MASTER=bond0 \n')
fp.write('SLAVE=yes \n')
elif interface == 'ifcfg-eth1':
fp.write('DEVICE=%s \n'%new_interface)
fp.write('BOOTPROTO=static \n')
fp.write('ONBOOT=yes \n')
fp.write('NM_CONTROLLED=no \n')
fp.write('MASTER=bond0 \n')
fp.write('SLAVE=yes \n')
fp.close()

def _configure_bond(inter):
"""
DEVICE=bond0
BOOTPROTO=static
ONBOOT=yes
IPADDR=192.168.0.100
NETMASK=255.255.255.0
NETWORK=192.168.0.0
BROADCAST=192.168.0.255
"""
#bond name message
if inter == 'eth0':
bond_name = 'ifcfg-bond0'
elif inter == 'em1':
bond_name =  'ifcfg-bond0'
elif inter == 'eth1':
bond_name = 'ifcfg-bond0'
elif inter == 'em2':
bond_name = 'ifcfg-bond0'
# ip address message
if _interface_get_ip(inter):
ipaddr  = _interface_get_ip(inter)
else:
ipaddr = '0.0.0.0'
# ip net mask info
try:
net_mk  = os.popen('ip a |grep %s|grep inet' %inter).readlines()[0]
res = net_mk.split()[1]
net_masklen = res.split('/')[-1]
except:
net_masklen = 18
net_mask = _interface_sum_master(net_masklen)
# default gateway is ....
try:
net_gate = os.popen('ip route |grep default').readlines()[0]
net_gateway = net_gate = net_gate.split(' ')[2]
except:
net_gateway = '0.0.0.0'

try:
if ipaddr == '0.0.0.0':
return ''
fp = open(bond_name,'w')
bond = bond_name.split('-')[-1]
fp.write("DEVICE=%s \n" %bond)
fp.write("BOOTPROTO=static \n")
fp.write("ONBOOT=yes \n")
fp.write("IPADDR=%s \n" %ipaddr)
fp.write("NETMASK=%s \n" % net_mask)
if bond == 'bond0':
fp.write("GATEWAY=%s \n" % net_gateway)
fp.write("DNS1=202.106.0.20 \n")
fp.write("DNS2=8.8.8.8 \n")
LOG("ifcfg-bond* configure is Ok")
return True
except Exception,e:
return False

def _interface_get_ip(inter):
try:
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
IP = socket.inet_ntoa(fcntl.ioctl(
s.fileno(),
0x8915,  # SIOCGIFADDR
struct.pack('24s',inter))[20:24])
return  IP
except Exception,e:
pass
return

# Add modprobe bonding model
def _interface_modprobe():
try:
fp = open('/etc/modprobe.d/bonding.conf','w')
fp.write("#Module options and blacklists written by bonding \n")
fp.write("alias bond0 bonding \n")
fp.write("options bond0 miimon=100 mode=1 \n")
fp.close()
x,y  = commands.getstatusoutput('modprobe bonding')
if x != 0:
LOG("modprobe bonding is failed")
return True
except:
LOG(traceback.format_exc())
return

# Restart Network
def _rester_network():
x,y  = commands.getstatusoutput('service network restart')
if x == 0:
LOG("restart netowrk is Ok ")
return True
else:
LOG("restart netowrk is Faild ")
return

# According to the CIDR calculation.net master
def _interface_sum_master(net_master):
mask =  (2** 8) - 2 ** (24 - int(net_master))
return '255.255.%s.0' % mask

if __name__  == "__main__":
sc = main()