osm_mon/core/database.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright 2018 Whitestack, LLC
   4 # *************************************************************
   5
   6 # This file is part of OSM Monitoring module
   7 # All Rights Reserved to Whitestack, LLC
   8
   9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
  10 # not use this file except in compliance with the License. You may obtain
  11 # a copy of the License at
  12
  13 #         http://www.apache.org/licenses/LICENSE-2.0
  14
  15 # Unless required by applicable law or agreed to in writing, software
  16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  18 # License for the specific language governing permissions and limitations
  19 # under the License.
  20
  21 # For those usages not covered by the Apache License, Version 2.0 please
  22 # contact: bdiaz@whitestack.com or glavado@whitestack.com
  23 ##
  24
  25 import logging
  26 import os
  27 from typing import Iterable
  28
  29 from peewee import CharField, FloatField, Model, AutoField, Proxy, ForeignKeyField
  30 from peewee_migrate import Router
  31 from playhouse.db_url import connect
  32
  33 from osm_mon import migrations
  34 from osm_mon.core.config import Config
  35
  36 log = logging.getLogger(__name__)
  37
  38 db = Proxy()
  39
  40
  41 class BaseModel(Model):
  42     id = AutoField(primary_key=True)
  43
  44     class Meta:
  45         database = db
  46
  47
  48 class Alarm(BaseModel):
  49     uuid = CharField(unique=True)
  50     name = CharField()
  51     severity = CharField()
  52     threshold = FloatField()
  53     operation = CharField()
  54     statistic = CharField()
  55     metric = CharField()
  56
  57
  58 class AlarmTag(BaseModel):
  59     name = CharField()
  60     value = CharField()
  61     alarm = ForeignKeyField(Alarm, related_name='tags', on_delete='CASCADE')
  62
  63
  64 class DatabaseManager:
  65     def __init__(self, config: Config):
  66         db.initialize(connect(config.get('sql', 'database_uri')))
  67
  68     def create_tables(self) -> None:
  69         db.connect()
  70         with db.atomic():
  71             router = Router(db, os.path.dirname(migrations.__file__))
  72             router.run()
  73         db.close()
  74
  75
  76 class AlarmTagRepository:
  77     @staticmethod
  78     def create(**query) -> Alarm:
  79         return AlarmTag.create(**query)
  80
  81
  82 class AlarmRepository:
  83     @staticmethod
  84     def create(**query) -> Alarm:
  85         return Alarm.create(**query)
  86
  87     @staticmethod
  88     def get(*expressions) -> Alarm:
  89         return Alarm.select().where(*expressions).get()
  90
  91     @staticmethod
  92     def list(*expressions) -> Iterable[Alarm]:
  93         if expressions == ():
  94             return Alarm.select()
  95         else:
  96             return Alarm.select().where(*expressions)