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()