python練習——第3題

題目：將 0001 題生成的 200 個激活碼（或者優惠券）保存到 Redis 非關係型數據庫中。

代碼：

import uuid
import redis

# 生成激活碼，number爲數量，length爲長度
def generate_code(number, length):
    codes = []
    k = 0
    while(True):
        temp = str(uuid.uuid1()).replace("-", "")[:length]
        if k == number:
            break
        elif temp not in codes:
            codes.append(temp)
            k += 1
        else:
            continue
    return codes

# 插入數據庫
def insert_into_redis(codes):
    r = redis.Redis(host='127.0.0.1', port=6379, decode_responses=True)

    counter = 0
    for code in codes:
        r.set('code-%s' % counter, code)
        counter += 1
    print(r.get('code-0'))


if __name__ == '__main__':
    codes = generate_code(20, 12)
    insert_into_redis(codes)