diff --git a/RoutesApp/models.py b/RoutesApp/models.py index 4c105cf..9b4d3ac 100644 --- a/RoutesApp/models.py +++ b/RoutesApp/models.py @@ -95,6 +95,17 @@ class Route(BaseModel): return False + def get_permission_for_highlight(self): + from SubscribesApp.funcs import get_cur_user_subscribe + user_subscribe = get_cur_user_subscribe(self.owner) + if not user_subscribe: + return False + data_Dict = user_subscribe.remains_route_adding_options() + if data_Dict['remains_route_highlight_count'] > 0: + return True + + return False + def from_country_n_city_str(self): res = _('Неизвестно') diff --git a/SubscribesApp/models.py b/SubscribesApp/models.py index 89b599d..36c945e 100644 --- a/SubscribesApp/models.py +++ b/SubscribesApp/models.py @@ -132,6 +132,10 @@ class SubscribeForUser(BaseModel): ).exclude( id=self.id ) - subscribes_for_user.update(enable=False) + subscribes_for_user.update( + enable=False, + used_route_rising_count=0, + used_route_highlight_count=0, + ) return self \ No newline at end of file