def add_product_to_agent(product_type, agent=None): from mu.model.domain.user import UserDomain if UserDomain.username_of_session() is None or not ProductDomain.is_valid_product_type(product_type): # , 401 return redirect(url_for('home.show_home')) from mu.form.add_product import AddProductForm, AddProductTypeForm if product_type.lower() == 'release': add_product_form = AddProductForm(request.form, obj={ 'agent_type': 'Artist' }) else: add_product_form = AddProductTypeForm(request.form, obj={ 'agent_type': 'Artist', 'product_type': product_type }) if request.method == "POST" and add_product_form.validate(): product_info = { 'agent_name': request.form.get('agent'), 'agent_type_id': request.form.get('agent_type'), 'content_owner_name': request.form.get('content_owner'), 'event_release_date': request.form.get('release_date'), 'product_title': request.form.get('title'), 'product_type_id': request.form.get('type'), 'product_status_id': request.form.get('status'), 'product_medium_id': request.form.get('medium') } ProductDomain.add_product(**product_info) return render_template('product/add_product.html', product_type=product_type, agent=agent, form=add_product_form)
def show_products(product_type=None): products = ProductDomain.get_products(product_type) return render_template('product/show_products.html', products=products)