数据库查询转excel小工具

import psycopg2
import psycopg2.pool
import xlsxwriter

class Sql:
    ‘‘‘
    database:库名
    user:用户名
    password:密码
    host:地址
    port:端口
    ‘‘‘
    def __init__(self,database,user,password,host,port):
        self.database=database
        self.user=user
        self.password=password
        self.host=host
        self.port=port
    def _sql(self):
        self.conn = psycopg2.connect(database=self.database, user=self.user,
                                     password=self.password,host=self.host, port=self.port)
        self.cursor = self.conn.cursor()

    def close(self):
        self.cursor.close()
        self.conn.close()


    def select(self,sql,args=None):
        self._sql()
        #查询语句
        try:
            self.cursor.execute(sql,args)
            coloumns = [row[0] for row in self.cursor.description]
            result = [[str(item) for item in row] for row in self.cursor.fetchall()]
            return [dict(zip(coloumns, row)) for row in result],coloumns
        except Exception as ex:
            print(ex)
        finally:
            self.close()

    def execute(self,sql,args=None):
        ###修改提交语句
        self._sql()
        try:
            self.cursor.execute(sql,args)
            self.conn.commit()
            res=self.cursor.rowcount
            return res
        except Exception as x:
            print(x)
        finally:
            self.close()




def run(title,field,wbk,data,sheet_name):

    sheet1 = wbk.add_worksheet(sheet_name)
    get_excel(sheet1, title, data, field)


def get_excel(sheet,cov,res,rov):
    row=1
    sheet.write_row(‘A1‘, cov)
    for line in res:
        l=[]
        for i in range(len(rov)):
             l.append(line[rov[i]])
        sheet.write_row(row, 0, l)
        row += 1

if __name__ == ‘__main__‘:
    wbk = xlsxwriter.Workbook(‘weibo.xlsx‘)####生成excel对象
    sql = Sql("uniqlo_reviews_analytics","bigodata","B1g0uniqlo","117.50.14.201","22432")###生成Sql类
    post_data , columns = sql.select(‘‘‘select * from stagingdata.posts WHERE createrdtime::date BETWEEN ‘{0}‘::date and ‘{1}‘::date 
                  ‘‘‘.format(‘2019-05-01‘,‘2019-05-31‘))####返回数据集  和所有字段集
    title=[‘帖子id‘,‘发帖人‘,‘微博内容‘,‘发帖时间‘,‘评论数‘,‘转发数‘,‘点赞数‘]#标题名 也可以传columns
    field=[‘posterid‘,‘postername‘,‘weibo‘,‘createrdtime‘,‘hnum‘,‘forward‘,‘aproval‘]#字段名  也可以传columns
    
    reply_data, columns = sql.select(‘‘‘select * from stagingdata.reply t1 WHERE  EXISTS 
                  (select 0 from stagingdata.posts WHERE createrdtime::date BETWEEN ‘{0}‘::date and ‘{1}‘ and 
                t1.posterid=posterid)
                      ‘‘‘.format(‘2019-05-01‘, ‘2019-05-31‘))  ####返回数据集  和所有字段集
    title1 = [‘评论id‘, ‘评论内容‘, ‘评论时间‘,‘帖子id‘]  # 标题名 也可以传columns
    field1 = [‘commentid‘, ‘comment_text‘, ‘comment_time‘,  ‘posterid‘]  # 字段名  也可以传columns

    run(title,field,wbk,post_data,‘帖子‘)
    run(title1, field1, wbk, reply_data,‘评论‘)
    wbk.close()