commit-confirm-notify.py
No OneTemporary
Actions

Size

981 B

Referenced Files

None

Subscribers

None

commit-confirm-notify.py
View Options

	#!/usr/bin/env python3
	import os
	import sys
	import time

	# Minutes before reboot to trigger notification.
	intervals = [1, 5, 15, 60]

	def notify(interval):
	s = "" if interval == 1 else "s"
	time.sleep((minutes - interval) * 60)
	message = ('"[commit-confirm] System is going to reboot in '
	f'{interval} minute{s} to rollback the last commit.\n'
	'Confirm your changes to cancel the reboot."')
	os.system("wall -n " + message)

	if __name__ == "__main__":
	# Must be run as root to call wall(1) without a banner.
	if len(sys.argv) != 2 or os.getuid() != 0:
	exit(1)
	minutes = int(sys.argv[1])
	# Drop the argument from the list so that the notification
	# doesn't kick in immediately.
	if minutes in intervals:
	intervals.remove(minutes)
	for interval in sorted(intervals, reverse=True):
	if minutes >= interval:
	notify(interval)
	minutes -= (minutes - interval)
	exit(0)