def test_add_candidate(self): """ Test the add_candidate function. """ create_elections(self.session) self.assertRaises( nuancierlib.NuancierException, nuancierlib.add_candidate, session=self.session, candidate_file='test.png', candidate_name='test image', candidate_author='pingou', candidate_license='CC-BY-SA', candidate_submitter='pingou', submitter_email='[email protected]', candidate_original_url=None, election_id=2, ) nuancierlib.add_candidate( session=self.session, candidate_file='test.png', candidate_name='test image', candidate_author='pingou', candidate_license='CC-BY-SA', candidate_submitter='pingou', submitter_email='[email protected]', candidate_original_url=None, election_id=2, user='******', ) self.session.commit() candidates = nuancierlib.get_candidates(self.session, 2, False) self.assertEqual(1, len(candidates)) self.assertEqual('test image', candidates[0].candidate_name) self.assertEqual('test.png', candidates[0].candidate_file) candidates = nuancierlib.get_candidates(self.session, 2, True) self.assertEqual(0, len(candidates)) self.assertRaises( nuancierlib.NuancierException, nuancierlib.add_candidate, session=self.session, candidate_file='test.png', candidate_name='test image', candidate_author='pingou', candidate_license='CC-BY-SA', candidate_submitter='pingou', submitter_email='[email protected]', candidate_original_url='http://example.org', election_id=2, user='******', )
def test_add_candidate(self): """ Test the add_candidate function. """ create_elections(self.session) nuancierlib.add_candidate( session=self.session, candidate_file='test.png', candidate_name='test image', candidate_author='pingou', election_id=2, ) self.session.commit() candidates = nuancierlib.get_candidates(self.session, 2) self.assertEqual(1, len(candidates)) self.assertEqual('test image', candidates[0].candidate_name) self.assertEqual('test.png', candidates[0].candidate_file)
def contribute(election_id): ''' Display the index page for interested contributor. ''' election = nuancierlib.get_election(SESSION, election_id) if not election: flask.flash('No election found', 'error') return flask.render_template('msg.html') elif not election.submission_open: flask.flash('This election is not open for submission', 'error') return flask.redirect(flask.url_for('elections_list')) candidates = nuancier.lib.model.Candidates.get_by_submitter( SESSION, flask.g.fas_user.username, election_id) if election.user_n_candidates and \ len(candidates) >= election.user_n_candidates: flask.flash( 'You have uploaded the maximum number of candidates (%s) you ' 'can upload for this election' % election.user_n_candidates, 'error') return flask.redirect(flask.url_for('elections_list')) form = nuancier.forms.AddCandidateForm() if form.validate_on_submit(): candidate_file = flask.request.files['candidate_file'] try: validate_input_file(candidate_file) except nuancierlib.NuancierException as err: LOG.debug( 'ERROR: Uploaded file is invalid - user: "******" ' 'election: "%s"', flask.g.fas_user.username, election_id) LOG.exception(err) flask.flash('%s' % err, 'error') return flask.render_template('contribute.html', election=election, form=form) filename = secure_filename( '%s-%s' % (flask.g.fas_user.username, candidate_file.filename)) # Only save the file once everything has been safely saved in the DB upload_folder = os.path.join(APP.config['PICTURE_FOLDER'], election.election_folder) if not os.path.exists(upload_folder): # pragma: no cover try: os.mkdir(upload_folder) except OSError as err: LOG.debug('ERROR: cannot add candidate file') LOG.exception(err) flask.flash( 'An error occured while writing the file, please ' 'contact an administrator', 'error') return flask.render_template('contribute.html', election=election, form=form) # Save candidate to the database try: nuancierlib.add_candidate( SESSION, candidate_file=filename, candidate_name=form.candidate_name.data, candidate_author=form.candidate_author.data, candidate_original_url=form.candidate_original_url.data, candidate_license=form.candidate_license.data, candidate_submitter=flask.g.fas_user.username, submitter_email=flask.g.fas_user.email, election_id=election.id, user=flask.g.fas_user.username, ) except nuancierlib.NuancierException as err: flask.flash('%s' % err, 'error') return flask.render_template('contribute.html', election=election, form=form) # The PIL module has already read the stream so we need to back up candidate_file.seek(0) candidate_file.save(os.path.join(upload_folder, filename)) try: SESSION.commit() except SQLAlchemyError as err: # pragma: no cover SESSION.rollback() # Remove file from the system if the db commit failed os.unlink(os.path.join(upload_folder, filename)) LOG.debug( 'ERROR: cannot add candidate - user: "******" ' 'election: "%s"', flask.g.fas_user.username, election_id) LOG.exception(err) flask.flash( 'Someone has already upload a file with the same file name' ' for this election', 'error') return flask.render_template('contribute.html', election=election, form=form) flask.flash('Thanks for your submission') return flask.redirect(flask.url_for('index')) elif flask.request.method == 'GET': form.candidate_author.data = flask.g.fas_user.username return flask.render_template('contribute.html', election=election, form=form)
def contribute(election_id): ''' Display the index page for interested contributor. ''' election = nuancierlib.get_election(SESSION, election_id) if not election: flask.flash('No election found', 'error') return flask.render_template('msg.html') elif not election.submission_open: flask.flash('This election is not open for submission', 'error') return flask.redirect(flask.url_for('elections_list')) candidates = nuancier.lib.model.Candidates.get_by_submitter( SESSION, flask.g.fas_user.username, election_id) if election.user_n_candidates and \ len(candidates) >= election.user_n_candidates: flask.flash( 'You have uploaded the maximum number of candidates (%s) you ' 'can upload for this election' % election.user_n_candidates, 'error') return flask.redirect(flask.url_for('elections_list')) form = nuancier.forms.AddCandidateForm() if form.validate_on_submit(): candidate_file = flask.request.files['candidate_file'] try: validate_input_file(candidate_file) except nuancierlib.NuancierException as err: LOG.debug('ERROR: Uploaded file is invalid - user: "******" ' 'election: "%s"', flask.g.fas_user.username, election_id) LOG.exception(err) flask.flash(err.message, 'error') return flask.render_template( 'contribute.html', election=election, form=form) filename = secure_filename('%s-%s' % (flask.g.fas_user.username, candidate_file.filename)) # Only save the file once everything has been safely saved in the DB upload_folder = os.path.join( APP.config['PICTURE_FOLDER'], election.election_folder) if not os.path.exists(upload_folder): # pragma: no cover try: os.mkdir(upload_folder) except OSError, err: LOG.debug('ERROR: cannot add candidate file') LOG.exception(err) flask.flash( 'An error occured while writing the file, please ' 'contact an administrator', 'error') return flask.render_template( 'contribute.html', election=election, form=form) # Save candidate to the database try: nuancierlib.add_candidate( SESSION, candidate_file=filename, candidate_name=form.candidate_name.data, candidate_author=form.candidate_author.data, candidate_original_url=form.candidate_original_url.data, candidate_license=form.candidate_license.data, candidate_submitter=flask.g.fas_user.username, submitter_email=flask.g.fas_user.email, election_id=election.id, user=flask.g.fas_user.username, ) except nuancierlib.NuancierException as err: flask.flash(err.message, 'error') return flask.render_template( 'contribute.html', election=election, form=form) # The PIL module has already read the stream so we need to back up candidate_file.seek(0) candidate_file.save( os.path.join(upload_folder, filename)) try: SESSION.commit() except SQLAlchemyError as err: # pragma: no cover SESSION.rollback() # Remove file from the system if the db commit failed os.unlink(os.path.join(upload_folder, filename)) LOG.debug('ERROR: cannot add candidate - user: "******" ' 'election: "%s"', flask.g.fas_user.username, election_id) LOG.exception(err) flask.flash( 'Someone has already upload a file with the same file name' ' for this election', 'error') return flask.render_template( 'contribute.html', election=election, form=form) flask.flash('Thanks for your submission') return flask.redirect(flask.url_for('index'))