xemeds
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
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")
|