def sys_log(cls, string): if EnvUtil.get_pslx_env_variable(var='PSLX_LOG'): try: caller = getframeinfo(stack()[1][0]) print('[SYS-LOG] ' + ColorsUtil.make_foreground_green( '[file: %s]' % FileUtil.base_name(caller.filename)) + ' ' + ColorsUtil.make_foreground_yellow('[line: %d]' % caller.lineno) + ' ' + ColorsUtil.make_foreground_red( '[%s]' % str(TimezoneUtil.cur_time_in_pst())) + ': ' + string) except Exception as _: print('[SYS-LOG] ' + ColorsUtil.make_foreground_red( '[%s]' % str(TimezoneUtil.cur_time_in_pst())) + ': ' + string)
def initialize_from_file(self, file_name): if '.pb' not in file_name: self.sys_log("Please use .pb extension for proto files.") self._file_name = FileUtil.create_file_if_not_exist(file_name=file_name) if FileUtil.is_file_empty(file_name=self._file_name): self._table_message = ProtoTable() else: with FileLockTool(self._file_name, read_mode=True): self._table_message = FileUtil.read_proto_from_file( proto_type=ProtoTable, file_name=self._file_name ) if not self._table_message.table_path: self._table_message.table_path = self._file_name if not self._table_message.table_name: self._table_message.table_name = FileUtil.base_name(file_name=file_name) if not self._table_message.created_time: self._table_message.created_time = str(TimezoneUtil.cur_time_in_pst())
def initialize_from_file(self, file_name): if '.pb' not in file_name: self._SYS_LOGGER.warning( "Please use .pb extension for proto files.") self._file_name = FileUtil.normalize_file_name(file_name=file_name) self._table_message = FileUtil.read_proto_from_file( proto_type=ProtoTable, file_name=self._file_name) self.increment_rpc_count_by(n=1) if self._table_message is None: self._table_message = ProtoTable() if not self._table_message.table_path: self._table_message.table_path = self._file_name if not self._table_message.table_name: self._table_message.table_name = FileUtil.base_name( file_name=file_name) if not self._table_message.created_time: self._table_message.created_time = str( TimezoneUtil.cur_time_in_pst())
def _file_to_shard(self, file): base_name = FileUtil.base_name(file) return int(base_name.replace('.pb', '').split('@')[1])
def test_base_name(self): file_name = 'foo/bar/file.txt' self.assertEqual(FileUtil.base_name(file_name), 'file.txt')