#coding:utf-8 import MySQLdb def get_mysql_data(sql,data=None): ''' 这个函数用来执行sql,返回数据 :sql: sql语句 :return: :配置文件里要定义MySQL_CONFIG字典,里面要包含所需要的变量 ''' try: #捕捉mysql连接异常 conn = MySQLdb.connect(host=MySQL_CONFIG['HOST'], port=MySQL_CONFIG['PORT'], user=MySQL_CONFIG['USER'], passwd=MySQL_CONFIG['PASSWORD'], db=MySQL_CONFIG['DB'], charset='utf8' ) except Exception as e: print('mysql连接失败') return 'mysql_error' else: cursor = conn.cursor() #创建游标,需要指定游标的类型,字典类型 try: #捕捉sql错误异常 cursor.execute(sql,data)#这样执行sql是为了防止sql注入 except Exception as e: print('sql执行失败,请检查sql!') return 'sql_error' else: #如果sql执行成功的话,提交一下,返回数据 conn.commit() if sql[:6]=='select':#如果是select语句的话,返回查询结果 return cursor.fetchone() else: return cursor.lastrowid#如果不是select语句的话,返回自增长id finally:#不管sql执行成功没执行成功,都关闭连接 cursor.close() conn.close()
python 数据库的方法封装
猜你喜欢
转载自www.cnblogs.com/yueli85627/p/9179116.html
今日推荐
周排行