functions.py

import random
import datetime

####################################
def read_list_from_csv(fname, start=1):
    with open(fname, 'r') as f:
        lines = f.readlines()
    
    out = []
    for line in lines[start:]:
        content = line.split(',')
        out.append(content[0].strip())
    
    return out

####################################
def create_random_groups(names, group_size=2, allow_less=False):
    n = len(names)
    groups = []

    if group_size > n:
        return groups

    while n >= group_size:
        group = []
        for i in range(group_size):
            name = random.choice(names)
            group.append(name)
            names.remove(name)
        
        groups.append(group)
        n = len(names)
    
    if allow_less:
        groups.append(names)
    else:
        n_rem = len(names)
        for i in range(n_rem):
            group = random.choice(groups)
            group.append(names[i])
    
    return groups

####################################
def show_one_group(group, id):
    print(f'Group {id}')
    print('------------')
    for item in group:
        print(f'{item} ', end='')
    print('')

####################################
def show_all_groups(groups):
    print('+++++++++++++++++++++++++++++++++')
    for i, group in zip(range(len(groups)), groups):
        show_one_group(group, i)
    print('_________________________________')

####################################
def save_to_csv(fname, groups):
    n = len(groups)
    with open(fname, 'w') as w:
        for i, group in zip(range(n), groups):
            names = ', '.join(group)
            w.write(f'Group {i + 1}, {names} \n')

        now = datetime.datetime.now()
        w.write(f' , \nCreated by GroupRandomiser @ {now.strftime("%Y-%m-%d %H:%M:%S")}\n')

####################################
def main(dataFile, groupFile, groupSize, allowLess, start=1):
    
    names = read_list_from_csv(dataFile, start=start)
    groups = create_random_groups(names, group_size=groupSize, allow_less=allowLess)
    if not groups:
        return False
    save_to_csv(groupFile, groups)
    return True