123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- import Alt
- import tomlkit, sqlite3
- # uid related
- # ----- Toml Methods -----
- def GetTomlDoc(tomlName):
- 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:
- input("error 0: could not found correct config file")
- exit()
-
- def MatchTomlKey(tomlName, key, table=None) -> str:
- doc = GetTomlDoc(tomlName)
- if table == None:
- return str(doc.item(key))
-
- elif table != None:
- d = doc.unwrap()
- return str(d[table][key])
- # no differernt between ↑ MatchTomlKey() ↑ except receives and returns in list
- def MatchTomlKeys(tomlName, keys, table=None) -> list:
- doc = GetTomlDoc(tomlName)
- if table == None:
- # rl == 'r'eturn 'l'ist
- rl = []
- for key in keys:
- try:
- rl.append(doc.item(key))
- except:
- pass
- return rl
-
- elif table != None:
- rl = []
- d = doc.unwrap()
- for key in keys:
- try:
- rl.append(d[table][key])
- except:
- pass
- return rl
-
- # ----- Sqlite Methods -----
- def Exec_one(dbPath, commands):
- con = sqlite3.connect(dbPath)
- cur = con.cursor()
- cur.execute(commands)
- con.commit()
- re = cur.fetchall()
- con.close()
- return re
- def IsExist(exec_commands, returnBool=True):
- tableName = str(exec_commands[1]).capitalize()
- ItemName = str(exec_commands[2])
- sqls = "SELECT name FROM {table} WHERE name='{name}'".format(table=tableName, name=ItemName)
- ie = Exec_one(dbPath, sqls)
- if ie != [] and returnBool == False:
- return ie
-
- elif ie != [] and returnBool == True:
- return True
- elif ie == []:
- return False
-
- else:
- # Alt.Err(errCode)
- print("err <Code>: unexpected error in existence check")
- class command():
- def __init__(self, dbPath, tableName, columnName, newColumnName) -> None:
- self.dp = dbPath
- self.table = tableName
- self.name = columnName
- self.newName = newColumnName
- def select(self, aliveOnly=True):
- if aliveOnly == True:
- sqls = "SELECT name FROM {table} WHERE name='{name}' AND status='alive';".format(table=self.table, name=self.name)
- elif aliveOnly == False:
- sqls = "SELECT name FROM {table} WHERE name='{name}';".format(table=self.table, name=self.name)
- res = Exec_one(self.dp, sqls)
- # 也许需要一个后处理res = [('a'),('b')]这样的格式问题
- return res
- def add(self, values, addObj="board", addType="new"):
- # insert
- # add board: insert into Board
- # add class(className, usingBoard):
- # add event(className)
- pass
- def delete(self):
- sqls = "UPDATE {table} SET status='deleted' WHERE name='{name}';".format(table=self.table, name=self.name)
- res = Exec_one(self.dp, sqls)
- def edit(self):
- # edit boardName
- # edit className
- # edit eventName
- # edit event dscrp
- # edit event blabla
- pass
-
- def move(self, moveObj):
- if moveObj == "KB":
- pass
-
- elif moveObj == "CL":
- pass
-
- elif moveObj == "EV":
- pass
- def backPrevious(self):
- pass
- def backHome(self):
- sqls = "SELECT name FROM sqlite_master WHERE type='table' AND name is NOT 'sqlite_sequence';"
- Exec_one(dbPath, sqls)
- def back(self, ):
- pass
- class objBoard():
- pass
- class objClass():
- pass
- class objEvent():
- pass
- def Handler():
- pass
- if __name__ == "__main__":
- dbPath = "dev.db"
|