Add a python script to upgrade

src/model.py

@@ -310,8 +310,8 @@ class PortMapping(db.Model):
     def __init__(self, node_name, node_ip, node_port, host_port):
         self.node_name = node_name
         self.node_ip = node_ip
-        self.node_port = node_port
-        self.host_port = host_port
+        self.node_port = int(node_port)
+        self.host_port = int(host_port)
 
     def __repr__(self):
         return "{\"id\":\"%d\", \"node_name\":\"%s\", \"node_ip\": \"%s\", \"node_port\":\"%s\", \"host_port\":\"%s\"}" % (self.id, self.node_name, self.node_ip, self.node_port, self.host_port)

tools/upgrade_file2db.py

@@ -0,0 +1,77 @@
+import sys
+sys.path.append("../src/")
+import os,json,datetime,time
+from model import db, VCluster, Container, PortMapping, Image
+
+timeFormat = "%Y-%m-%d %H:%M:%S"
+dockletPath = "/opt/docklet/global"
+userdir = dockletPath + "/users/"
+
+try:
+    VCluster.query.all()
+except Exception as err:
+    print("Create database...")
+    db.create_all()
+
+print("Update vcluster...")
+for user in os.listdir(usersdir):
+    if not os.path.exists(usersdir+"clusters"):
+        continue
+    print("Update User: "+str(user))
+    clusterfiles = os.listdir(usersdir+"clusters")
+    for cluname in clusterfiles:
+        cluFile = open(cluname,"r")
+        cluinfo = json.loads(cluFile.read())
+        vcluster = VCluster(cluinfo['clusterid'],cluname,user,cluinfo['status'],cluinfo['size'],cluinfo['nextcid'],cluinfo['proxy_server_ip'],cluinfo['proxy_public_ip'])
+        vcluster.create_time = time.strptime(cluinfo['create_time'],timeFormat)
+        vcluster.start_time = cluinfo['start_time']
+        for coninfo in cluinfo['containers']:
+            lastsavet = time.strptime(coninfo['lastsave'],timeFormat)
+            con = Container(coninfo['containername'], coninfo['hostname'], coninfo['ip'], coninfo['host'], coninfo['image'], lastsavet, cluinfo['setting'])
+            vcluster.containers.append(con)
+        for pminfo in cluinfo['port_mapping']:
+            pm = PortMapping(pminfo['node_name'], pminfo['node_ip'], int(pminfo['node_port']), int(pminfo['host_port']))
+            vcluster.port_mapping.append(pm)
+        for nodename in cluinfo['billing_history'].keys():
+            bhinfo = cluinfo['billing_history'][nodename]
+            bh = BillingHistory(nodename,bhinfo['cpu'],bhinfo['mem'],bhinfo['disk'],bhinfo['port'])
+            vcluster.billing_history.append(bh)
+        try:
+            db.session.add(vcluster)
+            db.session.commit()
+        except Exception as err:
+            print(err)
+        cluFile.close()
+
+print("Update Images...")
+for shareStr in ['private/','public/']:
+    print("Update "+shareStr+" Images...")
+    for user in os.listdir(dockletPath+"/images/"+shareStr):
+        print("Update User: "+user)
+        tmppath = dockletPath+"/images/"+shareStr+user+"/"
+        files = os.listdir(tmppath)
+        images = []
+        for file in files:
+            if file[0] == "." or file[-3] != ".":
+                continue
+            images.append(file[:-3])
+        for img in images:
+            infofile = open("."+img+".info","r")
+            imginfo = infofile.read().split('\n')
+            infofile.close()
+            desfile = open("."+img+".description","r")
+            desinfo = desfile.read()
+            dbimage = Image.query.filter_by(imagename=img,ownername=user).first()
+            if dbimage is None:
+                dbimage = Image(img,False,False,user,desinfo)
+                dbimage.create_time = time.strptime(imginfo[0],timeFormat)
+            if shareStr == 'public/':
+                dbimage.hasPublic = True
+            else:
+                dbimage.hasPrivate = True
+            try:
+                db.session.add(Image)
+                db.session.commit()
+            except Exception as err:
+                print(err)
+print("Finished!")