from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileAllowed
from wtforms import validators
from wtforms.validators import Length
from wtforms import (
    FileField,
    SubmitField,
    StringField,
)


class UploadForm(FlaskForm):
    """File upload class for a new site in distribusi-verse"""
    
    sitename = StringField(
        "Name of your website:",
        validators=[validators.InputRequired(), Length(6, 100)],
    )
    zipfile = FileField(
        'Upload your zip file with content here:',
        validators=[FileAllowed(['zip'], 'Zip archives only!')]
    )

    submit = SubmitField("Upload")