daemonize server

import os

import signal

import sys

import socket

import errno

from django.conf import settings

from django.core.management.base import BaseCommand

from django.core.handlers.wsgi import WSGIHandler

from django.core.servers.basehttp import get_internal_wsgi_application, run 

from django.utils.encoding import force_text

class Command(BaseCommand):

    help = 'HTTP/OPDS built-in server'

    def add_arguments(self, parser):

        parser.add_argument('command', help='Start, stop or restart server')

        #parser.add_argument('stop',help='Start server')

        parser.add_argument('--host',action='store', dest='host', default="0.0.0.0", help='Set server binding address')

        parser.add_argument('--port',action='store', dest='port', default=8001, help='Ser server port')

        parser.add_argument('--daemon',action='store_true', dest='daemonize', default=False, help='Daemonize server')

    def handle(self, *args, **options):

        self.pidfile = os.path.join(settings.BASE_DIR, "sopds-server.pid")

        try:

            action = options['command']

        except IndexError:

            print("You must provide an action. Possible actions are start, stop and restart.")

            raise SystemExit

        self.addr = options['host']

        self.port = int(options['port'])

        if options["daemonize"]:

            daemonize()

        if action == "start":

            self.start()

        elif action == "stop":

            pid = open(self.pidfile, "r").read()

            self.stop(pid)

        elif action == "restart":

            pid = open(self.pidfile, "r").read()

            self.restart(pid)

    def start(self):

        writepid(self.pidfile)

        quit_command = 'CTRL-BREAK' if sys.platform == 'win32' else 'CONTROL-C' 

        self.stdout.write((

            "Starting SOPDS server at http://%(addr)s:%(port)s/\n"

            "Quit the server with %(quit_command)s.\n"

        ) % {

            "addr": '[%s]' % self.addr,

            "port": self.port,

            "quit_command": quit_command,

        })

        try:

            handler = get_internal_wsgi_application()

            run(self.addr, int(self.port), handler,ipv6=False, threading=True)

        except socket.error as e:

            # Use helpful error messages instead of ugly tracebacks.

            ERRORS = {

                errno.EACCES: "You don't have permission to access that port.",

                errno.EADDRINUSE: "That port is already in use.",

                errno.EADDRNOTAVAIL: "That IP address can't be assigned to.",

            }

            try:

                error_text = ERRORS[e.errno]

            except KeyError:

                error_text = force_text(e)

            self.stderr.write("Error: %s" % error_text)

            # Need to use an OS exit because sys.exit doesn't work in a thread

            os._exit(1)

        except KeyboardInterrupt:

            sys.exit(0) 

    def stop(self, pid):

        os.kill(int(pid), signal.SIGTERM)

    def restart(self, pid):

        self.stop(pid)

        self.start()

def writepid(pid_file):

    """

    Write the process ID to disk.

    """

    fp = open(pid_file, "w")

    fp.write(str(os.getpid()))

    fp.close()

def daemonize():

    """

    Detach from the terminal and continue as a daemon.

    """

    # swiped from twisted/scripts/twistd.py

    # See http://www.erlenstar.demon.co.uk/unix/faq_toc.html#TOC16

    if sys.platform == "win32":

        print("On Windows platform Daemonize not working.")

        return

    if os.fork():   # launch child and...

        os._exit(0) # kill off parent

    os.setsid()

    if os.fork():   # launch child and...

        os._exit(0) # kill off parent again.

    os.umask(0)

    null = os.open("/dev/null", os.O_RDWR)

    for i in range(3):

        try:

            os.dup2(null, i)

        except OSError as e:

            if e.errno != errno.EBADF:

                raise

    os.close(null)

12644

 No newline at end of file