From b211d0e92bfeb79cf5abe825c331d9f038e2313c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marco=20Trevisan=20=28Trevi=C3=B1o=29?= <mail@3v1n0.net>
Date: Wed, 10 Jun 2020 18:25:03 +0200
Subject: [PATCH] validity-sensors-initializer: Use calibration file if any is
 available

Otherwise save generated one in temporary path
---
 validity-sensors-initializer.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/validity-sensors-initializer.py b/validity-sensors-initializer.py
index 3dbe38f..a287f8b 100644
--- a/validity-sensors-initializer.py
+++ b/validity-sensors-initializer.py
@@ -168,8 +168,18 @@ class VFSInitializer():
         self.sleep()
         self.restart()
 
-        print('Calibrating...')
-        calibrate()
+        print('Calibrating, re-using calib-data.bin if any...')
+        if os.path.exists('calib-data.bin'):
+            calibrate(calib_data_path='calib-data.bin')
+        else:
+            try:
+                calib_data = os.path.join(tempfile.mkdtemp(), 'calib-data.bin')
+                calibrate(calib_data_path=calib_data)
+                print('Calibration data saved at {}'.format(calib_data))
+            except Exception as e:
+                print('Calibration failed using device data ({}), ' \
+                      'You can try loading a local `calib-data.bin` blob, ' \
+                      'the device should work anyway, so skipping...'.format(e))
 
         print('Init database...')
         init_db()