python-接口自动化-数据库操作

下面以PostgreSQL为例，讲解下python3.7利用psycopg2进行数据库链接。

1、psycopg2第三方库是python3.7链接数据库的重要的py文件。
附：如果公司是mysql数据库，请用pymysql，第三方库进行操作。这里不在细说。

2、首先下载第三方库，然后进行库的倒入。

import psycopg2
from scripts.handle_config import do_config     # 配置文件中会用到。

3、对数据库进行连接，初始化处理

class HandleDB:

    def __init__(self):
        # 1、创建数据库连接
        self.conn = psycopg2.connect(host=do_config.get_value("ev_bike", "host"),
                                     user=do_config.get_value("ev_bike", "user"),
                                     password=do_config.get_value("ev_bike", "password"),
                                     dbname=do_config.get_value("ev_bike", "dbname"),
                                     port=do_config.get_int("ev_bike", "port"))

        # 2. 创建游标
        self.cursor = self.conn.cursor()

4、执行sql语句的方法，执行sql语句后提取的数据类型是元组的类型，后面根据入参需要转化，锁需要的数据类型，如字符串，int等。

    def get_count(self, sql):
        self.conn.commit()
        return self.cursor.execute(sql)

    def select_one_sql(self, sql):
        """
        获取sql的一条数据 主要还是看sql语句的要查询的数据
        :param sql:
        :return:
        """
        self.conn.commit()
        self.cursor.execute(sql)
        return self.cursor.fetchone()

    def select_all_sql(self, sql):
        """
        获取sql的全部数据
        :param sql:
        :return:
        """
        self.conn.commit()
        self.cursor.execute(sql)
        return self.cursor.fetchall()

5、关闭游标链接，和数据库链接，释放资源。

    def close_coon(self):
        """
        # 关闭游标,# 关闭连接
        :return:
        """
        self.cursor.close_coon()
        self.conn.close_coon.coon()

最后，一个类封装后，一定要测试下代码。