def refresh_data(self): if Config.is_cluster_enabled(): self.data = {**self.data, **self.refresh_data_of_cluster()} else: self.data['query_count'] += 1 self.data['last_time'] = str(datetime.datetime.now()) self.save_data()
def init_data(cls): self = cls() # 获取上次记录 # if Const.IS_TEST: return result = False if not Config.is_cluster_enabled() and path.exists(self.data_path): with open(self.data_path, encoding='utf-8') as f: result = f.read() if result: result = json.loads(result) if Config.is_cluster_enabled(): result = self.get_data_from_cluster() if result: self.data = {**self.data, **result} self.print_data_restored()
def init_data(cls): self = cls() # 获取上次记录 result = False if not Config.is_cluster_enabled() and path.exists(self.data_path): with open(self.data_path, encoding='utf-8') as f: result = f.read() try: result = json.loads(result) except json.JSONDecodeError as e: result = {} # self.add_quick_log('加载status.json失败, 文件内容为: {}.'.format(repr(result))) # self.flush() # 这里可以用不用提示 if Config.is_cluster_enabled(): result = self.get_data_from_cluster() if result: self.data = {**self.data, **result} self.print_data_restored()
def handler_exit(self, *args, **kwargs): """ 程序退出 :param args: :param kwargs: :return: """ if Config.is_cluster_enabled(): from py12306.cluster.cluster import Cluster Cluster().left_cluster() sys.exit()
def __init__(self, *args): if Config.is_cluster_enabled(): args = { 'host': Config().REDIS_HOST, 'port': Config().REDIS_PORT, 'db': 0, 'password': Config().REDIS_PASSWORD, 'decode_responses': True } super().__init__(**args) RedisLog.add_quick_log(RedisLog.MESSAGE_REDIS_INIT_SUCCESS) else: super().__init__(**args) return self
def __init__(self, *args): if Config.is_cluster_enabled(): self.session = Redis() return self
def init_class(self): from py12306.cluster.cluster import Cluster if Config.is_cluster_enabled(): Cluster().run()