[大数据]Python数据库增删改查脚本

import pymysql
from pymysql.cursors import DictCursor

def con(host='127.0.0.1',user='root',password='',port=3306,db='login',charset='utf8'):
#     # 创建连接
    conne=pymysql.connect(host=host,user=user,password=password,port=port,db=db,charset=charset)
# 创建游标
    cursor=conne.cursor(cursor=DictCursor)
    #<class 'tuple'>
    return conne,cursor

def sql(sql):
    c,corsor=con()

     # 执行sql
    corsor.execute(query=sql)
    if sql[:6]=='select' :
        result = corsor.fetchall()
        if result:
            # <class 'list'>
            # print(result)
            corsor.close()
            c.close()
            return result
        else:
            return 0
    # 增删改需要提交
    else:
        c.commit()
        corsor.close()
        c.close()
        return 1

if __name__ == '__main__':
    a= f'UPDATE stu SET PASSWORD=\'qq\' WHERE sname=\'qq\''
    c=sql(a)