forked from LiteyukiStudio/LiteyukiBot
102 lines
2.3 KiB
Python
102 lines
2.3 KiB
Python
import nonebot
|
|
|
|
|
|
def convert_duration(text: str, default) -> float:
|
|
"""
|
|
转换自然语言时间为秒数
|
|
Args:
|
|
text: 1d2h3m
|
|
default: 出错时返回
|
|
|
|
Returns:
|
|
float: 总秒数
|
|
"""
|
|
units = {
|
|
"d" : 86400,
|
|
"h" : 3600,
|
|
"m" : 60,
|
|
"s" : 1,
|
|
"ms": 0.001
|
|
}
|
|
|
|
duration = 0
|
|
current_number = ''
|
|
current_unit = ''
|
|
try:
|
|
for char in text:
|
|
if char.isdigit():
|
|
current_number += char
|
|
else:
|
|
if current_number:
|
|
duration += int(current_number) * units[current_unit]
|
|
current_number = ''
|
|
if char in units:
|
|
current_unit = char
|
|
else:
|
|
current_unit = ''
|
|
|
|
if current_number:
|
|
duration += int(current_number) * units[current_unit]
|
|
|
|
return duration
|
|
|
|
except BaseException as e:
|
|
nonebot.logger.info(f"convert_duration error: {e}")
|
|
return default
|
|
|
|
|
|
def convert_time_to_seconds(time_str):
|
|
"""转换自然语言时长为秒数
|
|
Args:
|
|
time_str: 1d2m3s
|
|
|
|
Returns:
|
|
|
|
"""
|
|
seconds = 0
|
|
current_number = ''
|
|
|
|
for char in time_str:
|
|
if char.isdigit() or char == '.':
|
|
current_number += char
|
|
elif char == 'd':
|
|
seconds += float(current_number) * 24 * 60 * 60
|
|
current_number = ''
|
|
elif char == 'h':
|
|
seconds += float(current_number) * 60 * 60
|
|
current_number = ''
|
|
elif char == 'm':
|
|
seconds += float(current_number) * 60
|
|
current_number = ''
|
|
elif char == 's':
|
|
seconds += float(current_number)
|
|
current_number = ''
|
|
|
|
return int(seconds)
|
|
|
|
|
|
def convert_seconds_to_time(seconds):
|
|
"""转换秒数为自然语言时长
|
|
Args:
|
|
seconds: 10000
|
|
|
|
Returns:
|
|
|
|
"""
|
|
d = seconds // (24 * 60 * 60)
|
|
h = (seconds % (24 * 60 * 60)) // (60 * 60)
|
|
m = (seconds % (60 * 60)) // 60
|
|
s = seconds % 60
|
|
|
|
# 若值为0则不显示
|
|
time_str = ''
|
|
if d:
|
|
time_str += f"{d}d"
|
|
if h:
|
|
time_str += f"{h}h"
|
|
if m:
|
|
time_str += f"{m}m"
|
|
if not time_str:
|
|
time_str = f"{s}s"
|
|
return time_str
|