-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
75 lines (60 loc) · 2.04 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from fastapi import FastAPI, HTTPException, status, Query, Response
from pydantic import BaseModel
from typing import Optional
from random import randrange
app = FastAPI()
class Post(BaseModel):
title: str
content: str
published: bool = True
ratings: Optional[int] = None
my_list = [
{"title": "title of post 1", "content": "content of post 1", "id": 1},
{"title": "title of post 2", "content": "content of post 2", "id": 2}
]
@app.get("/posts")
def get_all_posts():
return {"data": my_list}
@app.post("/posts", status_code=status.HTTP_201_CREATED)
def create_post(post: Post):
post_dict = post.dict()
post_dict["id"] = randrange(0, 1000000)
my_list.append(post_dict)
return {"data": post_dict}
@app.get("/posts/latest")
def get_latest_post():
post = my_list[-1]
return {"post_detail": post}
@app.get("/posts/{id}")
def get_post_by_id(id: int):
post = find_post(id)
if not post:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
detail=f"Post with ID {id} not found")
return {"post_detail": post}
@app.delete("/posts/{id}", status_code=status.HTTP_204_NO_CONTENT)
def delete_post(id: int):
indx = find_index_post(id)
if indx is None:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
detail=f"Post with ID {id} does not exist")
my_list.pop(indx)
return {"message": f"Post with ID {id} successfully deleted"}
@app.put("/posts/{id}")
def update_post(id: int, post: Post):
indx = find_index_post(id)
if indx is None:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
detail=f"Post with ID {id} does not exist")
post_dict = post.dict()
post_dict['id'] = id
my_list[indx] = post_dict
return {"message": f"Post with ID {id} successfully updated"}
def find_post(id):
for p in my_list:
if p["id"] == id:
return p
def find_index_post(id):
for i, p in enumerate(my_list):
if p['id'] == id:
return i