Python小练习_将数据库中表数据存到redis里

# ##练习：将xxx数据库中my_user表中数据存到redis里面
# 分析： pymysql、json、redis
#     1.连接数据库，查到数据库里面所有的数据，游标类型要用pymysql.cursor.Dictcour
#     2.查到所有数据[{"id":1,"passwd":"49487dd4f94008a6110275e48ad09448","username":"6j","is_admin":1}]
#     3.循环这个list,取到username，把username当作key
#     4.再把这个小字典转成json，存进去就ok

 import json,redis
 def my_db(table_name):
     import pymysql
     coon =pymysql.connect(
         user='xxx',passwd='',host='xxx.xxx.xx.xxx',port=3306,
         db='xxx',charset='utf8'
     )
     cur = coon.cursor(cursor=pymysql.cursors.DictCursor) #建立游标，指定cursor类型返回的是字典
     # cur = coon.cursor()
     sql = 'select * from %s;'%table_name
     cur.execute(sql)
     if sql.strip()[:6].upper()=='SELECT':
         res = cur.fetchall()
     cur.close()
     coon.close()
     return res

 all_date = my_db('my_user')
 r = redis.Redis(host='xxx.xxx.xxx.xxx',password='',db=0)  #端口号默认6379
 # print(all_date)

 for date in all_date:
     # print(date)
     key = date.get('username')
     # print(key)
     value = json.dumps(date)
     # print(value)
     r.hset('stu_info_6j',key,value)

简洁版：

 import pymysql,json,redis
 r = redis.Redis(host='xxx.xxx.xx.xxx',password='',db=1,port=6379)
 conn = pymysql.connect(host='xxx.xxx.xx.xxx',user='xxx',passwd='',db='xxx',charset='utf8')
 cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
 cur.execute('select * from my_user;')
 all_data = cur.fetchall()
 for data in all_data:
    k = data.get('username')
    r.hset('stu_info_nhy',k,json.dumps(data))
 cur.close()
 conn.close()