From d14c36e451c052ec96bcb7835937059464064666 Mon Sep 17 00:00:00 2001
From: su530201 <olivier.kaufmann@umons.ac.be>
Date: Sat, 10 Dec 2022 19:45:33 +0100
Subject: [PATCH] Adds a shutdown command...
---
ohmpi.py | 54 +++++++++++++++++++++++++++++++++++-------------------
1 file changed, 35 insertions(+), 19 deletions(-)
diff --git a/ohmpi.py b/ohmpi.py
index cf050889..373651c3 100644
--- a/ohmpi.py
+++ b/ohmpi.py
@@ -651,6 +651,26 @@ class OhmPi(object):
shutil.rmtree('data')
os.mkdir('data')
+ def reset_mux(self, cmd_id=None):
+ """Switches off all multiplexer relays.
+
+ Parameters
+ ----------
+ cmd_id : str, optional
+ Unique command identifier
+ """
+ if self.on_pi and self.use_mux:
+ roles = ['A', 'B', 'M', 'N']
+ for i in range(0, 4):
+ for j in range(1, self.max_elec + 1):
+ self._switch_mux(j, 'off', roles[i])
+ self.exec_logger.debug('All MUX switched off.')
+ elif not self.on_pi:
+ self.exec_logger.warning('Cannot reset mux while in simulation mode...')
+ else:
+ self.exec_logger.warning('You cannot use the multiplexer because use_mux is set to False.'
+ ' Set use_mux to True to use the multiplexer...')
+
def restart(self, cmd_id=None):
"""Restarts the Raspberry Pi
@@ -1299,6 +1319,21 @@ class OhmPi(object):
if quadrupole[i] > 0:
self._switch_mux(quadrupole[i], 'off', roles[i])
+ def shutdown(self, cmd_id=None):
+ """Shuts the Raspberry Pi down
+
+ Parameters
+ ----------
+ cmd_id : str, optional
+ Unique command identifier
+ """
+
+ if self.on_pi:
+ self.exec_logger.info(f'Shutting down the pi following command {cmd_id}...')
+ os.system('reboot')
+ else:
+ self.exec_logger.warning('Not on Raspberry Pi, skipping shutdown...')
+
def test_mux(self, activation_time=1.0, address=0x70):
"""Interactive method to test the multiplexer.
@@ -1348,25 +1383,6 @@ class OhmPi(object):
time.sleep(activation_time)
print('Test finished.')
- def reset_mux(self, cmd_id=None):
- """Switches off all multiplexer relays.
-
- Parameters
- ----------
- cmd_id : str, optional
- Unique command identifier
- """
- if self.on_pi and self.use_mux:
- roles = ['A', 'B', 'M', 'N']
- for i in range(0, 4):
- for j in range(1, self.max_elec + 1):
- self._switch_mux(j, 'off', roles[i])
- self.exec_logger.debug('All MUX switched off.')
- elif not self.on_pi:
- self.exec_logger.warning('Cannot reset mux while in simulation mode...')
- else:
- self.exec_logger.warning('You cannot use the multiplexer because use_mux is set to False.'
- ' Set use_mux to True to use the multiplexer...')
def _update_acquisition_settings(self, config):
warnings.warn('This function is deprecated, use update_settings() instead.', DeprecationWarning)
--
GitLab