Python 小练习二 数据库MySQL、Redis

 import pymysql,redis
 def op_mysql(host,user,password,db,sql,port=3306,charset='utf8'):
     conn = pymysql.connect(host=host,user=user,
                            password=password,
                            port=port,
                            charset=charset,db=db)
     cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
     cur.execute(sql)
     sql_start = sql[:6].upper() #取sql前6个字符串，判断它是什么类型的sql语句
     if sql_start=='SELECT' :
         res = cur.fetchall()
     else:
         conn.commit()
         res = 'ok'
     cur.close()
     conn.close()
     return res

 def op_redis(host,password,k,v=None,port=6379,db=0):
     r = redis.Redis(host=host,password=password,port=port,db=db)
     if v:
         r.set(k,v)
         res = 'ok'
     else:
         res = r.get(k)
         if res: #这里是判断有没有get到数据
             res = res.decode()
         else:
             res = None
     return res

 print(__name__)
 print('前面哈')
 if __name__=='__main__':
     #别人导入这个python文件的时候，下面的代码不会被执行
     #自己测试的时候用
     # print(__name__)  # __main__
     print('后面哈')
     sql = 'select * from bt_stu limit 5;'
     sql2 = 'update bt_stu set class="test123" where id=503;'
     res = op_mysql(
         host='211.149.xxx.16',
         user='jxz',password='',#port这里一定要写int类型
         port=3306,db='jxz',charset='utf8',sql=sql2)
     print(res)