Chinaunix首页 | 论坛 | 博客
  • 博客访问: 136130
  • 博文数量: 34
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 252
  • 用 户 组: 普通用户
  • 注册时间: 2014-05-27 20:04
文章分类

全部博文(34)

文章存档

2018年(1)

2017年(23)

2016年(10)

我的朋友

分类: 系统运维

2017-07-01 17:18:43

#encoding=utf8  
import logging
import MySQLdb

def olddb():
    return DB("200db",3306,"user","password","dbname")

def newdb():
    return DB("200db",3306,"user","password","dbname")

class DB():
    def __init__(self,db_host,db_port,db_user,db_pwd,db_name):
        self.db_host = db_host
        self.db_port = db_port
        self.db_user = db_user
        self.db_pwd = db_pwd
        self.db_name = db_name
        self.conn = self.getconnection()  

    def getconnection(self):
        return MySQLdb.connect(
        host=self.db_host,
        port=self.db_port,
        user=self.db_user,
        passwd=self.db_pwd,
        db=self.db_name,
        charset="utf8")

    def query(self, sqlstr, params=None):
        cursor = self.conn.cursor()
        if params:
            cursor.execute(sqlstr, params)
        else:
            cursor.execute(sqlstr)
        res = cursor.fetchall()
        self.conn.commit()
        cursor.close()
        self.conn.close()
        return res
    
    def saveall(self, table_name, res):
        cursor = self.conn.cursor()
        reslen = len(res)
        if reslen > 0:
            cursor.execute("""truncate table %s""" % table_name)
            colums=("%s,"*len(res[0])).rstrip(',')
            cursor.executemany("""insert into %s """ % table_name + """values(%s)""" % colums, res)
            self.conn.commit()
            print '导入完成'    
        else:
            print '不用导入'
        cursor.close()
        self.conn.close()
阅读(2039) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~