fix dump_login_status to wrong file. (littlecodersh#349)

Changing auto_login's statusStorageDir, then KeyboardInterrupt will dump status to default file path, not new path.
daisyligf · May 9, 2017 · c8ef672 · c8ef672
1 parent f8c8c80
commit c8ef672
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/itchat/components/register.py b/itchat/components/register.py
@@ -23,14 +23,14 @@ def auto_login(self, hotReload=False, statusStorageDir='itchat.pkl',
         logger.info("You can't get access to internet or wechat domain, so exit.")
         sys.exit()
     self.useHotReload = hotReload
+    self.hotReloadDir = statusStorageDir
     if hotReload:
         if self.load_login_status(statusStorageDir,
                 loginCallback=loginCallback, exitCallback=exitCallback):
             return
         self.login(enableCmdQR=enableCmdQR, picDir=picDir, qrCallback=qrCallback,
             loginCallback=loginCallback, exitCallback=exitCallback)
         self.dump_login_status(statusStorageDir)
-        self.hotReloadDir = statusStorageDir
     else:
         self.login(enableCmdQR=enableCmdQR, picDir=picDir, qrCallback=qrCallback,
             loginCallback=loginCallback, exitCallback=exitCallback)