Use decorators with decorator syntax

openhealthcare · Nov 20, 2018 · 3a05941 · 3a05941
1 parent 4ccc433
commit 3a05941
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 16 deletions.
diff --git a/intrahospital_api/services/appointments/service.py b/intrahospital_api/services/appointments/service.py
@@ -87,11 +87,6 @@ def load_patients():
     return updated
 
 
-# not an invalid, name, its not a constant, seperate out
-# for testing purposes
-# pylint: disable=invalid-name
-batch_load = load_utils.batch_load(
-    service_name="appointments"
-)(
-    load_patients
-)
+@load_utils.batch_load('appointments')
+def batch_load():
+    return load_patients()
diff --git a/intrahospital_api/services/demographics/service.py b/intrahospital_api/services/demographics/service.py
@@ -161,11 +161,7 @@ def for_hospital_number(hospital_number):
 
     return result
 
-# not an invalid, name, its not a constant, seperate out
-# for testing purposes
-# pylint: disable=invalid-name
-batch_load = load_utils.batch_load(
-    service_name="demographics"
-)(
-    load_patients
-)
+
+@load_utils.batch_load('demographics')
+def batch_load():
+    return load_patients()