import os
import sys
import json
import sqlite3

OUTPUT_FILE = sys.path[0] + "/../data/table_info.json"
LOCAL_TABLE_INFO_FILE = sys.path[0] + "/table_info.list"

def read_raw_file(file_path):
    f = open(file_path)
    content = f.read()
    f.close()
    return content

def read_line_file(file_path):
    return read_raw_file(file_path).split("\n")


def write_json_file(file_path, obj):
    f = open(file_path, "w")
    content = f.write(json.dumps(obj, indent=1))
    f.close()
    return file_path


lines = read_line_file(LOCAL_TABLE_INFO_FILE)

table_dict = {}

for line in lines:
    kv_list = line.rstrip().split("\t")
    if len(kv_list) != 2:
        continue
    if kv_list[0] not in table_dict:
        table_dict[kv_list[0]] = []
        pass
    table_dict[kv_list[0]].append(kv_list[1])
    pass


total_table_list = []
for k,v in table_dict.items():
    item = {"database": "sample_data", "sql": "", "table": k, "col_names": v}
    total_table_list.append(item)
    pass

write_json_file(OUTPUT_FILE, total_table_list)