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='123456',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='123456',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='123456',db=1,port=6379)
conn = pymysql.connect(host='xxx.xxx.xx.xxx',user='xxx',passwd='123456',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()