68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
import socketserver
|
|
import http.server
|
|
import json
|
|
import re
|
|
|
|
try:
|
|
config_file = open('config.json', 'r')
|
|
except FileNotFoundError:
|
|
print('Config file not found')
|
|
exit()
|
|
config = json.load(config_file)
|
|
|
|
def do_shutdown():
|
|
if config['shutdown_active'] == False:
|
|
return
|
|
|
|
import subprocess
|
|
import shlex
|
|
command = shlex.split('sudo shutdown -h now')
|
|
subprocess.call(command)
|
|
|
|
class MyHandler(http.server.SimpleHTTPRequestHandler):
|
|
def do_POST(self):
|
|
self.data_string = self.rfile.read(int(self.headers['Content-Length']))
|
|
try:
|
|
data = json.loads(self.data_string)
|
|
except:
|
|
data = ""
|
|
|
|
if self.path == '/shutdown':
|
|
match = False
|
|
for keyword in config['shutdown_keywords']:
|
|
if re.search(keyword, data['message']):
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
self.wfile.write(b'Shutting down...')
|
|
print('Shutting down...')
|
|
do_shutdown()
|
|
match = True
|
|
return
|
|
|
|
if not match:
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
self.wfile.write(b'Not shutting down')
|
|
print('Not shutting down')
|
|
|
|
elif self.path == '/toggle_shutdown':
|
|
config['shutdown_active'] = not config['shutdown_active']
|
|
with open('config.json', 'w') as config_file:
|
|
json.dump(config, config_file)
|
|
self.send_response(303)
|
|
self.send_header('Location', '/')
|
|
self.end_headers()
|
|
print('Shutdown toggled')
|
|
|
|
def do_GET(self):
|
|
if self.path == '/':
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
self.wfile.write(b'<html><head><title>Synology Webhook</title></head><body><p>Shutdown active: ' + str(config['shutdown_active']).encode() + b'</p><form method="post" action="/toggle_shutdown"><input type="submit" value="Toggle shutdown"></form></body></html>')
|
|
else:
|
|
self.send_response(404)
|
|
self.end_headers()
|
|
self.wfile.write(b'Not found')
|
|
|
|
httpd = socketserver.TCPServer(('', 48080), MyHandler)
|
|
httpd.serve_forever() |