def test_signup_download_staff(self):
# insert some vendors
v1 = Vendor.query.filter(Vendor.email == '*****@*****.**').first()
v1.categories.add(Category.query.first())
db.session.commit()
v2 = Vendor.create(**{'email': '*****@*****.**', 'business_name': 'foo'})
for i in Category.query.all():
v2.categories.add(i)
db.session.commit()
self.login_user(self.staff)
request = self.client.get('/beacon/admin/signups')
self.assertEquals(request.mimetype, 'text/tsv')
self.assertEquals(
request.headers.get('Content-Disposition'),
'attachment; filename=vendors-{}.tsv'.format(
datetime.date.today()))
# python adds an extra return character to the end,
# so we chop it off. we should have the header row and
# the two rows we inserted above
tsv_data = request.data.split('\n')[:-1]
self.assertEquals(len(tsv_data), 3)
for row in tsv_data:
self.assertEquals(len(row.split('\t')), 11)
def setUp(self):
super(TestOpportunitiesPublic, self).setUp()
self.opportunity3.is_public = False
self.opportunity3.categories = set([Category.query.all()[-1]])
self.opportunity3.planned_submission_end = datetime.datetime.today(
) + datetime.timedelta(days=5)
self.vendor = Vendor.create(business_name='foobar',
email='*****@*****.**',
categories=set([Category.query.all()[-1]]))
db.session.commit()
self.opportunity1.created_by = self.staff
self.opportunity3.created_by = self.staff
def setUp(self):
super(TestOpportunitiesAdminBase, self).setUp()
try:
mkdir(current_app.config.get('UPLOAD_DESTINATION'))
except OSError:
rmtree(current_app.config.get('UPLOAD_DESTINATION'))
mkdir(current_app.config.get('UPLOAD_DESTINATION'))
import_nigp(current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/nigp.csv')
self.admin_role = insert_a_role('admin')
self.staff_role = insert_a_role('staff')
self.department1 = DepartmentFactory(name='test')
self.opportunity_type = ContractTypeFactory.create(allow_opportunities=True)
self.admin = insert_a_user(email='*****@*****.**', role=self.admin_role)
self.staff = insert_a_user(email='*****@*****.**', role=self.staff_role)
self.document = insert_a_document()
self.vendor = Vendor.create(email='*****@*****.**', business_name='foo2')
self.opportunity1 = insert_an_opportunity(
contact=self.admin, created_by=self.staff,
title=u'tést unïcode title', description=u'tést unïcode déscription',
is_public=True, is_archived=False, planned_publish=datetime.date.today() + datetime.timedelta(1),
planned_submission_start=datetime.date.today() + datetime.timedelta(2),
planned_submission_end=datetime.datetime.today() + datetime.timedelta(2),
documents=[self.document.id], categories=set([Category.query.first()])
)
self.opportunity2 = insert_an_opportunity(
contact=self.admin, created_by=self.staff,
is_public=True, is_archived=False, planned_publish=datetime.date.today(),
planned_submission_start=datetime.date.today() + datetime.timedelta(2),
planned_submission_end=datetime.datetime.today() + datetime.timedelta(2),
categories=set([Category.query.first()])
)
self.opportunity3 = insert_an_opportunity(
contact=self.admin, created_by=self.staff,
is_public=True, is_archived=False, planned_publish=datetime.date.today() - datetime.timedelta(2),
planned_submission_start=datetime.date.today() - datetime.timedelta(2),
planned_submission_end=datetime.datetime.today() - datetime.timedelta(1),
categories=set([Category.query.first()])
)
self.opportunity4 = insert_an_opportunity(
contact=self.admin, created_by=self.staff,
is_public=True, is_archived=False, planned_publish=datetime.date.today() - datetime.timedelta(1),
planned_submission_start=datetime.date.today(),
planned_submission_end=datetime.datetime.today() + datetime.timedelta(2),
title='TEST TITLE!', categories=set([Category.query.first()])
)
def setUp(self):
super(TestOpportunitiesPublic, self).setUp()
self.opportunity3.is_public = False
self.opportunity3.categories = set([Category.query.all()[-1]])
self.opportunity3.planned_submission_end = datetime.datetime.today() + datetime.timedelta(2)
self.vendor = Vendor.create(
business_name='foobar',
email='*****@*****.**',
categories=set([Category.query.all()[-1]])
)
db.session.commit()
self.opportunity1.created_by = self.staff
self.opportunity3.created_by = self.staff
def test_manage_subscriptions(self):
v = Vendor.create(
email='*****@*****.**', business_name='foo', subscribed_to_newsletter=True
)
for i in Category.query.filter(Category.id.in_([1,2,3])):
v.categories.add(i)
db.session.commit()
manage = self.client.post('/beacon/manage', data=dict(
email='*****@*****.**'
))
self.assert200(manage)
form = self.get_context_variable('form')
self.assertEquals(len(form.categories.choices), 3)
# it shouldn't unsubscribe you if you click the wrong button
not_unsub_button = self.client.post('/beacon/manage', data=dict(
email='*****@*****.**',
categories=[1, 2],
))
self.assert200(not_unsub_button)
form = self.get_context_variable('form')
self.assertEquals(len(form.categories.choices), 3)
unsubscribe = self.client.post('/beacon/manage', data=dict(
email='*****@*****.**',
categories=[1, 2],
button='Update email preferences'
))
self.assert200(unsubscribe)
form = self.get_context_variable('form')
self.assertEquals(len(form.categories.choices), 1)
# it shouldn't matter if you somehow unsubscribe from things
# you are accidentally subscribed to
unsubscribe_all = self.client.post('/beacon/manage', data=dict(
email='*****@*****.**',
categories=[3, 5, 6],
button='Update email preferences',
subscribed_to_newsletter=False
))
self.assert200(unsubscribe_all)
self.assertTrue('You are not subscribed to anything!' in unsubscribe_all.data)
def signup():
'''The signup page for vendors
'''
session_vendor = Vendor.query.filter(
Vendor.email == session.get('email'),
Vendor.business_name == session.get('business_name')
).first()
form = init_form(VendorSignupForm, model=session_vendor)
if form.validate_on_submit():
vendor = Vendor.query.filter(Vendor.email == form.data.get('email')).first()
if vendor:
current_app.logger.info('''
OPPUPDATEVENDOR - Vendor updated:
EMAIL: {old_email} -> {email} at
BUSINESS: {old_bis} -> {bis_name} signed up for:
CATEGORIES:
{old_cats} ->
{categories}'''.format(
old_email=vendor.email, email=form.data['email'],
old_bis=vendor.business_name, bis_name=form.data['business_name'],
old_cats=[i.__unicode__() for i in vendor.categories],
categories=[i.__unicode__() for i in form.data['categories']]
))
vendor.update(
**form.pop_categories(categories=False)
)
flash("You are already signed up! Your profile was updated with this new information", 'alert-info')
else:
current_app.logger.info(
'OPPNEWVENDOR - New vendor signed up: EMAIL: {email} at BUSINESS: {bis_name} signed up for:\n' +
'CATEGORIES: {categories}'.format(
email=form.data['email'],
bis_name=form.data['business_name'],
categories=[i.__unicode__() for i in form.data['categories']]
)
)
vendor = Vendor.create(
**form.pop_categories(categories=False)
)
confirmation_sent = Notification(
to_email=vendor.email, subject='Thank you for signing up!',
html_template='opportunities/emails/signup.html',
txt_template='opportunities/emails/signup.txt',
categories=form.data['categories']
).send()
if confirmation_sent:
admins = db.session.query(User.email).join(Role, User.role_id == Role.id).filter(
Role.name.in_(['admin', 'superadmin'])
).all()
Notification(
to_email=admins, subject='A new vendor has signed up on beacon',
categories=form.data['categories'],
vendor=form.data['email'], convert_args=True,
business_name=form.data['business_name']
).send()
flash('Thank you for signing up! Check your email for more information', 'alert-success')
else:
flash('Uh oh, something went wrong. We are investigating.', 'alert-danger')
session['email'] = form.data.get('email')
session['business_name'] = form.data.get('business_name')
return redirect(url_for('opportunities.splash'))
page_email = request.args.get('email', None)
if page_email:
current_app.logger.info(
'OPPSIGNUPVIEW - User clicked through to signup with email {}'.format(page_email)
)
session['email'] = page_email
return redirect(url_for('opportunities.signup'))
if 'email' in session:
if not form.email.validate(form):
session.pop('email', None)
form.display_cleanup()
return render_template(
'opportunities/front/signup.html', form=form,
categories=form.get_categories(),
subcategories=form.get_subcategories()
)
def signup():
'''
The signup page for vendors
'''
all_categories = Category.query.all()
categories, subcategories = set(), defaultdict(list)
for category in all_categories:
categories.add(category.category)
subcategories['Select All'].append((category.id, category.subcategory))
subcategories[category.category].append((category.id, category.subcategory))
form = SignupForm()
form.categories.choices = [(None, '---')] + list(sorted(zip(categories, categories))) + [('Select All', 'Select All')]
form.subcategories.choices = []
if form.validate_on_submit():
vendor = Vendor.query.filter(Vendor.email == form.data.get('email')).first()
form_data = {c.name: form.data.get(c.name, None) for c in Vendor.__table__.columns if c.name not in ['id', 'created_at']}
form_data['categories'] = []
subcats = set()
# manually iterate the form fields
for k, v in request.form.iteritems():
if not k.startswith('subcategories-'):
continue
else:
subcat_id = int(k.split('-')[1])
# make sure the field is checked (or 'on') and we don't have it already
if v == 'on' and subcat_id not in subcats:
subcats.add(subcat_id)
subcat = Category.query.get(subcat_id)
# make sure it's a valid subcategory
if subcat is None:
raise ValidationError('{} is not a valid choice!'.format(subcat))
form_data['categories'].append(subcat)
if vendor:
current_app.logger.info('OPPPUPDATEVENDOR - Vendor updated: EMAIL: {old_email} -> {email} at BUSINESS: {old_bis} -> {bis_name} signed up for:\n CATEGORIES: {old_cats} ->\n {categories}'.format(
old_email=vendor.email,
email=form_data['email'],
old_bis=vendor.business_name,
bis_name=form_data['business_name'],
old_cats=[i.__unicode__() for i in vendor.categories],
categories=[i.__unicode__() for i in form_data['categories']]
))
vendor.update(
**form_data
)
flash("You are already signed up! Your profile was updated with this new information", 'alert-info')
else:
current_app.logger.info(
'OPPNEWVENDOR - New vendor signed up: EMAIL: {email} at BUSINESS: {bis_name} signed up for:\n CATEGORIES: {categories}'.format(
email=form_data['email'],
bis_name=form_data['business_name'],
categories=[i.__unicode__() for i in form_data['categories']]
)
)
vendor = Vendor.create(
**form_data
)
confirmation_sent = vendor_signup(vendor, categories=form_data['categories'])
if confirmation_sent:
flash('Thank you for signing up! Check your email for more information', 'alert-success')
else:
flash('Uh oh, something went wrong. We are investigating.', 'alert-danger')
return redirect(url_for('opportunities.index'))
page_email = request.args.get('email', None)
if page_email:
current_app.logger.info('OPPSIGNUPVIEW - User clicked through to signup with email {}'.format(page_email))
session['email'] = page_email
return redirect(url_for('opportunities.signup'))
if 'email' in session:
form.email.data = session['email']
form.email.validate(form)
session.pop('email')
display_categories = subcategories.keys()
display_categories.remove('Select All')
return render_template(
'opportunities/signup.html', form=form,
subcategories=json.dumps(subcategories),
categories=json.dumps(
sorted(display_categories) + ['Select All']
)
)
def signup():
"""The signup page for vendors
"""
session_vendor = Vendor.query.filter(
Vendor.email == session.get("email"), Vendor.business_name == session.get("business_name")
).first()
form = init_form(VendorSignupForm, model=session_vendor)
if form.validate_on_submit():
vendor = Vendor.query.filter(Vendor.email == form.data.get("email")).first()
if vendor:
current_app.logger.info(
"""
OPPUPDATEVENDOR - Vendor updated:
EMAIL: {old_email} -> {email} at
BUSINESS: {old_bis} -> {bis_name} signed up for:
CATEGORIES:
{old_cats} ->
{categories}""".format(
old_email=vendor.email,
email=form.data["email"],
old_bis=vendor.business_name,
bis_name=form.data["business_name"],
old_cats=[i.__unicode__() for i in vendor.categories],
categories=[i.__unicode__() for i in form.data["categories"]],
)
)
vendor.update(**form.pop_categories(categories=False))
flash("You are already signed up! Your profile was updated with this new information", "alert-info")
else:
current_app.logger.info(
"OPPNEWVENDOR - New vendor signed up: EMAIL: {email} at BUSINESS: {bis_name} signed up for:\n"
+ "CATEGORIES: {categories}".format(
email=form.data["email"],
bis_name=form.data["business_name"],
categories=[i.__unicode__() for i in form.data["categories"]],
)
)
vendor = Vendor.create(**form.pop_categories(categories=False))
confirmation_sent = Notification(
to_email=vendor.email,
subject="Thank you for signing up!",
html_template="opportunities/emails/signup.html",
txt_template="opportunities/emails/signup.txt",
categories=form.data["categories"],
).send()
if confirmation_sent:
admins = (
db.session.query(User.email)
.join(Role, User.role_id == Role.id)
.filter(Role.name.in_(["admin", "superadmin"]))
.all()
)
Notification(
to_email=admins,
subject="A new vendor has signed up on beacon",
categories=form.data["categories"],
vendor=form.data["email"],
convert_args=True,
business_name=form.data["business_name"],
).send()
flash("Thank you for signing up! Check your email for more information", "alert-success")
else:
flash("Uh oh, something went wrong. We are investigating.", "alert-danger")
session["email"] = form.data.get("email")
session["business_name"] = form.data.get("business_name")
return redirect(url_for("opportunities.splash"))
page_email = request.args.get("email", None)
if page_email:
current_app.logger.info("OPPSIGNUPVIEW - User clicked through to signup with email {}".format(page_email))
session["email"] = page_email
return redirect(url_for("opportunities.signup"))
if "email" in session:
if not form.email.validate(form):
session.pop("email", None)
form.display_cleanup()
return render_template(
"opportunities/front/signup.html",
form=form,
categories=form.get_categories(),
subcategories=form.get_subcategories(),
)