arti/doors
2
0
Fork 1
Code Issues 5 Pull Requests Activity

Teach DB to create schema in existing connection

Browse Source
This commit is contained in:
Arti Zirk 2020-10-08 20:43:48 +03:00
parent ea3d617e9c
commit 7878cadb30
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kdoorweb/kdoorweb/db.py
View File

@ -51,6 +51,9 @@ class DB:
@staticmethod @staticmethod
def create_db(dbfile): def create_db(dbfile):
if type(dbfile) is DB:
db = dbfile.db
else:
db = sqlite3.connect(dbfile) db = sqlite3.connect(dbfile)
db.executescript(""" db.executescript("""
create table versions ( create table versions (

6
kdoorweb/tests/test_db.py
View File

@ -4,9 +4,15 @@ from kdoorweb.db import DB
class TestDB(unittest.TestCase): class TestDB(unittest.TestCase):
def setUp(self) -> None:
self.db = DB()
def test_create_db_in_memory(self): def test_create_db_in_memory(self):
DB.create_db(dbfile=":memory:") DB.create_db(dbfile=":memory:")
def test_create_db_in_connection(self):
DB.create_db(dbfile=self.db)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()