修复不立即登录问题

2019-01-15 12:55:55 +08:00
parent 71f2972e3f
commit e4f20350fe
1 changed files with 3 additions and 2 deletions
--- a/py12306/user/job.py
+++ b/py12306/user/job.py
@@ -103,9 +103,10 @@ class UserJob:
        return self.last_heartbeat

    def set_last_heartbeat(self, time=None):
+        time = time if time != None else time_int()
        if Config().is_cluster_enabled():
-            return self.cluster.session.set(Cluster.KEY_USER_LAST_HEARTBEAT, time if time != None else time_int())
-        self.last_heartbeat = time_int()
+            return self.cluster.session.set(Cluster.KEY_USER_LAST_HEARTBEAT, time)
+        self.last_heartbeat = time

    # def init_cookies
    def is_first_time(self):