qa_bot/back_end/bot_app/fulfillment/utility.py

from flask import jsonify
from sqlalchemy import create_engine
db_engine = None

def simple_response(text_content='', fulfillmentObj=None):
    if not fulfillmentObj==None:
        jsonResp = fulfillmentObj
    else:
        jsonResp = {
            'fulfillmentText': text_content,
            'fulfillmentMessages': [
                # {
                #     'image' : {
                #     'imageUri':'https://yt3.ggpht.com/a/AGF-l78sCrWnJHZlRs-DP1imkaINg2KkpT5Gomkahw=s900-mo-c-c0xffffffff-rj-k-no',
                #     'accessibilityText':'Hello World!!'
                #     }
                # }

                # {
                #     'card': {
                #         'title': 'card title',
                #         'subtitle': 'card text',
                #         'imageUri': 'https://assistant.google.com/static/images/molecule/Molecule-Formation-stop.png',
                #         'buttons': [
                #             {
                #                 'text': 'button text',
                #                 'postback': 'https://assistant.google.com/'
                #             }
                #         ]
                #     }
                # }
            ],
            'source': 'richard_shih@wanhai.com'
        }

    return jsonify(jsonResp)

def lookup_context(fulfillment,lookup_pattern):
    contexts = fulfillment.get('queryResult').get('outputContexts')
    search_key = fulfillment.get('session') + '/contexts/' + lookup_pattern
    return next((x for x in contexts if x['name']== search_key),None)

def get_db_conn():
    try:
        conn = db_engine.connect()
        return conn
    except Exception as e:
        print(str(e))

def init_app(app, pre_connect = True):
    global db_engine
    db_engine = create_engine(app.config['SQLALCHEMY_DATABASE_URI'], encoding='utf8') 
    pre_connect = app.config['DATABASE_CONNECT_OPTIONS']['PRE_CONNECT']
    if pre_connect:
        try:
            conn = db_engine.connect()
        except Exception as e:    
            print(str(e))