python动态SQL并执行查询

最新推荐文章于 2024-07-20 03:22:02 发布

IT-例子

最新推荐文章于 2024-07-20 03:22:02 发布

阅读量1.2k

点赞数 1

分类专栏： python 文章标签： python sql 数据库

本文链接：https://blog.csdn.net/qq_43001132/article/details/126637244

版权

python 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

python动态SQL

# coding=utf-8
# sqlserver的连接
import pymssql
import datetime
import time

start = time.perf_counter()
print('程序正在运行, 请稍等...')
print("数据读取中...")
today_now = datetime.datetime.now()
print("现在时间是：", today_now)

"""
这是一个数据库连接类
"""
class MSSQL:
    def __init__(self, host, user, pwd, db):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.db = db

    def __GetConnect(self):
        """
        得到连接信息
        返回: conn.cursor()
        """
        if not self.db:
            raise (NameError, "没有设置数据库信息")
        self.conn = pymssql.connect(host=self.host, user=self.user, password=self.pwd, database=self.db, charset="utf8")
        cur = self.conn.cursor()
        if not cur:
            raise (NameError, "连接数据库失败")
        else:
            return cur

    def ExecQuery(self, sql):
        """
        执行查询语句
        返回的是一个包含tuple的list，list的元素是记录行，tuple的元素是每行记录的字段

        """
        cur = self.__GetConnect()
        cur.execute(sql)
        resList = cur.fetchall()

        # 查询完毕后必须关闭连接
        self.conn.close()
        return resList

    def ExecNonQuery(self, sql):
        """
        执行非查询语句

        调用示例：
            cur = self.__GetConnect()
            cur.execute(sql)
            self.conn.commit()
            self.conn.close()
        """
        cur = self.__GetConnect()
        cur.execute(sql)
        self.conn.commit()
        self.conn.close()


def main():
    # host默认为127.0.0.1，如果打开了TCP动态端口的需要加上端口号，如'127.0.0.1:1433'
    # user默认为sa
    # pwd为自己设置的密码
    # db为数据库名字
    ms = MSSQL(host='177.20.128.0', user="sa", pwd="sa123456", db="db_jkgd")
    table_name = input("请输入查询的数据库表名:")
    resList = ms.ExecQuery("SELECT COUNT(*) AS 总数 FROM " + str(table_name))
    print(resList)


if __name__ == '__main__':
    main()
    print("文件输出成功")

    today_now_end = datetime.datetime.now()
    print("现在时间是：", today_now_end)
    end = time.perf_counter()
    print('本次耗时:%0.0f秒' % (end - start))