DBHelper.py 文件源码

def insertDb(self, ip_list):
        print '[+]', 'insert db'
        save_pool = []
        # print ip_list
        for ip in ip_list:
            old_ip = self.session.query(IpPool).filter(and_(IpPool.ip == ip['ip'], IpPool.port == ip['port'])).first()
            if old_ip:
                continue
            if len(save_pool) > 100:
                self.session.add_all(save_pool)
                self.session.commit()
                print '[+] ', "session commit"
                save_pool = []
            else:
                ip_obj = IpPool(ip=ip['ip'], port=ip['port'], location=ip['location'], iptype=ip['iptype'], protocol=ip['protocol'])
                save_pool.append(ip_obj)
        if save_pool:
            self.session.add_all(save_pool)
            self.session.commit()
        # self.session.close()