from service.cus_vanna_srevice import CustomVanna
from util import train_ddl
from  util import q_and_a_dict
table_ddls = [
    train_ddl.person_ddl_sql,train_ddl.user_status_ddl,
    train_ddl.user_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_dict.question_and_answer:
        vn.add_question_sql(question=d['question'], sql=d['answer'])