|
@@ -3,8 +3,7 @@ import os
|
|
|
import base64, datetime
|
|
|
import tomlkit, sqlite3
|
|
|
|
|
|
-# kw == Key Word
|
|
|
-# dt_now == date&&time in UTC
|
|
|
+# kw == 'k'ey 'w'ord; dt_now = date&&time in UTC
|
|
|
def generate_uid(kw):
|
|
|
dt_now = str(datetime.datetime.now(datetime.timezone.utc))
|
|
|
genText = str.encode(kw + "_" + dt_now)
|
|
@@ -21,7 +20,7 @@ def getTomlDoc(tomlName):
|
|
|
try:
|
|
|
with open(tomlName, "rb") as t:
|
|
|
doc = tomlkit.load(t)
|
|
|
-
|
|
|
+
|
|
|
if doc == {}:
|
|
|
print("Error 0: Could not find correct 'config.toml'")
|
|
|
exit()
|
|
@@ -33,10 +32,39 @@ def getTomlDoc(tomlName):
|
|
|
exit()
|
|
|
|
|
|
|
|
|
-def matchTomlKey(tomlName, key):
|
|
|
+def matchTomlKey(tomlName, key, table=None) -> str:
|
|
|
doc = getTomlDoc(tomlName)
|
|
|
- return doc.item(key)
|
|
|
+
|
|
|
+ if table == None:
|
|
|
+ return str(doc.item(key))
|
|
|
|
|
|
+ elif table != None:
|
|
|
+ d = doc.unwrap()
|
|
|
+ return str(d[table][key])
|
|
|
+
|
|
|
+
|
|
|
+def matchTomlKeys(tomlName, keys, table=None) -> list:
|
|
|
+ doc = getTomlDoc(tomlName)
|
|
|
+
|
|
|
+ if table == None:
|
|
|
+ # rl == 'r'eturn 'l'ist
|
|
|
+ rl = []
|
|
|
+
|
|
|
+ for key in keys:
|
|
|
+ rl.append(doc.item(key))
|
|
|
+
|
|
|
+ return rl
|
|
|
+
|
|
|
+ elif table != None:
|
|
|
+ rl = []
|
|
|
+ d = doc.unwrap()
|
|
|
+
|
|
|
+ for key in keys:
|
|
|
+ rl.append(d[table][key])
|
|
|
+
|
|
|
+ return rl
|
|
|
+
|
|
|
+
|
|
|
|
|
|
# Sqlite3
|
|
|
|