116 lines
5.0 KiB
Python
116 lines
5.0 KiB
Python
"""
|
|
Project: PARTS Website
|
|
Author: Edward Middleton-Smith
|
|
Precision And Research Technology Systems Limited
|
|
|
|
Technology: App Routing
|
|
Feature: Dog - Company Routes
|
|
|
|
Description:
|
|
Contact Page Controller.
|
|
"""
|
|
|
|
# IMPORTS
|
|
# internal
|
|
from business_objects.api import API
|
|
from business_objects.dog.company import Company
|
|
from datastores.datastore_dog import DataStore_Dog
|
|
from datastores.datastore_user import DataStore_User
|
|
from forms.dog.company import Filters_Company
|
|
from helpers.helper_app import Helper_App
|
|
from models.model_view_user_company import Model_View_User_Company
|
|
from models.model_view_home import Model_View_Home
|
|
import lib.argument_validation as av
|
|
# external
|
|
from flask import Flask, render_template, jsonify, request, render_template_string, send_from_directory, redirect, url_for, session, Blueprint, current_app, flash
|
|
from flask_mail import Mail, Message
|
|
from extensions import db, oauth, mail
|
|
from urllib.parse import quote_plus, urlencode
|
|
from authlib.integrations.flask_client import OAuth
|
|
from authlib.integrations.base_client import OAuthError
|
|
from urllib.parse import quote, urlparse, parse_qs
|
|
import json
|
|
import base64
|
|
import hmac
|
|
import hashlib
|
|
import datetime
|
|
from altcha import ChallengeOptions, create_challenge, verify_solution
|
|
|
|
|
|
routes_user_company = Blueprint('routes_user_company', __name__)
|
|
|
|
"""
|
|
@routes_user_company.route(Model_View_User_Company.HASH_PAGE_USER_COMPANIES, methods=['GET'])
|
|
def companies():
|
|
Helper_App.console_log('companies')
|
|
Helper_App.console_log(f'request_args: {request.args}')
|
|
try:
|
|
form_filters = Filters_Company.from_json(request.args)
|
|
except Exception as e:
|
|
Helper_App.console_log(f'Error: {e}')
|
|
form_filters = Filters_Company()
|
|
Helper_App.console_log(f'form_filters={form_filters}')
|
|
model = Model_View_User_Company(form_filters_old = form_filters)
|
|
if not model.is_user_logged_in:
|
|
return redirect(url_for('routes_dog_home.home'))
|
|
Helper_App.console_log(f'form_filters={form_filters}')
|
|
return render_template('pages/user/_companies.html', model = model)
|
|
"""
|
|
@routes_user_company.route(Model_View_User_Company.HASH_PAGE_USER_COMPANY, methods=['GET'])
|
|
def company():
|
|
Helper_App.console_log('company')
|
|
Helper_App.console_log(f'request_args: {request.args}')
|
|
try:
|
|
form_filters = Filters_Company.from_json(request.args)
|
|
except Exception as e:
|
|
Helper_App.console_log(f'Error: {e}')
|
|
form_filters = Filters_Company()
|
|
Helper_App.console_log(f'form_filters={form_filters}')
|
|
model = Model_View_User_Company(form_filters_old = form_filters)
|
|
if (not model.is_user_logged_in) or (not model.user.can_edit_company) or (len(model.companies) == 0):
|
|
return redirect(url_for('routes_dog_home.home'))
|
|
Helper_App.console_log(f'form_filters={form_filters}')
|
|
return render_template('pages/user/_company.html', model = model)
|
|
|
|
@routes_user_company.route(Model_View_User_Company.HASH_SAVE_USER_COMPANY, methods=['POST'])
|
|
def save_company():
|
|
data = Helper_App.get_request_data(request)
|
|
try:
|
|
form_filters = Filters_Company.from_json(data[Model_View_User_Company.FLAG_FORM_FILTERS])
|
|
if not form_filters.validate_on_submit():
|
|
return jsonify({
|
|
Model_View_User_Company.FLAG_STATUS: Model_View_User_Company.FLAG_FAILURE,
|
|
Model_View_User_Company.FLAG_MESSAGE: f'Filters form invalid.\n{form_filters.errors}'
|
|
})
|
|
model_return = Model_View_User_Company(form_filters_old=form_filters)
|
|
if not model_return.is_user_logged_in:
|
|
return redirect(url_for('routes_core_home.home'))
|
|
if not model_return.user.can_edit_company:
|
|
return redirect(url_for('routes_dog_home.home'))
|
|
|
|
companies = data[Model_View_User_Company.FLAG_COMPANY]
|
|
if len(companies) == 0:
|
|
return jsonify({
|
|
Model_View_User_Company.FLAG_STATUS: Model_View_User_Company.FLAG_FAILURE,
|
|
Model_View_User_Company.FLAG_MESSAGE: f'No companies.'
|
|
})
|
|
objs_company = []
|
|
for company in companies:
|
|
objs_company.append(Company.from_json(company))
|
|
Helper_App.console_log(f'objs_company={objs_company}')
|
|
errors = DataStore_User.save_companies(data.get('comment', 'No comment'), objs_company)
|
|
|
|
if (len(errors) > 0):
|
|
return jsonify({
|
|
Model_View_User_Company.FLAG_STATUS: Model_View_User_Company.FLAG_FAILURE,
|
|
Model_View_User_Company.FLAG_MESSAGE: f'Error saving companies.\n{model_return.convert_list_objects_to_json(errors)}'
|
|
})
|
|
return jsonify({
|
|
Model_View_User_Company.FLAG_STATUS: Model_View_User_Company.FLAG_SUCCESS,
|
|
Model_View_User_Company.FLAG_DATA: Model_View_User_Company.convert_list_objects_to_json(model_return.companies)
|
|
})
|
|
except Exception as e:
|
|
return jsonify({
|
|
Model_View_User_Company.FLAG_STATUS: Model_View_User_Company.FLAG_FAILURE,
|
|
Model_View_User_Company.FLAG_MESSAGE: f'Bad data received by controller.\n{e}'
|
|
}) |