[change] Added time limits to celery tasks #608

Closes #608

Co-authored-by: Gagan Deep <pandafy.dev@gmail.com>

Author: c4llous

openwisp_controller/config/tasks.py

@@ -7,6 +7,8 @@
 from django.core.exceptions import ObjectDoesNotExist
 from swapper import load_model
 
+from openwisp_utils.tasks import OpenwispCeleryTask
+
 logger = logging.getLogger(__name__)
 
 
@@ -54,7 +56,7 @@ def create_vpn_dh(vpn_pk):
         vpn.save()
 
 
-@shared_task
+@shared_task(base=OpenwispCeleryTask)
 def invalidate_devicegroup_cache_change(instance_id, model_name):
     from .api.views import DeviceGroupCommonName
 
@@ -70,7 +72,7 @@ def invalidate_devicegroup_cache_change(instance_id, model_name):
         DeviceGroupCommonName.certificate_change_invalidates_cache(instance_id)
 
 
-@shared_task
+@shared_task(base=OpenwispCeleryTask)
 def invalidate_devicegroup_cache_delete(instance_id, model_name, **kwargs):
     from .api.views import DeviceGroupCommonName
 
@@ -87,7 +89,7 @@ def invalidate_devicegroup_cache_delete(instance_id, model_name, **kwargs):
         )
 
 
-@shared_task(soft_time_limit=1200)
+@shared_task(base=OpenwispCeleryTask)
 def trigger_vpn_server_endpoint(endpoint, auth_token, vpn_id):
     response = requests.post(
         endpoint,

openwisp_controller/subnet_division/tasks.py

@@ -5,6 +5,8 @@
 from django.utils.translation import gettext_lazy as _
 from swapper import load_model
 
+from openwisp_utils.tasks import OpenwispCeleryTask
+
 logger = logging.getLogger(__name__)
 
 Subnet = load_model('openwisp_ipam', 'Subnet')
@@ -69,7 +71,7 @@ def update_subnet_name_description(rule_id):
     )
 
 
-@shared_task
+@shared_task(base=OpenwispCeleryTask)
 def provision_extra_ips(rule_id, old_number_of_ips):
     def _create_ipaddress_and_subnetdivision_index_objects(ips, indexes):
         IpAddress.objects.bulk_create(ips)