Some headway on database stuff
This commit is contained in:
parent
a30961f884
commit
16dc75a57d
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
*__pycache__*
|
||||
0
test/__init__.py
Normal file
0
test/__init__.py
Normal file
29
test/db_interface_test.py
Normal file
29
test/db_interface_test.py
Normal file
@ -0,0 +1,29 @@
|
||||
from utilities.database.sqlite import SQLite
|
||||
from pprint import pprint
|
||||
|
||||
test_db = SQLite("testdb.db")
|
||||
|
||||
test_db.query("CREATE TABLE vegetables (veggie_id INTEGER PRIMARY KEY AUTOINCREMENT, veggie_name TEXT NOT NULL)")
|
||||
|
||||
test_db.insert("vegetables", ["veggie_name"], ["Green Beans"])
|
||||
test_db.insert("vegetables", ["veggie_name"], ["Peas"])
|
||||
test_db.insert("vegetables", ["veggie_name"], ["Corn"])
|
||||
test_db.insert("vegetables", ["veggie_name"], ["Stephen Hawking"])
|
||||
|
||||
pprint(test_db.select("vegetables"))
|
||||
|
||||
pprint(test_db.update("vegetables", ["veggie_name"], ["Snow Peas", 2], [
|
||||
{
|
||||
"name": "veggie_id",
|
||||
"compare": "="
|
||||
}
|
||||
]))
|
||||
|
||||
pprint(test_db.delete("vegetables", [3], [
|
||||
{
|
||||
"name": "veggie_id",
|
||||
"compare": ">="
|
||||
}
|
||||
]))
|
||||
|
||||
pprint(test_db.select("vegetables"))
|
||||
0
utilities/database/__init__.py
Normal file
0
utilities/database/__init__.py
Normal file
@ -35,13 +35,13 @@ class Database():
|
||||
|
||||
return self.query(query_string, values)[0]
|
||||
|
||||
def delete(self, table: str, where: list[dict]=None) -> int:
|
||||
def delete(self, table: str, values: list, where: list[dict]=None) -> int:
|
||||
query_string = "DELETE FROM {table}{where}".format(
|
||||
table = table,
|
||||
where = self.__generate_basic_where_clause(where) if not where is None else ""
|
||||
)
|
||||
|
||||
return self.query(query_string)[0]
|
||||
return self.query(query_string, values)[0]
|
||||
|
||||
def __generate_basic_where_clause(self, where: list[dict]):
|
||||
return " WHERE {clauses}".format(
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
from database import Database
|
||||
from .database import Database
|
||||
import sqlite3
|
||||
|
||||
class SQLite(Database):
|
||||
def __init__(self, db_name: str):
|
||||
super(self, Database).__init__()
|
||||
super(Database, self).__init__()
|
||||
|
||||
self.db = sqlite3.connect(db_name)
|
||||
self.db.autocommit = True
|
||||
|
||||
Loading…
Reference in New Issue
Block a user