1 files changed, 34 insertions, 0 deletions
diff --git a/app/modules/auth/forms.py b/app/modules/auth/forms.py
new file mode 100644
index 0000000..a4b4555
--- /dev/null
+++ b/app/modules/auth/forms.py
@@ -0,0 +1,34 @@
+# -*- mode: python; -*-
+
+
+from flask_wtf import FlaskForm
+from wtforms import (
+    SubmitField,
+    HiddenField,
+    StringField,
+    PasswordField,
+    BooleanField,
+)
+from wtforms.validators import (
+    InputRequired,
+    Length,
+    ValidationError,
+)
+
+
+class LoginForm(FlaskForm):
+    username = StringField("Username", validators=[InputRequired()])
+    password = PasswordField("Password", validators=[InputRequired()])
+    remember = BooleanField("Remember")
+    submit = SubmitField("Login")
+
+
+class RegisterForm(LoginForm):
+    def validate_invite_code(self, field):
+        if field.data != "mdltesters2022":
+            raise ValidationError("Invitation code does not match")
+
+    invitation_code = StringField("Invitation code", validators=[InputRequired()])
+    name_first = StringField("First name", validators=[InputRequired()])
+    name_last = StringField("Last name", validators=[InputRequired()])
+    submit = SubmitField("Register")