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')
