chat_table_test.py 1.05 KB
Newer Older
1
from src.pgdb.chat.c_db import UPostgresDB
2 3 4 5 6 7 8 9
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
)
10

11

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


陈正乐 committed
15
def test():
16 17
    c_db = UPostgresDB(host=CHAT_DB_HOST, database=CHAT_DB_DBNAME, user=CHAT_DB_USER, password=CHAT_DB_PASSWORD,
                       port=CHAT_DB_PORT, )
18
    print(c_db)
19
    crud = CRUD(c_db)
20 21 22 23 24 25 26 27 28 29 30
    # 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')
31
    print(crud.get_history('2'))
32 33


陈正乐 committed
34
if __name__ == "__main__":
陈正乐 committed
35
    test()