python通过配置文件生成日志

logtemplate.py：

#coding:utf-8 import ConfigParser from string import Template import time import math import Queue from getIP import * def getLogByConf(num): cf = ConfigParser.ConfigParser() cf.read("logTemplate.conf") opts = cf.options(cf.sections()[0]) t = Template(cf.get(cf.sections()[0],opts[0])) dic={} for i in range(1,len(opts)): dic[opts[i]] = cf.get(cf.sections()[0],opts[i]) dic['date'] = time.ctime() getSysIp = getIP() dic['sysip'] = getSysIp.get_ip_address() return t.substitute(dic) * num def getLogByScreenPlay(num): cf = ConfigParser.ConfigParser() cf.read("logTemplate.conf") opts = cf.options(cf.sections()[0]) t = Template(cf.get(cf.sections()[0],opts[0])) dic={} for i in range(1,len(opts)): dic[opts[i]] = cf.get(cf.sections()[0],opts[i]) dic['date'] = time.ctime() getScreenPlayIp = getIP() dic['sysip'] = getScreenPlayIp.get_ip_address() dic['srcip'] = getScreenPlayIp.getIpByScreenPlay() return t.substitute(dic) * num def getLogByRandom(num): cf = ConfigParser.ConfigParser() cf.read("logTemplate.conf") opts = cf.options(cf.sections()[0]) t = Template(cf.get(cf.sections()[0],opts[0])) dic={} for i in range(1,len(opts)): dic[opts[i]] = cf.get(cf.sections()[0],opts[i]) getScreenPlayIp = getIP() dic['sysip'] = getScreenPlayIp.get_ip_address() q = Queue.Queue() for i in range(0,num): dic['date'] = time.ctime() getRandomIp = getIP() dic['srcip'] = getRandomIp.getIpByRandom() dic['srcport'] = int(random.uniform(0, 255)) log = t.substitute(dic) q.put(log) return q;

getIP.py：

#coding:utf-8 import ConfigParser import os import socket import fcntl import struct import sys import re import random """ 获取主机ip地址 参数ifname: 通过'lo'获取的为环回地址, 通过'eth0'获取的为主机ip地址 """ class getIP: def get_ip_address(self,ifname = 'lo'): try: s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) return socket.inet_ntoa(fcntl.ioctl( s.fileno(), 0x8915, # SIOCGIFADDR struct.pack('256s', ifname[:15]) )[20:24]) except: ips = os.popen("LANG=C ifconfig | grep \"inet addr\" | grep -v \"127.0.0.1\"" " | awk -F \":\" '{print $2}' | awk '{print $1}'").readlines() if len(ips) > 0: return ips[0] return '' def getIpByConf(self): cf = ConfigParser.ConfigParser() cf.read("logTemplate.conf") opts = cf.options(cf.sections()[0]) if 'srcip' in opts: return cf.get(cf.sections()[0],'srcip') else: return '配置文件未配置IP字段' def getIpByScreenPlay(self): IpByScreenPlay = raw_input("请输入ip:\n") IsIP = re.search('^([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])\.' '([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])$', IpByScreenPlay) if IsIP: return IpByScreenPlay else: return '127.0.0.1' def getIpByRandom(self): IpByRandom = (str(int(random.uniform(0, 255))) + '.' + str(int(random.uniform(0, 255))) + '.' + str(int(random.uniform(0, 255))) + '.' + str(int(random.uniform(0, 255)))) return IpByRandom

logTemplate.conf:

[conf] template = $date $sysip sshd[$shdcode]: $responsetype for $user from $srcip port $srcport ssh2 sysip = 127.0.0.1 shdcode = 6666 responsetype = Failed password user = root srcip = 255.255.255.0 srcport = 6666