Skip to content

Commit bc74b9d

Browse files
committed
fix: decorators
1 parent 2cf07af commit bc74b9d

File tree

1 file changed

+5
-13
lines changed

1 file changed

+5
-13
lines changed

algoliasearch_django/decorators.py

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -53,22 +53,14 @@ def __init__(self, model=None):
5353
if model is not None:
5454
self.models = [model]
5555
else:
56-
self.models = algolia_engine._AlgoliaEngine__registered_models
56+
self.models = algolia_engine.get_registered_models()
5757

5858
def __enter__(self):
5959
for model in self.models:
60-
post_save.disconnect(
61-
algolia_engine._AlgoliaEngine__post_save_receiver, sender=model
62-
)
63-
pre_delete.disconnect(
64-
algolia_engine._AlgoliaEngine__pre_delete_receiver, sender=model
65-
)
60+
post_save.disconnect(algolia_engine.__post_save_receiver, sender=model)
61+
pre_delete.disconnect(algolia_engine.__pre_delete_receiver, sender=model)
6662

6763
def __exit__(self, exc_type, exc_value, traceback):
6864
for model in self.models:
69-
post_save.connect(
70-
algolia_engine._AlgoliaEngine__post_save_receiver, sender=model
71-
)
72-
pre_delete.connect(
73-
algolia_engine._AlgoliaEngine__pre_delete_receiver, sender=model
74-
)
65+
post_save.connect(algolia_engine.__post_save_receiver, sender=model)
66+
pre_delete.connect(algolia_engine.__pre_delete_receiver, sender=model)

0 commit comments

Comments
 (0)