import os
from typing import Iterable
-from peewee import CharField, TextField, FloatField, Model, AutoField, Proxy
+from peewee import CharField, FloatField, Model, AutoField, Proxy
from peewee_migrate import Router
from playhouse.db_url import connect
database = db
-class VimCredentials(BaseModel):
- uuid = CharField(unique=True)
- name = CharField()
- type = CharField()
- url = CharField()
- user = CharField()
- password = CharField()
- tenant_name = CharField()
- config = TextField()
-
-
class Alarm(BaseModel):
uuid = CharField(unique=True)
name = CharField()
db.close()
-class VimCredentialsRepository:
- @staticmethod
- def upsert(**query) -> VimCredentials:
- vim_credentials = VimCredentials.get_or_none(**query)
- if vim_credentials:
- query.update({'id': vim_credentials.id})
- vim_id = VimCredentials.insert(**query).on_conflict_replace().execute()
- return VimCredentials.get(id=vim_id)
-
- @staticmethod
- def get(*expressions) -> VimCredentials:
- return VimCredentials.select().where(*expressions).get()
-
-
class AlarmRepository:
@staticmethod
def create(**query) -> Alarm: