SQLUtil.py 472 Bytes
# coding=utf-8
#author:        4N
#createtime:    2021/6/29
#email:         nheweijun@sina.com


class SQLUtil:


    @classmethod
    def get_count(cls,session,table_name):
        count = session.execute('select count(*) from "{}"'.format(table_name)).fetchone()[0]
        return count

    @classmethod
    def rename(cls,session,origin,after_name):
        rename_sql = 'alter table "{}" rename to "{}"'.format(origin, after_name)
        session.execute(rename_sql)