def __init__(self, database: Database, tname='Messages'):
"""params:
database: Database instance
tname: name of table to store information"""
self.database = database
self.tname = tname
self.primary_key = PrimaryKey.id_as_primary().name
self.fields = MsgWithTag.to_fields() # no auto-increment id
# create table
if tname not in database.get_all_tables_name():
database.create_table(tname, PrimaryKey.id_as_primary(),
self.fields)
def test_create():
"""create table"""
try:
os.remove(PATH)
except FileNotFoundError:
pass
database = Database(PATH)
primary_key = PrimaryKey.id_as_primary()
fields = [
Field("msg", "TEXT"),
Field("tags", "CHAR(50)"),
Field('time',
'CHAR(30)',
others=[f'DEFAULT (\'{str(datetime.now())}\')'])
]
database.create_table(TABLE, primary_key, fields)
assert TABLE in database.tables.keys()
database.close()