python用c/s實現服務器簡單管理

背景：

由於有大量的windows虛擬機用來做一些任務。這些windows上的機器程序要經常更新。每次部署升級，需要一臺臺的遠程桌面上去操作，進行簡單升級操作。這樣講花費大量時間。並且伴隨windows機器的增加，將更加難管理。

目標：

無需遠程桌面，即可完成基本操作，如：部署升級程序，遠程啓動及停止代理服務，重啓，關機等。

解決方法：

採用python socket 的C/S形式來實現，在管理端發送管理命令到被管理端執行，進行更新程序代碼等操作。需在windows機器上安裝python。實現代碼如下：

1、以下是server端，部署到被管理的windows或者linux機器

#!/usr/bin/python
#Script Name : pyserver.py

import os
import logging
import sys
import socket

host='0.0.0.0'
port=4567
maxclient=10
if sys.platform == 'win32':
    logfile='c:/server.log'
else:
    logfile='/tmp/server.log'

def initlog():
    logger=logging.getLogger()
    hdlr=logging.FileHandler(logfile)
    formatter=logging.Formatter('%(asctime)s %(levelname)s %(message)s')
    hdlr.setFormatter(formatter)
    logger.addHandler(hdlr)
    logger.setLevel(logging.NOTSET)
    return logger

def socketserver():
    s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    s.bind((host,port))
    s.listen(maxclient)
#    print "Server is running on port %d; press ctrl-c to terminate." % port
    while True:
        clientsock,clientaddr=s.accept()
#        print "connect from %s" % str(clientaddr)
        message="connect from : %s" % str(clientaddr)
        lg.info(message)
        clientfile=clientsock.makefile('rw',0)
        data=clientsock.recv(1024)
        message="Execute command : %s" %data
        lg.info(message)
        command=os.popen(data).read()
        clientfile.write("%s" % command)
        clientfile.close()
        clientsock.close()

try:
    lg=initlog()
    socketserver()
except KeyboardInterrupt:
    print "User Press Ctrl+C,Exit"

2、以下是管理端，用於發送命令到被管理端

 

#!/usr/bin/env python
#Script Name : pyclient.py

import os
import sys
import getopt
import socket
import logging
import time

port=4567
logfile='/tmp/client.log'
file=''
ip=''

def initlog():
    logger=logging.getLogger()
    hdlr=logging.FileHandler(logfile)
    formatter=logging.Formatter('%(asctime)s %(levelname)s %(message)s')
    hdlr.setFormatter(formatter)
    logger.addHandler(hdlr)
    logger.setLevel(logging.NOTSET)
    return logger

def usage():
    print '''
Usage: python client.py [options...]
Options:
    -f : read host list from file
    -I : read host from this command line
    -c : command run at remote host
    -h : this help info
    python client.py -I 10.16.134.164 -c "/sbin/ifconfig"
    '''

def socketclient(IP,port,cmd):
    s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.connect((IP,port))
    ISOTIMEFORMAT = '%Y-%m-%d %X'
    date = time.strftime(ISOTIMEFORMAT, time.localtime())
    result ="-------------------------------------------------------------------\n"
    result = "%sDATE    : %s\nHOST    : %sCOMMAND : %s\n" %(result,date,IP,cmd)
    while 1:
        if not cmd:
            break
        s.sendall(cmd)
        cmd=s.recv(1024)
        if not cmd:
            break
        #print cmd
        result = "%sRESULT  :\n\n%s" %(result,cmd)
        print result
        writeLog(logfile,result)
    s.close()

def writeLog(file,message):
    logger = open(file, 'a+')
    logger.writelines(message)
    logger.close()

try:
    opts,args = getopt.getopt(sys.argv[1:],'hI:f:c:')
except getopt.GetoptError:
    usage()
    sys.exit()

if len(opts) == 0:
    usage()
    sys.exit()

for opt,arg in opts:
    if opt in ('-h','--help'):
        usage()
        sys.exit()
    elif opt == '-f':
        #print 'read flie %s' %arg
        file = arg
    elif opt == '-I':
        #print 'server IP is %s' %arg
        ip = arg
    elif opt == '-c':
        #print 'command is %s' %arg
        command = arg


if file:
    for ip in os.popen('cat %s' %file).readlines():
        socketclient(ip,port,command)
else:
    if ip :
        socketclient(ip,port,command)
    else:
        print 'Error '

 

簡單的實現了以上需求。大家可以一起討論下，用更好的方法來實現以上的需求。