chats.py 7.1 KB
Newer Older
Anuraag Jain's avatar
Anuraag Jain committed
1
from fastapi import Depends, Request, HTTPException, status
Timothy J. Baek's avatar
Timothy J. Baek committed
2
3
from datetime import datetime, timedelta
from typing import List, Union, Optional
4
from utils.utils import get_current_user, get_admin_user
Timothy J. Baek's avatar
Timothy J. Baek committed
5
6
from fastapi import APIRouter
from pydantic import BaseModel
7
import json
Timothy J. Baek's avatar
Timothy J. Baek committed
8
9
10
11

from apps.web.models.users import Users
from apps.web.models.chats import (
    ChatModel,
12
    ChatResponse,
Timothy J. Baek's avatar
Timothy J. Baek committed
13
    ChatTitleForm,
Timothy J. Baek's avatar
Timothy J. Baek committed
14
15
16
17
18
    ChatForm,
    ChatTitleIdResponse,
    Chats,
)

19
20
21

from apps.web.models.tags import (
    TagModel,
22
    ChatIdTagModel,
23
24
25
26
27
    ChatIdTagForm,
    ChatTagsResponse,
    Tags,
)

Timothy J. Baek's avatar
Timothy J. Baek committed
28
29
30
31
32
33
34
35
36
37
from constants import ERROR_MESSAGES

router = APIRouter()

############################
# GetChats
############################


@router.get("/", response_model=List[ChatTitleIdResponse])
38
async def get_user_chats(
Timothy J. Baek's avatar
Timothy J. Baek committed
39
40
    user=Depends(get_current_user), skip: int = 0, limit: int = 50
):
41
    return Chats.get_chat_lists_by_user_id(user.id, skip, limit)
Timothy J. Baek's avatar
Timothy J. Baek committed
42
43


Timothy J. Baek's avatar
Timothy J. Baek committed
44
45
46
47
48
49
############################
# GetAllChats
############################


@router.get("/all", response_model=List[ChatResponse])
50
async def get_all_user_chats(user=Depends(get_current_user)):
Anuraag Jain's avatar
Anuraag Jain committed
51
    return [
Timothy J. Baek's avatar
Timothy J. Baek committed
52
53
        ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
        for chat in Chats.get_all_chats_by_user_id(user.id)
54
    ]
Timothy J. Baek's avatar
Timothy J. Baek committed
55
56


57
58
59
60
61
62
############################
# GetAllChatsInDB
############################


@router.get("/all/db", response_model=List[ChatResponse])
63
64
65
66
67
async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
    return [
        ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
        for chat in Chats.get_all_chats()
    ]
68
69


Timothy J. Baek's avatar
Timothy J. Baek committed
70
71
72
73
74
############################
# CreateNewChat
############################


75
@router.post("/new", response_model=Optional[ChatResponse])
76
async def create_new_chat(form_data: ChatForm, user=Depends(get_current_user)):
Timothy J. Baek's avatar
Timothy J. Baek committed
77
78
79
80
81
82
83
84
    try:
        chat = Chats.insert_new_chat(user.id, form_data)
        return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
    except Exception as e:
        print(e)
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT()
        )
Timothy J. Baek's avatar
Timothy J. Baek committed
85
86


Timothy J. Baek's avatar
Timothy J. Baek committed
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
############################
# GetAllTags
############################


@router.get("/tags/all", response_model=List[TagModel])
async def get_all_tags(user=Depends(get_current_user)):
    try:
        tags = Tags.get_tags_by_user_id(user.id)
        return tags
    except Exception as e:
        print(e)
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT()
        )


############################
# GetChatsByTags
############################


@router.get("/tags/tag/{tag_name}", response_model=List[ChatTitleIdResponse])
async def get_user_chats_by_tag_name(
    tag_name: str, user=Depends(get_current_user), skip: int = 0, limit: int = 50
):
    chat_ids = [
        chat_id_tag.chat_id
        for chat_id_tag in Tags.get_chat_ids_by_tag_name_and_user_id(tag_name, user.id)
    ]

    print(chat_ids)

    return Chats.get_chat_lists_by_chat_ids(chat_ids, skip, limit)


Timothy J. Baek's avatar
Timothy J. Baek committed
123
124
125
126
127
############################
# GetChatById
############################


128
@router.get("/{id}", response_model=Optional[ChatResponse])
129
130
async def get_chat_by_id(id: str, user=Depends(get_current_user)):
    chat = Chats.get_chat_by_id_and_user_id(id, user.id)
Timothy J. Baek's avatar
Timothy J. Baek committed
131

Anuraag Jain's avatar
Anuraag Jain committed
132
    if chat:
Timothy J. Baek's avatar
Timothy J. Baek committed
133
        return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
Timothy J. Baek's avatar
Timothy J. Baek committed
134
    else:
Timothy J. Baek's avatar
Timothy J. Baek committed
135
136
137
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
        )
Timothy J. Baek's avatar
Timothy J. Baek committed
138
139
140
141
142
143
144


############################
# UpdateChatById
############################


145
@router.post("/{id}", response_model=Optional[ChatResponse])
Timothy J. Baek's avatar
Timothy J. Baek committed
146
147
148
async def update_chat_by_id(
    id: str, form_data: ChatForm, user=Depends(get_current_user)
):
149
    chat = Chats.get_chat_by_id_and_user_id(id, user.id)
Anuraag Jain's avatar
Anuraag Jain committed
150
    if chat:
151
        updated_chat = {**json.loads(chat.chat), **form_data.chat}
Timothy J. Baek's avatar
Timothy J. Baek committed
152

153
        chat = Chats.update_chat_by_id(id, updated_chat)
Timothy J. Baek's avatar
Timothy J. Baek committed
154
        return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
Anuraag Jain's avatar
Anuraag Jain committed
155
    else:
156
157
158
159
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
        )
160
161
162
163
164
165
166
167


############################
# DeleteChatById
############################


@router.delete("/{id}", response_model=bool)
Timothy J. Baek's avatar
Timothy J. Baek committed
168
169
170
171
172
173
174
175
176
177
178
async def delete_chat_by_id(request: Request, id: str, user=Depends(get_current_user)):

    if (
        user.role == "user"
        and not request.app.state.USER_PERMISSIONS["chat"]["deletion"]
    ):
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
        )

179
180
    result = Chats.delete_chat_by_id_and_user_id(id, user.id)
    return result
Timothy J. Baek's avatar
Timothy J. Baek committed
181

182

183
184
185
186
187
188
189
190
191
############################
# GetChatTagsById
############################


@router.get("/{id}/tags", response_model=List[TagModel])
async def get_chat_tags_by_id(id: str, user=Depends(get_current_user)):
    tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)

192
    if tags != None:
193
194
195
196
197
198
199
200
201
202
203
204
        return tags
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
        )


############################
# AddChatTagById
############################


205
@router.post("/{id}/tags", response_model=Optional[ChatIdTagModel])
206
207
208
async def add_chat_tag_by_id(
    id: str, form_data: ChatIdTagForm, user=Depends(get_current_user)
):
209
210
211
212
    tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)

    if form_data.tag_name not in tags:
        tag = Tags.add_tag_to_chat(user.id, form_data)
213

214
215
216
217
218
219
220
        if tag:
            return tag
        else:
            raise HTTPException(
                status_code=status.HTTP_401_UNAUTHORIZED,
                detail=ERROR_MESSAGES.NOT_FOUND,
            )
221
222
    else:
        raise HTTPException(
223
            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT()
224
225
226
227
228
229
230
231
232
        )


############################
# DeleteChatTagById
############################


@router.delete("/{id}/tags", response_model=Optional[bool])
233
async def delete_chat_tag_by_id(
234
235
    id: str, form_data: ChatIdTagForm, user=Depends(get_current_user)
):
236
    result = Tags.delete_tag_by_tag_name_and_chat_id_and_user_id(
237
238
239
        form_data.tag_name, id, user.id
    )

240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
    if result:
        return result
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
        )


############################
# DeleteAllChatTagsById
############################


@router.delete("/{id}/tags/all", response_model=Optional[bool])
async def delete_all_chat_tags_by_id(id: str, user=Depends(get_current_user)):
    result = Tags.delete_tags_by_chat_id_and_user_id(id, user.id)

    if result:
        return result
259
260
261
262
263
264
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
        )


Timothy J. Baek's avatar
Timothy J. Baek committed
265
266
267
268
269
270
############################
# DeleteAllChats
############################


@router.delete("/", response_model=bool)
271
272
273
async def delete_all_user_chats(user=Depends(get_current_user)):
    result = Chats.delete_chats_by_user_id(user.id)
    return result