Disable shutdown/reboot unless on Raspberry Pi (02afe066) · Commits · OpenFlexure / openflexure-microscope-server

openflexure_microscope/api/v2/blueprints/actions/init.py

+2 −3

Original line number	Diff line number	Diff line
		@@ -3,7 +3,6 @@ Top-level representation of enabled actions
		"""

		from flask import Blueprint, url_for, jsonify
		from sys import platform

		from openflexure_microscope.api.utilities import blueprint_for_module
		from openflexure_microscope.utilities import get_docstring, description_from_view
		@@ -41,12 +40,12 @@ _actions = {
		"shutdown": {
		"rule": "/system/shutdown/",
		"view_class": system.ShutdownAPI,
		"conditions": (platform == "linux"),
		"conditions": system.is_raspberrypi(),
		},
		"reboot": {
		"rule": "/system/reboot/",
		"view_class": system.RebootAPI,
		"conditions": (platform == "linux"),
		"conditions": system.is_raspberrypi(),
		},
		}

+8 −0

Original line number	Diff line number	Diff line
		from openflexure_microscope.api.views import MicroscopeView

		import subprocess
		import os
		from sys import platform


		def is_raspberrypi():
		if (platform != "linux"):
		return False
		else:
		return (os.uname()[1] == 'raspberrypi')

		class ShutdownAPI(MicroscopeView):
		"""
		Attempt to shutdown the device