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)
