#!/usr/bin/python

from django.contrib.auth.models import User

from django.db import IntegrityError

from workshop.reg.models import Event

from workshop.reg.utils import gen_key

def create_user(email, password, firstname="", lastname=""):

    """ create a user with random username and set the password.

    """

    username = ('%s %s'%(firstname, lastname)).title().strip()

    new_user = User.objects.create_user(username, email, password)

    new_user.firstname = firstname

    new_user.lastname = lastname

    new_user.is_active = False

    new_user.save()

    return new_user

def create_event(title, desc, start_date, stop_date, created_by=None):

    """ make an event and save it.

    """

    new_event = Event()

    new_event.title = title

    new_event.description = desc

    new_event.start_date = start_date

    new_event.stop_date = stop_date

    new_event.save()

    if created_by:

        new_event.organizers.add(created_by)

    new_event.key = gen_key()

    while True:

        try:

            new_event.save()

            return new_event

        except IntegrityError:

            new_event.key = gen_key()