def addContact():
form = AddContactForm()
customer_choices = [(a.id, a.name) for a in Customer.query.filter_by(customer_type=CUSTOMER_TYPES['TYPE_CUSTOMER']).all()]
customer_choices = [(0, '')] + customer_choices
form.customer.choices = customer_choices
if form.validate_on_submit():
contact = Contact()
contact.first_name = form.first_name.data
contact.surname = form.surname.data
contact.phone = form.phone.data
contact.email = form.email.data
if form.customer.data and form.customer.data != '' and form.customer.data != 0:
contact.customer_id = form.customer.data
else:
contact.customer_id = None
db.session.add(contact)
db.session.commit()
flash(gettext("New contact successfully added."))
return redirect(url_for("contacts"))
return render_template('settings/addContact.html',
title=gettext("Add New Contact"),
form=form)
def add_contact_form():
form = AddContactForm()
if request.method == 'POST' and form.validate_on_submit():
connect(mongo)
new_contact = Contact()
new_contact.name = form.name.data
new_contact.phone = form.phone.data
new_contact.email = form.email.data
new_contact.registration_date = str(date.today())
new_contact.save()
return 'Contact added'
return render_template("add_contact.html", form=form)
def addcontact():
form = AddContactForm()
if form.validate_on_submit():
client = MongoClient(MONGODB_URI)
db = client.get_default_database()
contacts = db.contacts
newContact = {
"name": form.name.data,
"number": form.number.data,
"description": form.description.data,
"user_id": current_user.id
}
contacts.insert(newContact)
return redirect('/index')
return render_template('addcontact.html',
user = current_user,
title = 'Add New Contact',
form = form)
