在Python中覆盖基本信号（SIGINT，SIGQUIT，SIGKILL ??）
|
我正在编写一个根据公司政策添加普通UNIX帐户（即修改/ etc / passwd，/ etc / group和/ etc / shadow）的程序。它还做一些花哨的事情，例如向用户发送电子邮件。
我已经完成所有代码的工作，但是有三段非常关键的代码，它们会更新上面的三个文件。该代码已经相当健壮，因为它锁定了这些文件（例如/etc/passwd.lock），写入了一个临时文件（例如/etc/passwd.tmp），然后用该临时文件覆盖了原始文件。我很高兴它不会与我程序的其他运行版本或系统useradd，usermod，passwd等程序兼容。
我最担心的是这些部分中间的Ctrl + C，Ctrl + D或kill命令。这导致我进入信号模块，它似乎恰好满足了我的要求：在“关键”区域中忽略某些信号。
我使用的是Python的旧版本，它没有signal.SIG_IGN，所以我有一个很棒的“传递”功能：
def passer(*a):
pass
def passer(a=None, b=None):
pass
def signalhander(enable):
signallist = (signal.SIGINT, signal.SIGQUIT, signal.SIGABRT, signal.SIGPIPE, signal.SIGALRM, signal.SIGTERM, signal.SIGKILL)
if enable:
for i in signallist:
signal.signal(i, passer)
else:
for i in signallist:
signal.signal(i, abort)
return
def abort(a=None, b=None):
sys.exit(\'\\nAccount was not created.\\n\')
return
signalhander(True)
print(\'Enabled\')
time.sleep(10) # ^C during this sleep
try:
except:
KeyBoardInterrupt
EOFError
def criticalRemoveLock(file):
try:
if os.path.isFile(file):
os.remove(file)
else:
return True
except (KeyboardInterrupt, EOFError):
return criticalRemoveLock(file)
def criticalOverwrite(tmp, file):
try:
if os.path.isFile(tmp):
shutil.copy2(tmp, file)
os.remove(tmp)
else:
return True
except (KeyboardInterrupt, EOFError):
return criticalOverwrite(tmp, file)
没有找到相关结果
已邀请:
1 个回复
讹巳漓把备