From 69d5a098f830128414f178d9382c65221f06e4bb Mon Sep 17 00:00:00 2001
From: Mitch Riedstra <Mitch@riedstra.us>
Date: Fri, 10 Nov 2017 15:25:29 -0500
Subject: Initial version of registration system--needs work

---
 app/dispatchAuth/admin.py  |  4 ++--
 app/dispatchAuth/forms.py  | 11 +++++++++++
 app/dispatchAuth/models.py |  6 ++++++
 3 files changed, 19 insertions(+), 2 deletions(-)
 create mode 100644 app/dispatchAuth/forms.py

(limited to 'app/dispatchAuth')

diff --git a/app/dispatchAuth/admin.py b/app/dispatchAuth/admin.py
index 38ec7a6..d65390c 100644
--- a/app/dispatchAuth/admin.py
+++ b/app/dispatchAuth/admin.py
@@ -1,7 +1,7 @@
 from django import forms
 from django.contrib import admin
 from django.contrib.auth.models import Group
-from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
+from django.contrib.auth.admin import UserAdmin as BaseUserAdmin, UserAdmin
 from django.contrib.auth.forms import ReadOnlyPasswordHashField
 
 # Register your models here.
@@ -55,7 +55,7 @@ class UserChangeForm(forms.ModelForm):
         return self.initial["password"]
 
 
-class UserAdmin(BaseUserAdmin):
+class UserAdmin(UserAdmin):
     # The forms to add and change user instances
     form = UserChangeForm
     add_form = UserCreationForm
diff --git a/app/dispatchAuth/forms.py b/app/dispatchAuth/forms.py
new file mode 100644
index 0000000..99781f7
--- /dev/null
+++ b/app/dispatchAuth/forms.py
@@ -0,0 +1,11 @@
+from registration.forms import RegistrationForm
+from .models import User
+
+
+class UserForm(RegistrationForm):
+    # fields = ['email', 'first_name', 'last_name']
+    # model = User
+
+    class Meta:
+        fields = ['email', 'first_name', 'last_name']
+        model = User
diff --git a/app/dispatchAuth/models.py b/app/dispatchAuth/models.py
index 49e0a41..3132f1d 100644
--- a/app/dispatchAuth/models.py
+++ b/app/dispatchAuth/models.py
@@ -1,4 +1,5 @@
 from django.contrib.auth.models import AbstractUser, AbstractBaseUser, PermissionsMixin, BaseUserManager
+from django.core.mail import send_mail
 from django.core.urlresolvers import reverse
 from django.db import models
 
@@ -66,6 +67,11 @@ class User(PermissionsMixin, AbstractBaseUser):
     def __str__(self):
         return self.email
 
+    def email_user(self, subject, message, from_email=None, **kwargs):
+        """Send an email to this user."""
+        send_mail(subject, message, from_email, [self.email], **kwargs)
+
+
     @property
     def is_staff(self):
         return self.is_superuser
-- 
cgit v1.2.3