Add a ability to select image type of VM

migrations/versions/16740cdbdd7c_added_a_deleted_at_field_in_vm_table.py

@@ -0,0 +1,28 @@
+"""Added a deleted_at field in Vm table
+
+Revision ID: 16740cdbdd7c
+Revises: 
+Create Date: 2018-03-15 11:35:59.559663
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '16740cdbdd7c'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('vm', sa.Column('deleted_at', sa.DateTime(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('vm', 'deleted_at')
+    # ### end Alembic commands ###

migrations/versions/ac40d51a4114_.py

@@ -1,9 +1,7 @@
 """Add a new column deleted_at in table vm
-
 Revision ID: ac40d51a4114
 Revises: 45b003a9a66f
 Create Date: 2018-03-20 15:23:15.806539
-
 """
 from alembic import op
 import sqlalchemy as sa

softserve/lib.py

@@ -37,7 +37,7 @@ def wrap(*args, **kwargs):
 
 
 @celery.task()
-def create_node(counts, name, node_request, pubkey):
+def create_node(counts, name, node_request, pubkey, image):
     '''
     Create a node in the cloud provider
     '''
@@ -47,7 +47,7 @@ def create_node(counts, name, node_request, pubkey):
     nova = pyrax.cloudservers
 
     flavor = nova.flavors.find(name='1 GB General Purpose v1')
-    image = nova.images.find(name='CentOS 7 (PVHVM)')
+    image = nova.images.find(name=image)
     node_request = NodeRequest.query.get(node_request)
     keypair = nova.keypairs.create(name, pubkey)
     # create the nodes

softserve/model.py

@@ -30,6 +30,7 @@ class NodeRequest(db.Model):
     id = db.Column(db.Integer, primary_key=True, autoincrement=True)
     user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
     node_name = db.Column(db.String(100), nullable=False)
+    image_type = db.Column(db.String(20), nullable=False)
     node_counts = db.Column(db.Integer, nullable=False)
     hours = db.Column(db.Integer, nullable=False)
     pubkey = db.Column(db.VARCHAR(1024), nullable=False)

softserve/templates/form.html

@@ -21,6 +21,15 @@ <h3>Create Nodes</h3>
        <input type="text" class="form-control" name="node_name" data-toggle="tooltip" title='Should not contain any special characters except underscore' required>
     </div>
   </div>
+  <div class='row'>
+    <div class="form-group col-md-5">
+      <label for="image">Image type</label>
+      <select class="form-control" name="image">
+        <option value="CentOS 7 (PVHVM)">CentOS7</option>
+        <option value="centos7-devel">CentOS7-devel</option>
+      </select>
+    </div>
+  </div>
   <div class='row'>
     <div class="form-group col-md-5">
       <label for="hours">Number of hours</label>

softserve/views.py

@@ -84,6 +84,7 @@ def get_node_data():
         name = request.form['node_name']
         hours_ = request.form['hours']
         pubkey_ = request.form['pubkey']
+        image = request.form['image']
 
         # Validating the hours and node counts
         if int(counts) > 5 or int(hours_) > 4 or int(counts) > n:
@@ -107,12 +108,13 @@ def get_node_data():
                 node_request = NodeRequest(
                     user_id=g.user.id,
                     node_name=name,
+                    image_type=image,
                     node_counts=counts,
                     hours=hours_,
                     pubkey=pubkey_)
                 db.session.add(node_request)
                 db.session.commit()
-                create_node.delay(counts, name, node_request.id, pubkey_)
+                create_node.delay(counts, name, node_request.id, pubkey_, image)
                 flash('Creating your machine. Please wait for a moment.')
                 return redirect('/dashboard')
             else: