from flask import render_template, abort, redirect, url_for, abort, flash, request, current_app, make_response, g from flask_login import login_required, login_user, logout_user, current_user from flask_sqlalchemy import get_debug_queries from . import panel from .forms import OrderForm, MessageForm from .. import db from ..email import send_email from ..models import User, Permission, Recipe, Order, Server, Deployment, Service, Region, Address, Domain, SupportTopic, SupportLine, contact_proxmaster import base64 from datetime import date, time, datetime from dateutil.relativedelta import relativedelta @panel.after_app_request def after_request(response): for query in get_debug_queries(): if query.duration >= current_app.config['SLOW_DB_QUERY_TIME']: current_app.logger.warning('Slow query: %s\nParameters: %s\nDuration: %fs\nContext: %s\n' % (query.statement, query.parameters, query.duration, query.context)) return response @panel.route("/deploy", methods=['GET', 'POST']) @login_required def deploy(): if current_user.name is None: return redirect(url_for('settings.profile')) form = OrderForm() if form.validate_on_submit(): region = Region.query.filter_by(pid=int(form.region.data)).first() recipe = Recipe.query.filter_by(pid=int(form.recipe.data)).first() new_order = Order(user_id=int(current_user.pid), region_id=int(region.pid), recipe_id=int(recipe.pid), parameter1=str(form.alias.data), parameter2=str(form.cpu.data), parameter3=str(form.memory.data), parameter4=str(form.storage.data), status='new') db.session.add(new_order) db.session.commit() send_email(current_app.config['MAIL_USERNAME'], 'New order from {}'.format(current_user.email), 'panel/email/adm_neworder', user=current_user) return redirect(request.args.get('next') or url_for('panel.dashboard')) return render_template('panel/deploy.html', form=form) #DASHBOARD @panel.route("/dashboard", defaults={'user_pid': 0}, methods=['GET']) @panel.route("/dashboard/", methods=['GET']) @login_required def dashboard(user_pid): sys_regions = Region.query.all() if user_pid == 0: cuser = current_user else: cuser = User.query.filter_by(pid=user_pid).first() if cuser == None: abort(404) if not current_user.is_administrator(): abort(404) #hidden 403 inv_addresses = cuser.inv_addresses.order_by(Address.ip.asc()).all() inv_deployments = cuser.inv_deployments.filter_by(deleted=False).order_by(Deployment.machine_alias.asc()).all() regions = {} for region in sys_regions: regions[region.pid] = region.description inv_deploycubeids = [] warnflag = False for invcls in inv_deployments: if invcls.user_id == cuser.pid: inv_deploycubeids.extend([invcls.machine_id]) #warning detector if invcls.warning == True or invcls.enabled == False: warnflag = True inv_services = cuser.inv_services.filter_by(deleted=False).order_by(Service.date_last_charge.asc()).all() inv_domains = cuser.inv_domains.filter_by(deleted=False).order_by(Domain.date_created.desc()).all() #extract rrd and status from the deployments rrd = {} statuses = {} #current_app.logger.warning(str(inv_deploycubeids)) for unit_id in inv_deploycubeids: data = { 'unit_id': int(unit_id), 'type': 'kvm' } try: query = contact_proxmaster(data, 'vmrrd') graphs_list = ['net', 'cpu', 'mem', 'hdd'] rrd[unit_id] = {} for graph in graphs_list: raw = query[graph]['image'].encode('raw_unicode_escape') rrd[unit_id][graph] = base64.b64encode(raw).decode() status = { unit_id : query['status'] } statuses.update(status) except Exception as e: current_app.logger.error(e) for invcls in inv_deployments: if invcls.machine_id == unit_id: inv_deployments.remove(invcls) flash('Support is notified about {}.'.format(str(cuser.inv_deployments.filter_by(machine_id=unit_id).first().machine_alias))) if not current_user.is_administrator(): send_email(current_app.config['MAIL_USERNAME'], '{} experienced an error'.format(cuser.email), 'vmanager/email/adm_unreachable', user=current_user, unit_id=unit_id, error=repr(e)) continue supportform = MessageForm() return render_template('panel/dashboard.html', sys_regions=sys_regions, inv_deployments=inv_deployments, inv_services=inv_services, inv_domains=inv_domains, inv_addresses=inv_addresses, rrd=rrd, status=statuses, warnflag=warnflag, regions=regions, form=supportform) #SUPPORT @panel.route("/list", methods=['GET']) @login_required def support_list(): """ general enquiry and list all open support tasks """ cuser = current_user form = MessageForm() alltopics = cuser.inv_topics.all() return render_template('panel/support_list.html', form=form, inv_topics=alltopics) @panel.route("/topic//", methods=['GET', 'POST']) @login_required def support(topic): """ block item for support chatbox. invoked from vdc_pool or supportlist """ cuser = current_user form = MessageForm() if request.method == "GET": support_topic = SupportTopic.query.filter_by(hashtag=str(topic)).first() if support_topic == None: class EmptySupport(): hashtag=str(topic) timestamp=datetime.utcnow() support_topic = EmptySupport() return render_template('panel/support_item.html', form=form, support=support_topic) else: if support_topic.user_id != cuser.pid: abort(403) #TODO: hidden 403. there is a topic like that but its not yours! else: #topic is yours. show it. return render_template('panel/support_item.html', form=form, support=support_topic) if request.method == "POST" and form.validate_on_submit(): support_topic = SupportTopic.query.filter_by(hashtag=str(topic)).first() if support_topic == None: #no topic. create one? if cuser.inv_topics.all() != []: #check if other topics exist, and ratelimit last_topic = cuser.inv_topics.order_by(SupportTopic.timestamp.desc()).first() now = datetime.utcnow() time_last_topic = last_topic.timestamp expiry = time_last_topic + relativedelta(time_last_topic, minutes=+5) if now < expiry: flash('ratelimit. try again later') return redirect(url_for('panel.support_list')) #create new topic new_topic = SupportTopic(user_id=cuser.pid, hashtag=str(topic)) db.session.add(new_topic) new_line = SupportLine(topic_id=new_topic.pid, line=str(form.line.data)) db.session.add(new_line) else: if support_topic.user_id == cuser.pid: new_line = SupportLine(topic_id=support_topic.pid, line=form.line.data) db.session.add(new_line) else: abort(403) #TODO: hidden 404 db.session.commit() return redirect(url_for('panel.support_list'))