pythonによる４桁乱数（ユニーク生成）について

Question

python3 django2で商品管理システムを独学で作っています。

ある４桁の乱数をユニーク生成するコードなのですが、
必ず４桁にならなければいけないのですが０が前に来ると桁数が減ってしまいエラー原因になっています。

ハマってしまい、お知恵をお借りできれば幸いです。

また、こういう記述の方が良いよといったオススメの書き方があれば教えてください。

def number(specified_uuid=None):
    generate_number = lambda digits: ("{:0" + str(digits) + "d}").format(uuid.uuid4().int % 10 ** digits)

    uuid_candidate = specified_uuid if specified_uuid else generate_number(4)
    # DB に既に存在する乱数である場合、やり直す
    if Model.objects.filter(number=uuid_candidate).exists():
        return new_number(generate_number(4))

    return uuid_candidate

Answer 1

単純に、

impoer random
x = random.randrange(1000,9999)

ではどうでしょうか？

Comments

[hayataka2049]

これで良いですね。ただ、それだと1000 <= x <9999になります（第二引数は範囲に含まれません）
https://docs.python.jp/3/library/random.html#random.randrange
https://docs.python.jp/3/library/stdtypes.html#range

[otn]

ああ、失礼。
rangeと同じでしたね。

[Meteor]

ご回答ありがとうございました。
9999までの全ての数字を含めるとどのような記述になるのでしょうか？

また、ゼロパディングができる関数を見つけ欲していた機能でした。
例）１が出力された場合「0001」にしてくれる

[otn]

> 9999までの全ての数字を含めるとどのような記述になるのでしょうか？

??
0以上9999以下と言うことですか？
random.randrange(0,10000)
random.randrange(10000)
です。

> また、ゼロパディングができる関数を見つけ欲していた機能でした。

数値じゃなくて文字列で欲しいと言うことですね。
format(random.randrange(10000),"04")

Answer 2

そもそもなのですが、今回の４桁程度の重複しない番号を振りたい場合はDBに番号テーブルを用意すべきです。
これは0001～9999までをランダムにシャッフルして並べた以下のようなテーブルになります。

登録番号	商品番号
1	0158
2	6437
：	：

このテーブルは運用開始時に１回だけ作ればよいです。たとえば以下のようなコードで作れます。

def make_number_table(n):
    import random
    nums = list(range(1,10**n))
    random.shuffle(nums)

    DBに番号テーブルを新規作成する

    for reg_num, product_num in enumerate(nums):
        番号テーブルにレコード挿入する
        # print( '{rn},{pn:{width}d}'.format(rn=reg_num,pn=product_num,width=n))

すると、番号を振り出すnumber関数は以下のようになります。

def number(specified_uuid=None):

    if specified_uuid:
        商品番号がspecified_uuidに一致する行にカーソルを移動する
    else:
        登録番号が最も小さい（テーブルの先頭）行にカーソルを移動する

    if カーソルの指す行がない:
        throw KeyError

    uuid_candidate = カーソルの指す行の商品番号
    カーソルの指す行を番号テーブルから削除する

    return uuid_candidate

このような仕組みにすれば、番号の振り出し時にバッティングが発生しませんし
商品数が多くなっても（specified_uuid未指定なら）ほぼ定数時間で番号を振り出せます。

Answer 3

require random
random.randint(1000, 9999)
で 4 桁整数を得てから、DB 上のデータとの重複チェックをすればよいのでは？

質問文のコードで uuid を使っている理由は何でしょうか？

参考情報

• Pythonでランダムな小数・整数を生成するrandom, randrange, randintなど

https://note.nkmk.me/python-random-randrange-randint/

...
random.randint()
randomモジュールの関数randint(a, b)はa <= n <= bのランダムな整数intを返す
,,,

Answer 4

>>> import random
>>> import string
>>> int(random.choice(string.digits[1:]) + "".join(random.choices(string.digits, k=3)))
8328

ゼロパディングしたいのなら

>>> "{:0>4}".format(random.randrange(10000))
'5155'

これで'0000'から'9999'のランダムな文字列を返します。