from src.pgdb.chat.c_db import UPostgresDB
from src.pgdb.chat.crud import CRUD
from src.config.consts import (
    CHAT_DB_USER,
    CHAT_DB_HOST,
    CHAT_DB_PORT,
    CHAT_DB_DBNAME,
    CHAT_DB_PASSWORD
)


"""测试会话相关数据可的连接"""


def test():
    c_db = UPostgresDB(host=CHAT_DB_HOST, database=CHAT_DB_DBNAME, user=CHAT_DB_USER, password=CHAT_DB_PASSWORD,
                       port=CHAT_DB_PORT, )
    print(c_db)
    crud = CRUD(c_db)
    crud.create_table()
    # crud.insert_turn_qa("2", "wen4", "da1", 1, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 2, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 5, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 4, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 3, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 6, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 8, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 7, 0)
    # crud.insert_turn_qa("2", "wen4", "da1", 9, 0)
    crud.insert_c_user('zhangs', '111111')
    crud.insert_chat('1', '这是chat_id为1的问答info', 0)
    crud.insert_c_user('lis', '111111')
    crud.insert_chat('1', '这是chat_id为2的问答info', 0)
    crud.insert_chat('2', '这是chat_id为3的问答info', 0)
    crud.insert_chat('2', '这是chat_id为4的问答info', 0)
    print(crud.get_history('2'))


if __name__ == "__main__":
    test()