UUID7 functions for Python

uuid7.py

import random
import time
from uuid import UUID


UUID7_MASK = ((1 << 80) - 1) ^ (0xf << 76) ^ (0x3 << 62)
UUID7_SET = (0x7 << 76) | (0x2 << 62)


def uuid7() -> UUID:
    unix_ts_ms = ((time.time_ns() // 1_000_000) << 80) | UUID7_SET
    rand = random.getrandbits(80) & UUID7_MASK
    return UUID(int=unix_ts_ms | rand)


def uuid7_batch(n: int) -> list[UUID]:
    mask = UUID7_MASK
    getrandbits = random.getrandbits
    unix_ts_ms = ((time.time_ns() // 1_000_000) << 80) | UUID7_SET
    return [UUID(int=unix_ts_ms | (getrandbits(80) & mask)) for _ in range(n)]