ฉันกำลังทำงานกับสคริปต์หลามที่เริ่มต้นกระบวนการและการเชื่อมต่อฐานข้อมูลหลายอย่าง ทุก ๆ ครั้งที่ฉันต้องการฆ่าสคริปต์ด้วยสัญญาณCtrl+ Cและฉันต้องการจะล้างข้อมูล

ใน Perl ฉันจะทำสิ่งนี้:

$SIG{'INT'} = 'exit_gracefully';

sub exit_gracefully {
    print "Caught ^C \n";
    exit (0);
}

ฉันจะทำอะนาล็อกของสิ่งนี้ใน Python ได้อย่างไร

ลงทะเบียนตัวจัดการของคุณด้วยsignal.signalสิ่งนี้:

#!/usr/bin/env python
import signal
import sys

def signal_handler(sig, frame):
    print('You pressed Ctrl+C!')
    sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)
print('Press Ctrl+C')
signal.pause()

รหัสดัดแปลงมาจากที่นี่

เอกสารเพิ่มเติมเกี่ยวกับsignalสามารถพบได้ที่นี่  

คุณสามารถปฏิบัติต่อมันเหมือนมีข้อยกเว้น (KeyboardInterrupt) เหมือนกับอื่น ๆ สร้างไฟล์ใหม่และเรียกใช้จากเชลล์ด้วยเนื้อหาต่อไปนี้เพื่อดูว่าฉันหมายถึงอะไร:

import time, sys

x = 1
while True:
    try:
        print x
        time.sleep(.3)
        x += 1
    except KeyboardInterrupt:
        print "Bye"
        sys.exit()

และในฐานะผู้จัดการบริบท:

import signal

class GracefulInterruptHandler(object):

    def __init__(self, sig=signal.SIGINT):
        self.sig = sig

    def __enter__(self):

        self.interrupted = False
        self.released = False

        self.original_handler = signal.getsignal(self.sig)

        def handler(signum, frame):
            self.release()
            self.interrupted = True

        signal.signal(self.sig, handler)

        return self

    def __exit__(self, type, value, tb):
        self.release()

    def release(self):

        if self.released:
            return False

        signal.signal(self.sig, self.original_handler)

        self.released = True

        return True

ใช้:

with GracefulInterruptHandler() as h:
    for i in xrange(1000):
        print "..."
        time.sleep(1)
        if h.interrupted:
            print "interrupted!"
            time.sleep(2)
            break

ตัวจัดการซ้อน:

with GracefulInterruptHandler() as h1:
    while True:
        print "(1)..."
        time.sleep(1)
        with GracefulInterruptHandler() as h2:
            while True:
                print "\t(2)..."
                time.sleep(1)
                if h2.interrupted:
                    print "\t(2) interrupted!"
                    time.sleep(2)
                    break
        if h1.interrupted:
            print "(1) interrupted!"
            time.sleep(2)
            break

จากที่นี่: https://gist.github.com/2907502

คุณสามารถจัดการCTRL+ CโดยจับKeyboardInterruptข้อยกเว้น คุณสามารถใช้โค้ดการล้างข้อมูลใด ๆ ในตัวจัดการข้อยกเว้น

จากเอกสารของ Python :

import signal
import time

def handler(signum, frame):
    print 'Here you go'

signal.signal(signal.SIGINT, handler)

time.sleep(10) # Press Ctrl+c here

อีกตัวอย่างหนึ่ง

เรียกว่าmainเป็นฟังก์ชั่นหลักและexit_gracefullyเป็นตัวจัดการCTRL+c

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        pass
    finally:
        exit_gracefully()

ฉันดัดแปลงรหัสจาก @udi เพื่อรองรับสัญญาณหลายสัญญาณ (ไม่มีอะไรแฟนซี):

class GracefulInterruptHandler(object):
    def __init__(self, signals=(signal.SIGINT, signal.SIGTERM)):
        self.signals = signals
        self.original_handlers = {}

    def __enter__(self):
        self.interrupted = False
        self.released = False

        for sig in self.signals:
            self.original_handlers[sig] = signal.getsignal(sig)
            signal.signal(sig, self.handler)

        return self

    def handler(self, signum, frame):
        self.release()
        self.interrupted = True

    def __exit__(self, type, value, tb):
        self.release()

    def release(self):
        if self.released:
            return False

        for sig in self.signals:
            signal.signal(sig, self.original_handlers[sig])

        self.released = True
        return True

รหัสนี้รองรับการใช้อินเตอร์รัปต์คีย์บอร์ด ( SIGINT) และSIGTERM( kill <process>)

ตรงกันข้ามกับคำตอบของMatt Jเขาใช้วัตถุง่าย ๆ สิ่งนี้ทำให้ฉันมีความเป็นไปได้ที่จะแยกวิเคราะห์ตัวจัดการนี้ไปยังกระทู้ทั้งหมดที่จำเป็นต้องหยุดความลับ

class SIGINT_handler():
    def __init__(self):
        self.SIGINT = False

    def signal_handler(self, signal, frame):
        print('You pressed Ctrl+C!')
        self.SIGINT = True


handler = SIGINT_handler()
signal.signal(signal.SIGINT, handler.signal_handler)

ที่อื่น ๆ

while True:
    # task
    if handler.SIGINT:
        break

คุณสามารถใช้ฟังก์ชั่นในโมดูลสัญญาณในตัวของ Python เพื่อตั้งค่าตัวจัดการสัญญาณในไพ ธ อน โดยเฉพาะsignal.signal(signalnum, handler)ฟังก์ชั่นที่ใช้ในการลงทะเบียนฟังก์ชั่นการส่งสัญญาณhandlersignalnum

ขอบคุณสำหรับคำตอบที่มีอยู่ แต่เพิ่ม signal.getsignal()

import signal

# store default handler of signal.SIGINT
default_handler = signal.getsignal(signal.SIGINT)
catch_count = 0

def handler(signum, frame):
    global default_handler, catch_count
    catch_count += 1
    print ('wait:', catch_count)
    if catch_count > 3:
        # recover handler for signal.SIGINT
        signal.signal(signal.SIGINT, default_handler)
        print('expecting KeyboardInterrupt')

signal.signal(signal.SIGINT, handler)
print('Press Ctrl+c here')

while True:
    pass

หากคุณต้องการให้แน่ใจว่ากระบวนการล้างข้อมูลของคุณเสร็จสิ้นฉันจะเพิ่มคำตอบของ Matt Jโดยใช้SIG_IGNเพื่อที่SIGINTจะไม่สนใจสิ่งต่อไปซึ่งจะป้องกันไม่ให้การขัดจังหวะการล้างข้อมูลของคุณ

import signal
import sys

def signal_handler(signum, frame):
    signal.signal(signum, signal.SIG_IGN) # ignore additional signals
    cleanup() # give your process a chance to clean up
    sys.exit(0)

signal.signal(signal.SIGINT, signal_handler) # register the signal with the signal handler first
do_stuff()

โดยส่วนตัวฉันไม่สามารถใช้ลอง / ยกเว้น KeyboardInterrupt เพราะฉันใช้โหมดซ็อกเก็ตมาตรฐาน (IPC) ซึ่งกำลังบล็อกอยู่ ดังนั้น SIGINT จึงเข้ามา แต่มาหลังจากได้รับข้อมูลบนซ็อกเก็ตเท่านั้น

การตั้งค่าตัวจัดการสัญญาณทำงานเหมือนกัน

ในทางกลับกันสิ่งนี้ใช้ได้กับเทอร์มินัลจริงเท่านั้น สภาพแวดล้อมที่เริ่มต้นอื่น ๆ อาจไม่ยอมรับCtrl+ Cหรือจัดการสัญญาณล่วงหน้า

นอกจากนี้ยังมี "ข้อยกเว้น" และ "BaseExceptions" ใน Python ซึ่งแตกต่างในแง่ที่ล่ามจำเป็นต้องออกจากตัวเองอย่างหมดจดดังนั้นข้อยกเว้นบางอย่างมีลำดับความสำคัญสูงกว่าคนอื่น ๆ (ข้อยกเว้นมาจาก BaseException)