19 lines
624 B
Python
19 lines
624 B
Python
|
|
||
|
|
import ops.cmd
|
||
|
|
import ops
|
||
|
|
import ops.env
|
||
|
|
import ops.cmd.safetychecks
|
||
|
|
OpsCommandException = ops.cmd.OpsCommandException
|
||
|
|
|
||
|
|
def mySafetyCheck(self):
|
||
|
|
good = True
|
||
|
|
msgparts = []
|
||
|
|
if (ops.env.get('OPS_NOMEMORY').upper() == 'TRUE'):
|
||
|
|
good = False
|
||
|
|
msgparts.append('OPS_NOMEMORY is set to TRUE, you should probably not run this')
|
||
|
|
if (not self.validateInput()):
|
||
|
|
good = False
|
||
|
|
msgparts.append('Your command did not pass input validation')
|
||
|
|
msg = '\n\t'.join(msgparts)
|
||
|
|
return (good, msg)
|
||
|
|
ops.cmd.safetychecks.addSafetyHandler('memory', 'ops.cmd.memory.mySafetyCheck')
|