Added URL validation form

2020-07-18 16:40:51 +00:00
parent c783e7792a
commit d97b3266c8
5 changed files with 68 additions and 8 deletions
@@ -1,9 +1,26 @@
 from flask_wtf import FlaskForm
-from wtforms import StringField, SubmitField
-from wtforms.validators import DataRequired, Length, URL
+from wtforms import StringField, SubmitField, ValidationError
+from wtforms.validators import DataRequired, Length

-'''
+# Validates a URL
+def validate_URL(form, field):
+	# Make sure the url is not too short or long
+	if len(field.data) < 4 or len(field.data) > 2000:
+		# Raise a ValidationError
+		raise ValidationError()

-Declaration of forms
+	# If the url contains spaces or does not have any dots
+	if field.data.count(" ") > 0 or field.data.count(".") == 0:
+		# Raise a ValidationError
+		raise ValidationError()

-'''
+	# If the URL does not start with http:// and https://
+	if not(field.data.startswith("http://")) and not(field.data.startswith("https://")):
+		# Add https:// to the beginning of the URL
+		field.data = "https://" + field.data
+
+
+class URLForm(FlaskForm):
+	url = StringField(validators=[DataRequired(), Length(min=4, max=2000), validate_URL])
+
+	submit = SubmitField("Shorten URL")
@@ -1,9 +1,27 @@
 from flask import render_template, redirect, request, flash, url_for
 from tiny0 import app
-#from tiny0.forms import URLForm
+from tiny0.forms import URLForm
 #from tiny0.models import URLs
 #from token import gen_valid_token

-@app.route("/")
+# Index Page
+@app.route("/", methods=['GET', 'POST'])
 def index():
-	return "Hello, world!"
+	# Get request
+	if request.method == "GET":
+		# Create a instance of the form
+		form = URLForm()
+
+		# Return the index page with the form
+		return render_template("index.html", form=form)
+
+	# Post request
+	else:
+		# Create a instance of the form
+		form = URLForm()
+
+		# If the form was valid
+		if form.validate_on_submit():
+			return "Valid URL"
+
+		return render_template("index.html", form=form)
@@ -0,0 +1,3 @@
+body {
+	background: #2c2c2c;
+}
@@ -0,0 +1,11 @@
+{% extends "layout.html" %}
+
+{% block body %}
+	<div>
+		<form method="POST" action="">
+			{{ form.hidden_tag() }}
+			{{ form.url(placeholder="Enter the URL here") }}
+			{{ form.submit }}
+		</form>
+	</div>
+{% endblock %}
@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html>
+	<head>
+		<link href="/static/style.css" rel="stylesheet" type="text/css"/>
+		
+		<title>tiny0 - Custom URL Shortener</title>
+	</head>
+	<body>
+		{% block body %}{% endblock %}
+	</body>
+</html>