12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- import tomlkit
- # ----- Toml Methods -----
- def GetTomlDoc(tomlName:str):
- try:
- with open(tomlName, "rb") as t:
- doc = tomlkit.load(t)
- if doc == {}:
- input("error 0: could not found correct config file")
- exit()
- return doc
-
- except FileNotFoundError:
- input("error 0: could not found correct config file")
- # sec_respond?
- exit()
- def MatchTomlKey(tomlName:str, key:str, table:str="str") -> str:
- doc = GetTomlDoc(tomlName)
- if table == "str":
- return str(doc.item(key))
-
- elif table == "list":
- return list(doc.item(key))
-
- elif table != "table":
- d = doc.unwrap()
- return str(d[table][key])
- reg = MatchTomlKey("config.toml", "registered", "list")
- weekly = MatchTomlKey("config.toml", "weekly", "str").split("\r\n")
- weekly = weekly[2:-1]
- bouns = 0
- window = []
- for name in weekly:
- window = name.split(" ")
- try:
- if int(window[2]) >= 3:
- bouns = bouns + 1
- except:
- pass
- reg.remove(window[1])
- print("未打卡名单(%s): " %(len(reg)), reg)
- print("红包数量: ", bouns)
|