import os import secrets from PIL import Image from flask import url_for, current_app from flask_mail import Message from minibase.app import mail from minibase.blueprints.main.models import Industries import minibase.blueprints.database.utils as dbUtils class accountInfo: def __init__(self, title, description, short, status, image_file): self.title = title self.description = description self.short = short self.status = status self.image_file = image_file # FORM AND MODEL MANAGEMENT def fill_model(company, form): for item in form: modelFill(company, item) def populate_form(form, company): for item in form: item.data = formFill(item, company) def formFill(item, table): if item.type not in ["SubmitField", "CSRFTokenField"]: return getattr(table, item.name) def modelFill(table, item): if item.type not in ["SubmitField", "CSRFTokenField"]: if item.type == "FileField": if item.data == "None": print(f"File filed found with no data -> item name {item.name}") print(f"It will be filled with the existin tbale data -> item name {getattr(table, item.name)}") setattr(table, item.name, getattr(table, item.name)) else: setattr(table, item.name, item.data) # PICTURE MANAGEMENT def imageFileLink(imageFileName): return url_for('static', filename='pics/' + imageFileName) def save_picture(form_picture): random_hex = secrets.token_hex(8) _, f_ext = os.path.splitext(form_picture.filename) picture_fn = random_hex + f_ext picture_path = os.path.join(current_app.root_path, 'static/pics', picture_fn) output_size = (125, 125) i = Image.open(form_picture) i.thumbnail(output_size) i.save(picture_path) return picture_fn # MAIL MECHANISMS def send_reset_email(user): token = user.get_reset_token() msg = Message('Password Reset Request', sender='noreply@demo.com', recipients=[user.email_account]) msg.body = f'''To reset your password, visit the following link: {url_for('user.reset_token', token=token, _external=True)} If you didn't make this request, then simply ingnore this email and no changes will be made. ''' mail.send(msg) # QUERIES def queryIndustryNames(): choices = Industries.query.order_by(Industries.name.asc()) return choices def queryIndustryNamesWithDefault(defId): choices = dbUtils.queryNameWithDefaultId(Industries,defId) return choices