pjcunningham / flask-admin-modal Goto Github PK

@expose('/', methods=['POST'])
def index(self):
    if request.method == 'POST':
        # get the original checked ids
        url = get_redirect_target() or self.get_url('.index_view')
        ids = request.form.getlist('rowid')
        joined_ids = ','.join(ids)
        # could not encode as a `str` so it must be encoded
        encoded_ids = base64.b64encode(joined_ids.encode('utf-8'))
        change_form = ReconcileForm()
        # now the `byte` object needs to be encoded into a `str` for the form
        change_form.ids.data = str(encoded_ids, 'utf-8')
        self._template_args['url'] = url
        self._template_args['change_form'] = change_form
        self._template_args['change_modal'] = True
        return self.index_view()

@expose('/update/', methods=['POST'])
def update_view(self):
    if request.method == 'POST':
        url = get_redirect_target() or self.get_url('.index_view')
        change_form = ReconcileForm(request.form)
        if change_form.validate():
            decoded_ids = base64.b64decode(change_form.ids.data)
            # again `byte` object needs to be made into a `str`
            ids = str(decoded_ids, 'utf-8').split(',')
            cost = change_form.cost.data
            _update_mappings = [{'id': rowid, 'cost': cost} for rowid in ids]
            db.session.bulk_update_mappings(Project, _update_mappings)
            db.session.commit()
            flash("Set cost for {} record{} to {}."
                  .format(len(ids), 's' if len(ids) > 1 else '', cost),
                  category='info')
            return redirect(url)
        else:
            # Form failed validation
            self._template_args['url'] = url
            self._template_args['change_form'] = change_form
            self._template_args['change_modal'] = True
            return self.index_view()