Skip to content

Commit d42cb1e

Browse files
committed
chore(places): update service tests and tasks
1 parent 1247c2f commit d42cb1e

4 files changed

Lines changed: 5 additions & 123 deletions

File tree

apps/places/services.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from apps.places.models import Place, PlaceImage, PlaceTag
99
from apps.places.tasks import (
1010
suggest_tags_from_uploaded_images,
11-
transform_uploaded_images,
1211
)
1312
from apps.users.models import BaseUser
1413

@@ -111,11 +110,6 @@ def place_images_create(
111110
images_for_detection[place_image.id] = place_image.image.path
112111
created_place_images.append(place_image)
113112

114-
transform_uploaded_images.enqueue(
115-
user_id=user.id,
116-
file_path=place_image.image.path,
117-
parent_folder=str(place_image.id),
118-
)
119113
suggest_tags_from_uploaded_images.enqueue(
120114
user_id=user.id,
121115
place_id=place_id,

apps/places/tasks.py

Lines changed: 4 additions & 93 deletions
Original file line numberDiff line numberDiff line change
@@ -5,103 +5,14 @@
55
logger = logging.getLogger(__name__)
66

77

8-
@task(queue_name="place_images")
9-
def transform_uploaded_images(user_id: int, file_path: str, parent_folder: str) -> None:
10-
from apps.image_processing.services import image_local_transform
11-
from apps.image_processing_api.constants import (
12-
ImageTransformations,
13-
TransformationFilterBlurFilter,
14-
TransformationFilterDither,
15-
TransformationFilterThumbnailResampling,
16-
)
17-
from apps.image_processing_api.data_models import (
18-
ImageTransformationDefinition,
19-
TransformationFiltersBlackAndWhite,
20-
TransformationFiltersBlur,
21-
TransformationFiltersThumbnail,
22-
)
23-
24-
logger.info(f"Transforming image {file_path}")
25-
transformations = [
26-
ImageTransformationDefinition(
27-
identifier="THUMBNAIL/default",
28-
transformation=ImageTransformations.THUMBNAIL,
29-
),
30-
ImageTransformationDefinition(
31-
identifier="THUMBNAIL/size_64",
32-
transformation=ImageTransformations.THUMBNAIL,
33-
filters=TransformationFiltersThumbnail(size=(64, 64)),
34-
),
35-
ImageTransformationDefinition(
36-
identifier="THUMBNAIL/s_320_gap_4",
37-
transformation=ImageTransformations.THUMBNAIL,
38-
filters=TransformationFiltersThumbnail(size=(320, 320), reducing_gap=4),
39-
),
40-
ImageTransformationDefinition(
41-
identifier="THUMBNAIL/s_320_gap_8_lanczos",
42-
transformation=ImageTransformations.THUMBNAIL,
43-
filters=TransformationFiltersThumbnail(
44-
size=(320, 320),
45-
reducing_gap=8,
46-
resample=TransformationFilterThumbnailResampling.LANCZOS,
47-
),
48-
),
49-
ImageTransformationDefinition(
50-
identifier="BNW/default",
51-
transformation=ImageTransformations.BLACK_AND_WHITE,
52-
),
53-
ImageTransformationDefinition(
54-
identifier="BNW/floydsteinberg",
55-
transformation=ImageTransformations.BLACK_AND_WHITE,
56-
filters=TransformationFiltersBlackAndWhite(
57-
dither=TransformationFilterDither.FLOYDSTEINBERG
58-
),
59-
),
60-
ImageTransformationDefinition(
61-
identifier="BNW/none",
62-
transformation=ImageTransformations.BLACK_AND_WHITE,
63-
filters=TransformationFiltersBlackAndWhite(
64-
dither=TransformationFilterDither.NONE
65-
),
66-
),
67-
ImageTransformationDefinition(
68-
identifier="BLUR/default",
69-
transformation=ImageTransformations.BLUR,
70-
),
71-
ImageTransformationDefinition(
72-
identifier="BLUR/gaussian_86",
73-
transformation=ImageTransformations.BLUR,
74-
filters=TransformationFiltersBlur(
75-
filter=TransformationFilterBlurFilter.GAUSSIAN_BLUR,
76-
radius=86,
77-
),
78-
),
79-
ImageTransformationDefinition(
80-
identifier="BLUR/box_48",
81-
transformation=ImageTransformations.BLUR,
82-
filters=TransformationFiltersBlur(
83-
filter=TransformationFilterBlurFilter.BOX_BLUR,
84-
radius=48,
85-
),
86-
),
87-
]
88-
applied_transformations = image_local_transform(
89-
user_id=user_id,
90-
image_path=file_path,
91-
transformations=transformations,
92-
parent_folder=parent_folder,
93-
)
94-
95-
for transformation in applied_transformations:
96-
logger.info(f"{transformation.identifier}: {transformation.path}")
97-
98-
998
@task(queue_name="place_images")
1009
def suggest_tags_from_uploaded_images(
10110
user_id: int, place_id: int, images: dict[int, str]
10211
) -> None:
103-
from apps.image_processing.core.detectors import CommonObjectDetector
104-
from apps.image_processing.data_models import DetectorImage
12+
from apps.image_processing.core.detectors.base import DetectorImage
13+
from apps.image_processing.core.detectors.common_object_detector import (
14+
CommonObjectDetector,
15+
)
10516
from apps.places.models import Place, PlaceTag
10617

10718
user_place = (

apps/places/tests/test_services.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,8 @@ def test_place_create_with_tags(user):
125125

126126

127127
@patch("apps.places.services.suggest_tags_from_uploaded_images")
128-
@patch("apps.places.services.transform_uploaded_images")
129128
@pytest.mark.django_db
130-
def test_place_images_create(mock_transform_uploaded_images, mock_suggest_tags, user):
129+
def test_place_images_create(mock_suggest_tags, user):
131130
place = PlaceFactory(user=user)
132131

133132
image1 = SimpleUploadedFile(
@@ -149,11 +148,6 @@ def test_place_images_create(mock_transform_uploaded_images, mock_suggest_tags,
149148

150149
assert place.images.count() == 1
151150

152-
mock_transform_uploaded_images.enqueue.assert_called_once_with(
153-
user_id=user.id,
154-
file_path=f"{settings.MEDIA_ROOT}/place_images/test_image1.jpg",
155-
parent_folder=str(1),
156-
)
157151
mock_suggest_tags.enqueue.assert_called_once_with(
158152
user_id=user.id,
159153
place_id=place.id,

apps/places/tests/test_tasks.py

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)