Shulou(Shulou.com)06/03 Report--

This article shows you the Python third-party library package mysql database sample analysis, concise and easy to understand, absolutely can make you shine, through the detailed introduction of this article I hope you can gain something.

This time the database is encapsulated, even if it is not for automated testing, database operations must also be able to, why? Because in many cases, repairing existing network user data is achieved through scripts, if you don't use other languages, then Python's advantage in this area is very obvious.

'''

Created on October 10, 2019

@author: qguan

'''

import pymysql

from utils.HandleLogging import log as logging

mysql_info = {"host": '127.0.0.1',

"port": 3306,

"user": 'root',

"passwd": 'root',

"db": 'mysql',

"charset": 'utf8'}

class HandleMySQL(object):

'''

classdocs

'''

def __init__(self):

'''

Initialize mysql database information, create connection cursors

'''

self.db_info = mysql_info

self.logger = logging

try:

self.conn = pymysql.connect(host=self.db_info['host'],

port=self.db_info['port'],

user=self.db_info['user'],

passwd=self.db_info['passwd'],

db=self.db_info['db'],

charset=self.db_info['charset'],

cursorclass=pymysql.cursors.DictCursor)

self.logger.info("success! ")

except Exception as a:

self.logger.info("failture:%s" % a)

raise a

else:

self.logger.info("create cursor! ")

self.cursor = self.conn.cursor()

def __call__(self, sql, arg=None, is_more=False):

'''

invoke database execution method

:param sql: sql statement executed, str type

:param arg: Parameter must be sequence type: list, str, tuple

:param is_more: Whether the query displays more data, usually determined by the limit of sql

:return:

'''

self.mysql_execute(sql, arg)

if is_more:

result = self.cursor.fetchall()

else:

result = self.cursor.fetchone()

#Not suitable, if there is sql in the loop execution, close here, will cause sql execution failure

# self.mysql_close()

return result

def mysql_execute(self, sql, arg=None):

'' Zhengzhou abortion hospital which is good http://mobile.zhongyuan120.com/

Encapsulate methods that execute sql statements themselves

'''

try:

self.cursor.execute(sql)

except Exception as a:

self.conn.rollback() # sql rollback processing

self.logger.info ("rollback sql, exception as follows: {}".format(a))

raise a

else:

self.conn.commit() # sql commit

def mysql_getrows(self, sql):

'''

Return query result set

:return: Returns a tuple of nested dictionaries

'''

try:

self.cursor.execute(sql) # arg must be a sequence type

except Exception as a:

self.logger.info ("Exception executing SQL statement: %s" % a)

raise a #throws an exception directly without executing the following statement

else:

rows = self.cursor.fetchall()

return rows

def mysql_getrow(self, sql):

'''

Return query result set

return: The result is a dictionary

'''

try:

self.cursor.execute(sql) # arg must be a sequence type

except Exception as a:

self.logger.info ("Exception executing SQL statement: %s" % a)

raise a

else:

row = self.cursor.fetchone()

return row

def mysql_close(self):

'''

Close database connection, close cursor

'''

try:

self.conn.close()

self.cursor.close()

self.logger.info ("Close mysql connection normally! ")

except Exception as a:

self.logger.info ("Failed to close mysql connection pool, exception as follows: {}".format(a))

raise a

if __name__ == '__main__':

mq = HandleMySQL()

sql = 'SELECT *from user limit 3'

# print(mq(sql, is_more=True))

mq.mysql_close()

The above content is an example analysis of a third-party library encapsulating mysql database in Python. Have you learned any knowledge or skills? If you want to learn more skills or enrich your knowledge reserves, please pay attention to the industry information channel.

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.