Skip to content

Commit

Permalink
Fix for camera models; potential fix for CORS
Browse files Browse the repository at this point in the history
  • Loading branch information
ReallyWeirdCat committed Dec 7, 2024
1 parent c6fe5a7 commit a485f47
Show file tree
Hide file tree
Showing 4 changed files with 139 additions and 7 deletions.
4 changes: 3 additions & 1 deletion backend/app/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,9 @@ def main():
CORSMiddleware,
allow_origins=[
"http://localhost",
"http://0.0.0.0"
"http://0.0.0.0",
"http://127.0.0.1",
"*"
],
allow_credentials=True,
allow_methods=["*"],
Expand Down
117 changes: 115 additions & 2 deletions backend/app/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,121 @@ def create_database():
safe=True
)

Role.get_or_create(name="user")
Role.get_or_create(name="admin")
Role(name="user")
Role(name="admin")

if Camera.select().count() != 0: # pylint: disable=no-value-for-parameter
return

Camera(
id=1,
name="Dummy 1",
description="Located in room 12",
contamination=0.5,
date="2024-01-01",
url="https://t4.ftcdn.net/jpg/03/10/07/45/360_F_310074598_rBt50O0nwjydPjWStjdzyNdm0Oh1nAyV.jpg"
).save()

Camera(
id=2,
name="Dummy 2",
description="Located in room 15",
contamination=0.3,
date="2024-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=3,
name="Dummy 3",
description="Located in room 17",
contamination=0.13,
date="2022-01-01",
url="https://media.istockphoto.com/id/1141324902/photo/real-lens-flare-shot-in-studio-over-black-background-easy-to-add-as-overlay-or-screen-filter.jpg?s=612x612&w=0&k=20&c=zWGnDHkDJZKqaUdNIGkf_eSNJ17qRrnl7czqJxWZlLw="
).save()

Camera(
id=4,
name="Dummy 4",
description="Located in room 19",
contamination=0.1,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=5,
name="Dummy 5",
description="Located in room 21",
contamination=0.05,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=6,
name="Dummy 6",
description="Located in room 23",
contamination=0.01,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=7,
name="Dummy 7",
description="Located in room 25",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=8,
name="Dummy 8",
description="Located in room 27",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=9,
name="Dummy 9",
description="Located in room 29",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=10,
name="Dummy 10",
description="Located in room 31",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=11,
name="Dummy 11",
description="Located in room 33",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

Camera(
id=12,
name="Dummy 12",
description="Located in room 35",
contamination=0,
date="2022-01-01",
url="https://t4.ftcdn.net/jpg/05/00/56/83/360_F_500568328_HdltBEmUOLBcRfIQTzjSslOsfuH06OCh.jpg"
).save()

logger.debug("All tables have been initialized successfully.")

def wipe_database(database_name: str):
"""
Expand Down
5 changes: 3 additions & 2 deletions backend/app/models/camera.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,15 @@ class Camera(Base):

description = pw.CharField(null=True, max_length=500)

contamination = pw.FloatField()

date = pw.DateTimeField()

url = pw.CharField(null=True, max_length=100)
url = pw.CharField(null=True, max_length=500)

def validate(self):
"""
Function to run validation on.
"""
validate_field(self, "name", v_url)
validate_field(self, "url", v_url)
return self
20 changes: 18 additions & 2 deletions backend/app/services/camera.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,14 @@ def get_camera(cls, camera_id) -> CameraDto:
camera = Camera.get_or_none(Camera.id == camera_id)
if camera is None:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)
return CameraDto(**camera)
return CameraDto(
id=camera.id,
name=camera.name,
description=camera.description,
contamination=camera.contamination,
date=camera.date,
url=camera.url
)

@classmethod
def get_cameras(cls, page=1) -> CamerasResponse:
Expand All @@ -36,6 +43,15 @@ def get_cameras(cls, page=1) -> CamerasResponse:

return CamerasResponse(
page=page,
cameras=[CameraDto(**camera) for camera in cameras],
cameras=[
CameraDto(
id=camera.id,
name=camera.name,
description=camera.description,
contamination=camera.contamination,
date=camera.date,
url=camera.url
)
for camera in cameras],
total_pages=camera_count // 10
)

0 comments on commit a485f47

Please sign in to comment.