#!/usr/bin/python# coding=utf-8#python实现对MongoDB的操作#需要安装python2、pymongo、安装pymongo可能需要pip,logging打印日志#改脚本主要功能就是每5秒改一次mongodb中存储的ip,5秒后再改回来import pymongoimport loggingimport datetimeimport osimport timeimport tracebackimport sys,gc# 初始化logginglogging.basicConfig(level=logging.NOTSET, format='%(asctime)s %(filename)s[line:%(lineno)d] [%(levelname)s] %(message)s', datefmt='%Y-%m-%d %H:%M:%S', filename='clear_screenshot_based_on_db.log', filemode='a')# set to print log to console at the same timeconsole = logging.StreamHandler()console.setLevel(logging.NOTSET)formatter = logging.Formatter('%(asctime)s %(name)s [%(levelname)s] %(message)s')console.setFormatter(formatter)logging.getLogger('').addHandler(console)class MongoDB_Util():#写一个类，用于实例化pymongo对象以及对mongo的增删改查 def __init__(self, user, password, host, port, database, max_pool_size=2): try: client = pymongo.MongoClient(host=host, port=port, maxPoolSize=max_pool_size)#实例化对象，需要mongo的相关参数 if client == None: logging.error( "MongoDB_Util.__init__ : Initialize mongodb client object error,host:%s port:%s" % ( host, port)) raise Exception("Initialize mongodb client object error") try: admin_db = client['admin'] #指定连接的库 admin_db.authenticate(user, password) #身份验证 self.persist_db = admin_db #赋值属性，通过该属性执行其他操作 except Exception as exp: logging.error( "MongoDB_Util.__init__ : Authtication failed,host:%s port:%s database:%s user:%s password:%s" % (host, port, database, user, password)) raise exp except Exception as exp: logging.error( "MongoDB_Util.__init__ : Exception has occured : %s" % str(sys.exc_info()[1])) raise exp def update(self):#update方法，对mongodb执行update操作 tom = None try: if self.persist_db == None: logging.error('MongoDB_Util.delete : Persist object has not been initialised') raise Exception('Persist object has not been initialised') collection = self.persist_db['fuzzing_agent.configuration'] #链接fuzzing_agent.configuration表 abc=collection.find({"default_value" : "10.10.1.179"}) #执行find命令 for i in abc: tom = i["default_value"] if tom == "10.10.1.179": #判断该值是否为179,如果是则通过update改为134 collection.update({"default_value" : "10.10.1.179"}, {'$set':{'default_value': '10.10.1.134'}}) collection.update({"default_value" : "10.10.1.179:/run/media/root/disk01/fuzzing"}, {'$set':{"default_value" :"10.10.1.134:/run/media/root/disk01/fuzzing"}}) logging.debug("179 > 134") time.sleep(5) else: collection.update({"default_value" : "10.10.1.134"}, {'$set':{'default_value': '10.10.1.179'}}) collection.update({"default_value" : "10.10.1.134:/run/media/root/disk01/fuzzing"}, {'$set':{"default_value" :"10.10.1.179:/run/media/root/disk01/fuzzing"}}) logging.debug("134 > 179") time.sleep(5) return except Exception as exp: logging.error('Mongodb_Util.delete : Exception has occured : ' + str(sys.exc_info()[1])) raise expdef change_ip(mongodb_host) : result = False try : mongo_util = MongoDB_Util('username', 'password', mongodb_host, 27017, 'admin',max_pool_size = 2) #用户名，密码，mongodb的Ip，端口，连接库，连接池 mongo_util.update() result = True except : logging.error('change_ip: Exception has occured : ' + traceback.format_exc()) finally: return resultif __name__ == '__main__' : while True : try : if change_ip(mongodb_host = '127.0.0.1') : #mongodb的ip地址 logging.debug('update is done !!!') else : logging.error('not update done') except : logging.error('main : Exception has occured : ' + traceback.format_exc()) finally: logging.debug("one loop %s" %datetime.datetime.now()) 
