from service.cus_vanna_srevice import CustomVanna
from util import train_ddl
from  util import q_and_a_dict,q_and_a_test1
table_ddls = [
    train_ddl.person_database_ddl,train_ddl.person_status_ddl,
    train_ddl.person_attendance_ddl,train_ddl.person_ac_area,
    train_ddl.org_orgs_ddl

]
list_documentions = [
    train_ddl.train_document,
]


def add_ddl(vn: CustomVanna):
    for ddl in table_ddls:
        vn.add_ddl(ddl)


def add_documentation(vn: CustomVanna):
    for doc in list_documentions:
        vn.add_documentation(doc)

    for d in q_and_a_test1.question_and_answer:
        vn.add_question_sql(
            question=f"{d['question']} [分类:{d['category']}] [标签:{','.join(d['tags'])}]",
            sql=d['answer']
        )
        # vn.add_question_sql(question=d['question'], sql=d['answer'])