X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_mon%2Fcore%2Fdatabase.py;fp=osm_mon%2Fcore%2Fdatabase.py;h=0886c2bf0bcf9693f5ee12d8a1a8c64cc836d1a8;hb=b2b43da9ec8230664b19479c457dd54711b8986a;hp=eca08e9c8df63bd26baa7e170fab422d0ce2188a;hpb=8721c5ce76869719af22fb238a56fec612f76289;p=osm%2FMON.git diff --git a/osm_mon/core/database.py b/osm_mon/core/database.py index eca08e9..0886c2b 100644 --- a/osm_mon/core/database.py +++ b/osm_mon/core/database.py @@ -23,11 +23,14 @@ ## import logging +import os import uuid from peewee import CharField, TextField, FloatField, Model, AutoField, Proxy +from peewee_migrate import Router from playhouse.db_url import connect +from osm_mon import migrations from osm_mon.core.config import Config log = logging.getLogger(__name__) @@ -72,7 +75,8 @@ class DatabaseManager: def create_tables(self) -> None: with db.atomic(): - db.create_tables([VimCredentials, Alarm]) + router = Router(db, os.path.dirname(migrations.__file__)) + router.run() def get_credentials(self, vim_uuid: str = None) -> VimCredentials: with db.atomic():