def host_category_delete(host_id, hc_id): """ Delete a host_category. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') else: for url in hcobj.urls: SESSION.delete(url) for dirs in hcobj.dirs: SESSION.delete(dirs) SESSION.delete(hcobj) try: SESSION.commit() flask.flash('Host Category deleted') except SQLAlchemyError as err: SESSION.rollback() flask.flash('Could not delete Category of the host') APP.logger.debug('Could not delete Category of the host') APP.logger.exception(err) return flask.redirect(flask.url_for('host_view', host_id=host_id))
def host_category_url_delete(host_id, hc_id, host_category_url_id): """ Delete a host_category_url. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') hostcaturlobj = mmlib.get_host_category_url(SESSION, host_category_url_id) if hostcaturlobj is None: flask.abort(404, 'Host category URL not found') else: SESSION.delete(hostcaturlobj) try: SESSION.commit() flask.flash('Host category URL deleted') except SQLAlchemyError as err: SESSION.rollback() flask.flash('Could not delete category URL of the host') APP.logger.debug('Could not delete category URL of the host') APP.logger.exception(err) return flask.redirect( flask.url_for('host_category', host_id=hostobj.id, hc_id=hcobj.id))
def host_category_url_new(host_id, hc_id): """ Create a new host_category_url. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') if not (is_site_admin(flask.g.fas_user, hostobj.site) or is_mirrormanager_admin(flask.g.fas_user)): flask.abort(403, 'Access denied') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') host_cat_ids = [cat.id for cat in hostobj.categories] if hcobj.id not in host_cat_ids: flask.abort(404, 'Category not associated with this host') categories = mmlib.get_categories(SESSION) form = forms.AddHostCategoryUrlForm() if form.validate_on_submit(): host_category_u = model.HostCategoryUrl() host_category_u.host_category_id = hcobj.id form.populate_obj(obj=host_category_u) url = form.url.data.strip() if url.endswith('/'): url = url[:-1] host_category_u.url = url SESSION.add(host_category_u) try: SESSION.flush() flask.flash('Host Category URL added') except SQLAlchemyError as err: SESSION.rollback() flask.flash('Could not add Category URL to the host') APP.logger.debug('Could not add Category URL to the host') APP.logger.exception(err) SESSION.commit() return flask.redirect( flask.url_for('host_category', host_id=host_id, hc_id=hc_id)) return flask.render_template( 'host_category_url_new.html', form=form, host=hostobj, hostcategory=hcobj, )
def host_category(host_id, hc_id): """ View a host_category. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') if not (is_site_admin(flask.g.fas_user, hostobj.site) or is_mirrormanager_admin(flask.g.fas_user)): flask.abort(403, 'Access denied') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') host_cat_ids = [cat.id for cat in hostobj.categories] if hcobj.id not in host_cat_ids: flask.abort(404, 'Category not associated with this host') categories = mmlib.get_categories(SESSION) form = forms.EditHostCategoryForm(obj=hcobj) if form.validate_on_submit() and is_mirrormanager_admin(flask.g.fas_user): form.populate_obj(obj=hcobj) try: SESSION.flush() flask.flash('Host Category updated') except SQLAlchemyError as err: # pragma: no cover # We cannot check this because the code check before updating # and therefore the only situation where it could fail is a # failure at the DB server level itself. SESSION.rollback() flask.flash('Could not update Category to the host') APP.logger.debug('Could not update Category to the host') APP.logger.exception(err) SESSION.commit() return flask.redirect( flask.url_for('host_category', host_id=hostobj.id, hc_id=hcobj.id)) return flask.render_template( 'host_category.html', form=form, host=hostobj, hostcategory=hcobj, )
def host_category_url_delete(host_id, hc_id, host_category_url_id): """ Delete a host_category_url. """ form = forms.ConfirmationForm() if form.validate_on_submit(): hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') if not (is_site_admin(flask.g.fas_user, hostobj.site) or is_mirrormanager_admin(flask.g.fas_user)): flask.abort(403, 'Access denied') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') host_cat_ids = [cat.id for cat in hostobj.categories] if hcobj.id not in host_cat_ids: flask.abort(404, 'Category not associated with this host') hostcaturlobj = mmlib.get_host_category_url_by_id( SESSION, host_category_url_id) if hostcaturlobj is None: flask.abort(404, 'Host category URL not found') host_cat_url_ids = [url.id for url in hcobj.urls] if hostcaturlobj.id not in host_cat_url_ids: flask.abort(404, 'Category URL not associated with this host') else: SESSION.delete(hostcaturlobj) try: SESSION.commit() flask.flash('Host category URL deleted') except SQLAlchemyError as err: # pragma: no cover # We check everything before deleting so the only error we could # run in is DB server related, and that we can't fake in our # tests SESSION.rollback() flask.flash('Could not delete category URL of the host') APP.logger.debug('Could not delete category URL of the host') APP.logger.exception(err) return flask.redirect( flask.url_for('host_category', host_id=host_id, hc_id=hc_id))
def host_category_url_new(host_id, hc_id): """ Create a new host_category_url. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') categories = mmlib.get_categories(SESSION) form = forms.AddHostCategoryUrlForm() if form.validate_on_submit(): host_category_u = model.HostCategoryUrl() host_category_u.host_category_id = hcobj.id form.populate_obj(obj=host_category_u) SESSION.add(host_category_u) try: SESSION.flush() flask.flash('Host Category URL added') except SQLAlchemyError as err: SESSION.rollback() flask.flash('Could not add Category URL to the host') APP.logger.debug('Could not add Category URL to the host') APP.logger.exception(err) SESSION.commit() return flask.redirect( flask.url_for('host_category', host_id=host_id, hc_id=hc_id)) return flask.render_template( 'host_category_url_new.html', form=form, host=hostobj, hostcategory=hcobj, )
def host_category(host_id, hc_id): """ View a host_category. """ hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') categories = mmlib.get_categories(SESSION) form = forms.EditHostCategoryForm(obj=hcobj) if form.validate_on_submit(): form.populate_obj(obj=hcobj) try: SESSION.flush() flask.flash('Host Category updated') except SQLAlchemyError as err: SESSION.rollback() flask.flash('Could not update Category to the host') APP.logger.debug('Could not update Category to the host') APP.logger.exception(err) SESSION.commit() return flask.redirect( flask.url_for('host_category', host_id=hostobj.id, hc_id=hcobj.id)) return flask.render_template( 'host_category.html', form=form, host=hostobj, hostcategory=hcobj, )
def host_category_delete(host_id, hc_id): """ Delete a host_category. """ form = forms.ConfirmationForm() if form.validate_on_submit(): hostobj = mmlib.get_host(SESSION, host_id) if hostobj is None: flask.abort(404, 'Host not found') hcobj = mmlib.get_host_category(SESSION, hc_id) if hcobj is None: flask.abort(404, 'Host/Category not found') host_cat_ids = [cat.id for cat in hostobj.categories] if hcobj.id not in host_cat_ids: flask.abort(404, 'Category not associated with this host') else: for url in hcobj.urls: SESSION.delete(url) for dirs in hcobj.directories: SESSION.delete(dirs) SESSION.delete(hcobj) try: SESSION.commit() flask.flash('Host Category deleted') except SQLAlchemyError as err: # pragma: no cover # We check everything before deleting so the only error we could # run in is DB server related, and that we can't fake in our # tests SESSION.rollback() flask.flash('Could not delete Category of the host') APP.logger.debug('Could not delete Category of the host') APP.logger.exception(err) return flask.redirect(flask.url_for('host_view', host_id=host_id))