Merge "Upgrade db version to add cpu column for function"

qinling/db/sqlalchemy/migration/alembic_migrations/versions/003_add_cpu_column.py

@@ -0,0 +1,44 @@
+# Copyright 2018 OpenStack Foundation.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License");
+#    you may not use this file except in compliance with the License.
+#    You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS,
+#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#    See the License for the specific language governing permissions and
+#    limitations under the License.
+
+"""add cpu column for functions
+
+Revision ID: 003
+Revises: 002
+Create Date: 2018-04-19 06:03:46.687706
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '003'
+down_revision = '002'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        'functions',
+        sa.Column('cpu', sa.Integer(), nullable=False)
+    )
+
+    op.alter_column(
+        'functions',
+        'memory_size',
+        existing_type=sa.Integer(),
+        nullable=False
+    )
+    # ### end Alembic commands ###

qinling/db/sqlalchemy/models.py

@@ -42,7 +42,8 @@ class Function(model_base.QinlingSecureModelBase):
     runtime = relationship(
         'Runtime', back_populates="functions", innerjoin=True, lazy='select'
     )
-    memory_size = sa.Column(sa.Integer)
+    cpu = sa.Column(sa.Integer, default=0)
+    memory_size = sa.Column(sa.Integer, default=0)
     timeout = sa.Column(sa.Integer)
     code = sa.Column(st.JsonLongDictType(), nullable=False)
     entry = sa.Column(sa.String(80), nullable=True)