how can i define the action of exit

when I input "python xxx.py stop"，I need to save something for my program,But I cannot find the method to solve it ,even if I read your source code.I will very grateful to you if you solve my problem.Thank you first.