1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
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
13 # http://www.apache.org/licenses/LICENSE-2.0
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
21 # For those usages not covered by the Apache License, Version 2.0 please
22 # contact: bdiaz@whitestack.com or glavado@whitestack.com
24 """Peewee migrations -- 001_initial.py.
26 Some examples (model - class or model name)::
28 > Model = migrator.orm['model_name'] # Return model in current state by name
30 > migrator.sql(sql) # Run custom SQL
31 > migrator.python(func, *args, **kwargs) # Run python code
32 > migrator.create_model(Model) # Create a model (could be used as decorator)
33 > migrator.remove_model(model, cascade=True) # Remove a model
34 > migrator.add_fields(model, **fields) # Add fields to a model
35 > migrator.change_fields(model, **fields) # Change fields
36 > migrator.remove_fields(model, *field_names, cascade=True)
37 > migrator.rename_field(model, old_field_name, new_field_name)
38 > migrator.rename_table(model, new_table_name)
39 > migrator.add_index(model, *col_names, unique=False)
40 > migrator.drop_index(model, *col_names)
41 > migrator.add_not_null(model, *field_names)
42 > migrator.drop_not_null(model, *field_names)
43 > migrator.add_default(model, field_name, default)
52 def migrate(migrator
, database
, fake
=False, **kwargs
):
53 """Write your migrations here."""
55 @migrator.create_model
56 class BaseModel(pw
.Model
):
60 table_name
= "basemodel"
62 @migrator.create_model
63 class ScalingGroup(pw
.Model
):
65 nsr_id
= pw
.CharField(max_length
=255)
66 vnf_member_index
= pw
.IntegerField()
67 name
= pw
.CharField(max_length
=255)
68 content
= pw
.TextField()
71 table_name
= "scalinggroup"
73 @migrator.create_model
74 class ScalingPolicy(pw
.Model
):
76 name
= pw
.CharField(max_length
=255)
77 cooldown_time
= pw
.IntegerField()
78 last_scale
= pw
.DateTimeField()
79 scaling_group
= pw
.ForeignKeyField(backref
='scaling_policies', column_name
='scaling_group_id', field
='id',
80 model
=migrator
.orm
['scalinggroup'], on_delete
='CASCADE')
83 table_name
= "scalingpolicy"
85 @migrator.create_model
86 class ScalingCriteria(pw
.Model
):
88 name
= pw
.CharField(max_length
=255)
89 scaling_policy
= pw
.ForeignKeyField(backref
='scaling_criterias', column_name
='scaling_policy_id', field
='id',
90 model
=migrator
.orm
['scalingpolicy'], on_delete
='CASCADE')
93 table_name
= "scalingcriteria"
95 @migrator.create_model
96 class ScalingAlarm(pw
.Model
):
98 alarm_uuid
= pw
.CharField(max_length
=255, unique
=True)
99 action
= pw
.CharField(max_length
=255)
100 vnf_member_index
= pw
.IntegerField()
101 vdu_name
= pw
.CharField(max_length
=255)
102 scaling_criteria
= pw
.ForeignKeyField(backref
='scaling_alarms', column_name
='scaling_criteria_id', field
='id',
103 model
=migrator
.orm
['scalingcriteria'], on_delete
='CASCADE')
106 table_name
= "scalingalarm"
109 def rollback(migrator
, database
, fake
=False, **kwargs
):
110 """Write your rollback migrations here."""
112 migrator
.remove_model('scalingalarm')
114 migrator
.remove_model('scalingcriteria')
116 migrator
.remove_model('scalingpolicy')
118 migrator
.remove_model('scalinggroup')
120 migrator
.remove_model('basemodel')