flask_url_shortener/tiny0/forms.py

from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length, Optional
from tiny0.config import WEBSITE_DOMAIN
from tiny0 import db
from tiny0.models import URL

# 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:
		return

	# If the url contains spaces or does not have any dots
	if (" " in field.data) or not("." in field.data):
		# Raise a ValidationError
		raise ValidationError("Invalid URL")

	# If the url starts with a dot after http:// or after https:// or just starts with a dot
	if field.data.lower().startswith("http://.") or field.data.lower().startswith("https://.") or field.data.startswith("."):
		# Raise a ValidationError
		raise ValidationError("Invalid URL")

	# If the url starts with a slash after http:// or after https:// or just starts with a slash
	if field.data.lower().startswith("http:///") or field.data.lower().startswith("https:///") or field.data.startswith("/"):
		# Raise a ValidationError
		raise ValidationError("Invalid URL")

	# If the url ends with a dot and it is the only dot
	if field.data.endswith(".") and field.data.count(".") == 1:
		# Raise a ValidationError
		raise ValidationError("Invalid URL")

	# If the url contains the websites domain
	if WEBSITE_DOMAIN in field.data.lower():
		# Raise a ValidationError
		raise ValidationError("Invalid URL")

	# If the URL does not start with http:// and https://
	if not(field.data.lower().startswith("http://")) and not(field.data.lower().startswith("https://")):
		# Add http:// to the beginning of the URL
		field.data = "http://" + field.data

# Validates a token
def validate_token(form, field):
	# Make sure the token is not too short or long
	if len(field.data) < 6 or len(field.data) > 16:
		return

	# For each character in the token
	for char in field.data:
		# If it is not a valid character
		if not(char.isalpha()) and not(char.isdigit()) and not(char == "_") and not(char == '-'):
			# Raise a ValidationError
			raise ValidationError("Token contains invalid characters")

	# If the token exists in the database
	if db.session.query(db.session.query(URL).filter_by(token=field.data).exists()).scalar():
		# Raise a ValidationError
		raise ValidationError("Token already exists")

class URLForm(FlaskForm):
	url = StringField(validators=[DataRequired(), 
		Length(min=4, max=2000, message="Invalid URL length"), 
		validate_URL])

	token = StringField(validators=[Optional(), Length(min=6, max=16, message="Invalid token length"), validate_token])

	submit = SubmitField("Shorten")