summaryrefslogtreecommitdiff
path: root/ATRI/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/utils.py')
-rw-r--r--ATRI/utils.py47
1 files changed, 47 insertions, 0 deletions
diff --git a/ATRI/utils.py b/ATRI/utils.py
new file mode 100644
index 0000000..b0f1b12
--- /dev/null
+++ b/ATRI/utils.py
@@ -0,0 +1,47 @@
+import yaml
+from datetime import datetime
+from pathlib import Path
+
+
+def loadYaml(file: Path) -> dict:
+ '''
+ 读取yaml文件
+ :return: dict
+ '''
+ with open(file, 'r', encoding='utf-8') as f:
+ data = yaml.safe_load(f)
+ return data
+
+
+def countList(lst: list, aim) -> int:
+ '''
+ 检查某列表中目标出现的次数
+ :return: int
+ '''
+ count = 0
+ for ele in lst:
+ if ele == aim:
+ count = count + 1
+ return count
+
+
+def delListAim(lst: list, aim) -> list:
+ '''
+ 删除某列表中所有目标元素
+ :return: list
+ '''
+ while aim in lst:
+ lst.remove(aim)
+ return lst
+
+
+def nowTime() -> float:
+ '''
+ 获取当前时间(小时)
+ :return: float
+ '''
+ now_ = datetime.now()
+ hour = now_.hour
+ minute = now_.minute
+ now = hour + minute / 60
+ return now \ No newline at end of file