-
Notifications
You must be signed in to change notification settings - Fork 0
/
helpers.py
31 lines (22 loc) · 1.17 KB
/
helpers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import os
from jogoteca import app
from flask_wtf import FlaskForm
from wtforms import StringField, validators, SubmitField, PasswordField
class FormularioJogo(FlaskForm):
nome = StringField('Nome do Jogo', [validators.DataRequired(), validators.Length(min=1, max=50)])
categoria = StringField('Categoria', [validators.DataRequired(), validators.Length(min=1, max=40)])
console = StringField('Console', [validators.DataRequired(), validators.Length(min=1, max=20)])
salvar = SubmitField('Salvar')
class FormularioUsuario(FlaskForm):
nickname = StringField('Nickname', [validators.DataRequired(), validators.Length(min=1, max=8)])
senha = PasswordField('Senha', [validators.DataRequired(), validators.Length(min=1, max=100)])
login = SubmitField('Login')
def recuperar_img(id):
for nome_arquivo in os.listdir(app.config['UPLOAD_PATH']):
if f'capa{id}' in nome_arquivo:
return nome_arquivo
return 'capa_padrao.jpg'
def deletar_arquivo(id):
arquivo = recuperar_img(id)
if arquivo != 'capa_padrao.jpg':
os.remove(os.path.join(app.config['UPLOAD_PATH'], arquivo))