reg/events.py
changeset 2 c11afa8623f7
child 6 057498d12450
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/reg/events.py	Thu Apr 08 22:28:15 2010 +0530
@@ -0,0 +1,41 @@
+#!/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()