Das deutsche Python-Forum

Code: Alles auswählen

import os, atexit, fcntl

PIDFILEFD = None

# Öffnet die PID-Datei und gibt die PID des bestehenden Prozesses zurück falls
# dieser existiert, oder hält selbst das Lock auf den PIDFILE und gibt None zurück.
# In diesem Fall existierte kein Prozess.
def openpidfile(pidfile):
    if PIDFILEFD is not None:
        raise RuntimeError("Wir haben schon eine PID-Datei geöffnet!")
    # Diese Operation erstellt die Datei ohne eine bestehende truncaten
    # (wie bei open(...,"w")), deswegen lowlevel IO.
    PIDFILEFD = os.open(pidfile,os.O_RDWR|os.O_CREAT)
    try:
        # Wenn exklusives Locking geht läuft der Prozess noch nicht.
        fcntl.flock(PIDFILEFD,fcntl.LOCK_EX|fcntl.LOCK_NB)
    except IOError:
        # Der Prozess läuft und hat ein shared lock auf die Datei, was wir auch machen
        # können, und dadurch die PID aus der Datei auslesen. Das shared-lock kann
        # auch ruhig blockieren, das bedeutet nur dass der Prozess vor uns noch ein
        # exklusives lock hat, und wir warten dann bis er es freigibt.
        fcntl.flock(PIDFILEFD,fcntl.LOCK_SH)
        # PIDs sind maximal 10 Zeichen lang als Dezimalzahl (oder noch weniger
        # sogar).
        try:
            return int(os.read(PIDFILEFD,10))
        finally:
            # Hier könnte man auch den File offen lassen mit dem advisory lock, daraus
            # würde sich ergeben dass ein neuer Prozess so lange nicht gestartet werden
            # kann bis dieses Script sich beendet. Das ist aber im Normalfall nicht
            # notwendig.
            fcntl.flock(PIDFILEFD,fcntl.LOCK_UN)
            os.close(PIDFILEFD)
            PIDFILEFD = None
    # Wir haben das exklusive Lock, und schreiben jetzt die PID in die Datei.
    os.ftruncate(PIDFILEFD,0)
    os.write(PIDFILEFD,str(os.getpid()))
    # Und geben das lock frei auf LOCK_SH, so dass andere Prozesse merken dass
    # wir da sind und die PID lesen können.
    fcntl.flock(PIDFILEFD,fcntl.LOCK_SH)
    atexit.register(_closepidfile)

# Das folgende ist nicht _umbedingt_ notwendig da die Ressourcen eh freigegeben
# werden wenn der Prozess vom Betriebssystem zerstört wird, ist aber zumindest
# höflich die Dinge selbst freizugeben.
def _closepidfile():
    fcntl.flock(PIDFILEFD,fcntl.LOCK_UN)
    os.close(PIDFILEFD)