实现Python程序在重启后恢复到上次关闭时的状态，通常需要在程序退出前保存其状态，并在下次启动时加载该状态。以下是几种常见的实现方法： ### 1. 使用文件存储状态信息 可以将程序的关键状态信息（如变量值、运行上下文等）序列化并保存到本地文件中，例如使用`pickle`模块或`json`模块进行序列化。 ```python import pickle import os STATE_FILE = "program_state.pkl" def save_state(state): with open(STATE_FILE, 'wb') as f: pickle.dump(state, f) def load_state(): if os.path.exists(STATE_FILE): with open(STATE_FILE, 'rb') as f: return pickle.load(f) return None # 示例程序逻辑 state = load_state() if state is None: state = {"counter": 0} else: print("恢复状态:", state) # 模拟程序运行 try: while True: state["counter"] += 1 print("当前计数器:", state["counter"]) # 模拟错误触发重启 if state["counter"] > 5: raise Exception("模拟错误") except Exception as e: print("发生错误，保存状态并退出") save_state(state) ``` ### 2. 使用数据库存储状态 对于更复杂的应用场景，可以将状态信息存储在轻量级数据库中，如SQLite。这种方式适合需要持久化大量数据或支持多进程访问的场景。 ```python import sqlite3 def init_db(): conn = sqlite3.connect('state.db') c = conn.cursor() c.execute('''CREATE TABLE IF NOT EXISTS state (key TEXT PRIMARY KEY, value TEXT)''') conn.commit() conn.close() def save_state(key, value): conn = sqlite3.connect('state.db') c = conn.cursor() c.execute("REPLACE INTO state (key, value) VALUES (?, ?)", (key, str(value))) conn.commit() conn.close() def load_state(key): conn = sqlite3.connect('state.db') c = conn.cursor() c.execute("SELECT value FROM state WHERE key=?", (key,)) row = c.fetchone() conn.close() return int(row[0]) if row else 0 # 初始化数据库 init_db() # 示例程序逻辑 counter = load_state("counter") print("恢复状态:", counter) try: while True: counter += 1 print("当前计数器:", counter) save_state("counter", counter) # 模拟错误触发重启 if counter > 5: raise Exception("模拟错误") except Exception as e: print("发生错误，保存状态并退出") ``` ### 3. 使用环境变量或注册表（Windows平台） 在某些情况下，也可以使用环境变量或Windows注册表来存储状态信息。这种方式适用于需要跨进程共享状态的场景。 ```python import winreg def set_registry_value(key_name, value_name, value): try: key = winreg.CreateKey(winreg.HKEY_CURRENT_USER, key_name) winreg.SetValueEx(key, value_name, 0, winreg.REG_SZ, value) finally: winreg.CloseKey(key) def get_registry_value(key_name, value_name): try: key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, key_name) value, _ = winreg.QueryValueEx(key, value_name) return value except FileNotFoundError: return None finally: winreg.CloseKey(key) # 示例程序逻辑 key_name = r"Software/MyApp" value_name = "Counter" counter = get_registry_value(key_name, value_name) if counter is None: counter = 0 else: counter = int(counter) print("恢复状态:", counter) try: while True: counter += 1 print("当前计数器:", counter) set_registry_value(key_name, value_name, str(counter)) # 模拟错误触发重启 if counter > 5: raise Exception("模拟错误") except Exception as e: print("发生错误，保存状态并退出") ``` ### 4. 结合自动重启机制 可以结合引用中的自动重启机制，在程序崩溃时自动重启并恢复状态[^1]。 ```python import sys import time def restart_program(): print("重启程序...") python = sys.executable os.execl(python, python, *sys.argv) # 示例程序逻辑 counter = load_state() if counter is None: counter = {"counter": 0} else: print("恢复状态:", counter) try: while True: counter["counter"] += 1 print("当前计数器:", counter["counter"]) # 模拟错误触发重启 if counter["counter"] > 5: raise Exception("模拟错误") time.sleep(1) except Exception as e: save_state(counter) restart_program() ```