sqllineage/sqllineage.py

from flask import Flask, render_template, request, jsonify, send_file, send_from_directory
from utils.sql_parse import parse_create_table_sql
from utils.log import Log

app = Flask(__name__)
log = Log().getlog()


@app.route('/css/<path:filename>')
def css_files(filename):
    return send_from_directory('templates/css', filename)


@app.route('/')
def index():
    return send_file('templates/index.html')


@app.route('/a')
def index_a():
    return render_template('a.html')


@app.route('/b')
def index_b():
    return render_template('b.html')


@app.route('/convert', methods=['POST'])
def convert_sql():
    sql_input = request.form['sql']
    if sql_input.strip() == '':
        return jsonify({
            'parsed_result': "请在上面输入框输入hologres建表语句",
            'message': 'SQL processed.'
        })

    log.info("SQL Input: %s", sql_input)
    try:
        parsed_result = parse_create_table_sql(sql_input, '')
        print(parsed_result)
        result = {
            'parsed_result': parsed_result,
            'message': 'SQL processed successfully.'
        }
    except Exception as e:
        result = {'error': f'请检查输入sql:\n\n  {str(e)}'}
    log.info("SQL result: %s", result)
    return jsonify(result)


@app.route('/convert1', methods=['POST'])
def convert_sql1():
    createTableDDL = request.form['sql']
    if createTableDDL.strip() == '':
        return jsonify({
            'parsed_result': "请在上面输入框输入hologres建表语句",
            'message': 'SQL processed.'
        })

    log.info("SQL Input: %s", createTableDDL)
    try:

        parsed_result = ''
        createTableDDL = createTableDDL.split('CREATE')[1].split('with (')[0]
        createTableDDL = 'CREATE' + createTableDDL
        print(createTableDDL)
        createTableDDL = createTableDDL.replace('public.', '').replace('integer', 'int')
        createTableDDL = createTableDDL.replace('timestamp with time zone', 'DATETIME').replace('text', 'VARCHAR(64)')
        createTableDDL = createTableDDL.replace(',PRIMARY KEY', 'UNIQUE KEY')
        createTableDDL = createTableDDL.replace('numeric(10,2)', 'DECIMAL(10,2)')
        createTableDDL = createTableDDL.replace('numeric(10,4)', 'DECIMAL(10,4)')

        createTableDDL = createTableDDL.split('\n')
        createTableDDL = [(i + '\n').strip() for i in createTableDDL]
        UNIQUE_str = ''
        UNIQUE_key = ''
        for i in createTableDDL:
            if i.startswith('UNIQUE'):
                # 获取字符()里面的字符串
                UNIQUE_key = i.replace('UNIQUE KEY (', '').replace(")", "")
                UNIQUE_str += i
            else:
                if i:
                    parsed_result += i + '\n'

        UNIQUE_str += f' DISTRIBUTED BY HASH ({UNIQUE_key}) BUCKETS 32;'
        parsed_result += UNIQUE_str
        result = {
            'parsed_result': parsed_result,
            'message': 'SQL processed successfully.'
        }
    except Exception as e:
        result = {'error': f'请检查输入sql:\n\n  {str(e)}'}
    log.info("SQL result: %s", result)
    print(result)
    return jsonify(result)


if __name__ == '__main__':
    # 指定host和port，这里使用0.0.0.0可以让服务器被外部访问
    app.run(host='0.0.0.0', port=8778, debug=True)