Expérimente le cache des paramètres

openfisca_france/france_taxbenefitsystem.py

@@ -1,6 +1,10 @@
 # -*- coding: utf-8 -*-
 
 import os
+import pickle
+
+import checksumdir
+import appdirs
 
 from openfisca_core.taxbenefitsystems import TaxBenefitSystem
 
@@ -11,6 +15,7 @@
 
 
 COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))
+CACHE_PARAMETER_TREE = True
 
 
 class FranceTaxBenefitSystem(TaxBenefitSystem):
@@ -24,9 +29,7 @@ class FranceTaxBenefitSystem(TaxBenefitSystem):
     def __init__(self):
         TaxBenefitSystem.__init__(self, entities)
 
-        param_dir = os.path.join(COUNTRY_DIR, 'parameters')
-        self.load_parameters(param_dir)
-
+        self.load_parameters_from_cache()
         self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'model'))
         self.cache_blacklist = conf_cache_blacklist
 
@@ -36,6 +39,27 @@ def __init__(self):
             "simulation_example": couple,
             }
 
+    def load_parameters_from_cache(self):
+        param_dir = os.path.join(COUNTRY_DIR, 'parameters')
+        if not CACHE_PARAMETER_TREE:
+            return self.load_parameters(param_dir)
+        dir_hash = checksumdir.dirhash(param_dir)
+        cached_dir = appdirs.user_cache_dir(self.get_package_metadata()['name'])
+        pickle_path = os.path.join(cached_dir, f'{dir_hash}.pickle')
+
+        if os.path.isfile(pickle_path):
+            # print("Found cached parameters")
+            with open(pickle_path, "rb") as pickle_file:
+                self.parameters = pickle.load(pickle_file)
+        else:
+            # print("No cached parameters")
+            if not os.path.isdir(cached_dir):
+                os.mkdir(cached_dir)
+            self.load_parameters(param_dir)
+            with open(pickle_path, "wb+") as pickle_file:
+                pickle.dump(self.parameters, pickle_file)
+
+
     def prefill_cache(self):
         # Compute one "zone APL" variable, to pre-load CSV of "code INSEE commune" to "Zone APL".
         from .model.prestations import aides_logement

setup.py

@@ -50,6 +50,8 @@
     include_package_data = True,  # Will read MANIFEST.in
     install_requires = [
         "OpenFisca-Core >=31.0,<33.0",
+        "checksumdir == 1.1.6",
+        "appdirs == 1.4.3",
         ],
     message_extractors = {"openfisca_france": [
         ("**.py", "python", None),