Split xsetbg.py to xsetbg-wsgi.py to separate the core from the server.

[xsetbg.git] / xsetbg-wsgi.py

diff --git a/xsetbg-wsgi.py b/xsetbg-wsgi.py
new file mode 100755 (executable)
index 0000000..d7815a9
--- /dev/null
+++ b/xsetbg-wsgi.py
@@ -0,0 +1,75 @@
+#! /usr/bin/env python
+"""XSetBg (WSGI version)
+
+"""
+
+__version__ = "$Revision$"[11:-2]
+__revision__ = "$Id$"[5:-2]
+__date__ = "$Date$"[7:-2]
+
+__author__ = "Oleg Broytman <phd@phdru.name>"
+__copyright__ = "Copyright (C) 2000-2010 PhiloSoft Design"
+__license__ = "GNU GPL"
+
+
+from wsgiref import simple_server
+from wsgiref.handlers import SimpleHandler
+from wsgiref.simple_server import WSGIServer, make_server
+simple_server.ServerHandler = SimpleHandler # Stop logging to stdout
+
+from xsetbg import host, port, change as _change
+
+
+def published(func):
+    func._wsgi_published = True
+    return func
+
+@published
+def change(force=False):
+    _change()
+
+@published
+def force():
+    _change(force=True)
+
+@published
+def stop():
+    QuitWSGIServer._quit_flag = True
+
+
+g = globals().copy()
+commands = dict([(name, g[name]) for name in g
+    if getattr(g[name], '_wsgi_published', False)])
+del g
+
+class QuitWSGIServer(WSGIServer):
+    _quit_flag = False
+
+    def serve_forever(self):
+        while not self._quit_flag:
+            self.handle_request()
+
+def app(env, start_response):
+    command = env['PATH_INFO'][1:] # Remove the leading slash
+    if command not in commands:
+        status = '404 Not found'
+        response_headers = [('Content-type', 'text/plain')]
+        start_response(status, response_headers)
+        return ['The command was not found.\n']
+
+    try:
+        commands[command]()
+    except:
+        status = '500 Error'
+        response_headers = [('Content-type', 'text/plain')]
+        start_response(status, response_headers)
+        return ['Error occured!\n']
+
+    status = '200 OK'
+    response_headers = [('Content-type', 'text/plain')]
+    start_response(status, response_headers)
+    return ['Ok\n']
+
+force()
+httpd = make_server(host, port, app, server_class=QuitWSGIServer)
+httpd.serve_forever()