def signup_for_opp(form, opportunity, multi=False): '''Sign a vendor up for an opportunity Generic helper method to handle subscriptions from both the list view (signing up form multiple opportunities) and the detail view (signing up for a single opportunity). Responsible for creation of new Vendor objects if necessary, and sending emails based on the opportunities selected to receive updates about. Arguments: form: The relevant subscription form opportunity: Either an opportunity model or a list of opportunity ids multi: A boolean to flag if there are multiple opportunities that should to subscribe to or a single opportunity Returns: True if email sent successfully, false otherwise ''' send_email = True email_opportunities = [] if opportunity is None or (isinstance(opportunity, list) and len(opportunity) == 0): form.errors['opportunities'] = [ 'You must select at least one opportunity!' ] return False # add the email/business name to the session session['email'] = form.data.get('email') session['business_name'] = form.data.get('business_name') # subscribe the vendor to the opportunity vendor = Vendor.query.filter( Vendor.email == form.data.get('email')).first() if vendor is None: vendor = Vendor(email=form.data.get('email'), business_name=form.data.get('business_name')) db.session.add(vendor) db.session.commit() else: vendor.update(business_name=form.data.get('business_name')) if multi: for opp in opportunity: _opp = Opportunity.query.get(int(opp)) if not _opp.is_public: db.session.rollback() form.errors['opportunities'] = ['That\'s not a valid choice.'] return False if _opp in vendor.opportunities: send_email = False else: vendor.opportunities.add(_opp) email_opportunities.append(_opp) else: if opportunity in vendor.opportunities: send_email = False else: vendor.opportunities.add(opportunity) email_opportunities.append(opportunity) if form.data.get('also_categories'): # TODO -- add support for categories pass db.session.commit() current_app.logger.info( u'OPPSIGNUP - Vendor has signed up for opportunities: EMAIL: {email} at BUSINESS: {bis_name} signed up for:\n' + u'OPPORTUNITY: {opportunities}'.format( email=form.data.get('email'), business_name=form.data.get('business_name'), opportunities=', '.join([i.title for i in email_opportunities]))) if send_email: Notification(to_email=vendor.email, from_email=current_app.config['BEACON_SENDER'], subject='Subscription confirmation from Beacon', html_template='opportunities/emails/oppselected.html', txt_template='opportunities/emails/oppselected.txt', opportunities=email_opportunities).send() return True
def manage(): '''Manage a vendor's signups :status 200: render the :py:class:`~purchasing.opportunities.forms.UnsubscribeForm` :status 302: post the :py:class:`~purchasing.opportunities.forms.UnsubscribeForm` and change the user's email subscriptions and redirect them back to the management page. ''' form = init_form(UnsubscribeForm) form_categories = [] form_opportunities = [] vendor = None if form.validate_on_submit(): email = form.data.get('email') vendor = Vendor.query.filter(Vendor.email == email).first() if vendor is None: current_app.logger.info( 'OPPMANAGEVIEW - Unsuccessful search for email {}'.format( email)) form.email.errors = [ 'We could not find the email {}'.format(email) ] if request.form.get('button', '').lower() == 'update email preferences': remove_categories = set( [Category.query.get(i) for i in form.categories.data]) remove_opportunities = set( [Opportunity.query.get(i) for i in form.opportunities.data]) # remove none types if any remove_categories.discard(None) remove_opportunities.discard(None) vendor.categories = vendor.categories.difference(remove_categories) vendor.opportunities = vendor.opportunities.difference( remove_opportunities) if form.data.get('subscribed_to_newsletter'): vendor.subscribed_to_newsletter = False current_app.logger.info( u'''OPPMANAGEVIEW - Vendor {} unsubscribed from: Categories: {} Opportunities: {} Subscribed from newsletter: {} '''.format( email, u', '.join([ i.category_friendly_name for i in remove_categories if remove_categories and len(remove_categories) > 0 ]), u', '.join([ i.description for i in remove_opportunities if remove_opportunities and len(remove_opportunities) > 0 ]), vendor.subscribed_to_newsletter)) db.session.commit() flash('Preferences updated!', 'alert-success') if vendor: for subscription in vendor.categories: form_categories.append( (subscription.id, subscription.category_friendly_name)) for subscription in vendor.opportunities: form_opportunities.append( (subscription.id, subscription.title)) form = init_form(UnsubscribeForm) form.opportunities.choices = form_opportunities form.categories.choices = form_categories return render_template('opportunities/front/manage.html', form=form, vendor=vendor if vendor else Vendor())