# coding: utf-8
import json

import pika
# from selenium.webdriver.common.by import By

from house_info import HouseInfo

# CONFIG AREA
AMQP_CONFIG = {
    'host': '127.0.0.1',
    'port': 5672,
    'username': 'guest',
    'password': 'guest'
}
AMQP_QUEUE_NAME = 'house-inquery'
AMQP_FEEDBACK_QUEUE_NAME = 'house-inquery-result'


# 因为反馈回去的队列要求在不同队列,所以重新建立了一个新的队列连接
def process_feedback(msg):
    mq_credential = pika.PlainCredentials(
        username=AMQP_CONFIG['username'],
        password=AMQP_CONFIG['password']
    )
    mq_conn = pika.BlockingConnection(pika.ConnectionParameters(
        host=AMQP_CONFIG['host'],
        credentials=mq_credential
    ))
    mq_channel = mq_conn.channel()
    mq_channel.queue_declare(queue=AMQP_FEEDBACK_QUEUE_NAME, durable=True)
    return mq_channel.basic_publish(exchange='',
                                    routing_key=AMQP_FEEDBACK_QUEUE_NAME,
                                    body=msg,
                                    properties=pika.BasicProperties(
                                        delivery_mode=2,
                                        content_encoding='UTF-8',
                                        content_type='text/plain'
                                    )
                                    )


user_pwd = pika.PlainCredentials(AMQP_CONFIG['username'], AMQP_CONFIG['password'])
s_conn = pika.BlockingConnection(pika.ConnectionParameters(AMQP_CONFIG['host'], credentials=user_pwd))
channel = s_conn.channel()
channel.queue_declare(queue=AMQP_QUEUE_NAME, durable=True)


def callback(ch, method, properties, body):
    print(body)
    query_data = json.loads(body)
    if not query_data:
        return
    try:
        # 一些业务逻辑代码
        # 应答消息
        ch.basic_ack(delivery_tag=method.delivery_tag)
    except Exception as e:
        print(e)

# 设置一次只获取一条消息处理
channel.basic_qos(prefetch_count=1)
# 启用ack,防止异常消息丢失 确认处理后再删除消息
channel.basic_consume(callback, queue=AMQP_QUEUE_NAME, no_ack=False)
channel.start_consuming()

# coding: utf-8
from flask import Flask
from flask import render_template
from flask import request
import docker

app = Flask(__name__, template_folder='./templates', static_folder="", static_url_path="")
client = docker.from_env()


@app.route('/cmd', methods=['GET', 'POST'])
def exec_cmd():
    if request.method == 'GET':
        return render_template('cmd.html')
    if request.method == 'POST':
        container = client.containers.get('gmb-index') # 这里我写死了获取本地这个容器
        cmd = request.form.get('cmd')
        result = container.exec_run(cmd=cmd)
        output = result.output.decode()
        return render_template('cmd.html', output=output)


if __name__=='__main__':
    app.run(host='127.0.0.1', port=8888)