feat: add decorator to run functions only in prod (#436)

basedosdados_api/api/v1/tasks.py

@@ -4,13 +4,15 @@
 from loguru import logger
 
 from basedosdados_api.api.v1.admin import update_table_metadata
+from basedosdados_api.utils import prod_task
 
 
 @periodic_task(crontab(minute="*/10"))
 def every_ten_mins_task():
     logger.info("Am I alive between these periods?")
 
 
+@prod_task
 @periodic_task(crontab(day_of_week="0", hour="0", minute="0"))
 def update_table_metadata_task():
     update_table_metadata()

basedosdados_api/utils.py

@@ -47,3 +47,13 @@ def is_prod():
     if is_remote() and not is_dev() and not is_stag():
         return True
     return False
+
+
+def prod_task(func):
+    """Decorator that avoids function call if it isn't production"""
+
+    def wrapper(*args, **kwargs):
+        if is_prod():
+            return func(*args, **kwargs)
+
+    return wrapper