class BookmarkDatabase(PersistenceLayer):
def __init__(self):
self.table_name = 'bookmarks'
self.db = DatabaseManager('bookmarks.db')
self.db.create_table(self.table_name, {
'id': 'integer primary key autoincrement',
'title': 'text not null',
'url': 'text not null',
'notes': 'text',
'date_added': 'text not null'
})
def create(self, bookmark_data):
self.db.add(self.table_name, bookmark_data)
def list(self, order_by=None):
return self.db.select(self.table_name, order_by=order_by).fetchall()
def edit(self, bookmark_id, bookmark_data):
self.db.update(self.table_name, {'id': bookmark_id}, bookmark_data)
def delete(self, bookmark_id):
self.db.delete(self.table_name, {'id': bookmark_id})
class BookmarkDatabase(PersistenceLayer):
def __init__(self):
self.table_name = 'bookmarks'
self.db = DatabaseManager('bookmarks.db')
self.db.create_table(
'bookmarks', {
'id': 'INTEGER PRIMARY KEY AUTOINCREMENT',
'title': 'TEXT NOT NULL',
'url': 'TEXT NOT NULL',
'notes': 'TEXT',
'date_added': 'TEXT NOT NULL'
})
def create(self, bookmark_data):
self.db.add(self.table_name, bookmark_data)
def list(self, order_by=None):
return self.db.select(self.table_name, order_by=order_by).fetchall()
def delete(self, bookmark_id):
self.db.delete(self.table_name, {'id': bookmark_id})
def update(self, bookmark_id, bookmark_data):
self.db.update(self.table_name, {'id': bookmark_id}, bookmark_data)
def testadd(self):
tablename = "BookMarks"
data = Mock
expected = Mock
self.assertEqual(expected, DatabaseManager.add(self, tablename, data))
