from db.crud import ingest, read, update, delete
from db.schema import Feedback, Response, ModelRatings
from datetime import datetime


def test():
    # Sample data for ModelRatings
    model_ratings = ModelRatings(
        query_v_ratings={'v1': 5, 'v2': 4},
        query_p0_ratings={'p0_1': 3, 'p0_2': 4},
        query_p1_ratings={'p1_1': 2, 'p1_2': 5}
    )

    # Sample data for Response
    response = Response(
        config_id='config_123',
        model_ratings={'model_1': model_ratings},
        comment='This is a sample response.',
        timestamp='2025-02-01T18:13:48'
    )

    # Sample data for Feedback
    feedback = Feedback(
        id=1,
        user_id='user_123',
        time_stamp=datetime(2025, 2, 1, 18, 13, 48),
        responses=[response]
    )

    ingest(feedback)
    print(read("user_123"))

    # Read (Fetch)
    feedback_data = read("ashmi")
    if feedback_data:
        print(feedback_data)

    # Update (Modify)
    # updated_feedback = Feedback(
    #     id=1,
    #     user_id="user123",
    #     time_stamp=datetime.now(),
    #     responses=[
    #         {"q_id": "q1", "ans": 4},  # Updated answer
    #         {"q_id": "q2", "ans": 3}
    #     ]
    # )
    # update(1, updated_feedback)
    #
    # # Delete (Remove)
    # delete(1)


if __name__ == "__main__":
    test()