user_groups.py

import random
import string

def generate_unique_code(existing_codes, length=6):
    while True:
        code = ''.join(random.choices(string.ascii_lowercase + string.digits, k=length))
        if code not in existing_codes:
            return code

def generate_user_data(num_codes_per_group=10):
    user_data = {}
    existing_codes = set()
    groups = ["A", "B", "C"]
    
    for group in groups:
        for _ in range(num_codes_per_group):
            code = generate_unique_code(existing_codes)
            existing_codes.add(code)
            user_data[code] = {"group": group}
    
    return user_data

if __name__ == "__main__":
    user_data = generate_user_data()
    print(user_data)

    import json

    with open('user_data.json', 'w') as file:
        json.dump(user_data, file)
else:
    # test user data
    user_data = {
        "code123": {"group": "A"},
        "code456": {"group": "B"},
        "code789": {"group": "C"},
    }