Merge branch 'add-39-change-value-login-timeout' into 'master'

Add 39 change value login timeout

See merge request !69

tower/migrations/039_change_login_session_timeout.py

+
+# Third-party modules
+import yaml
+import json
+
+from peewee import (Model, CharField, TextField, ForeignKeyField)
+
+
+def migrate(migrator):
+    class Environment(Model):
+        class Meta(object):
+            database = migrator.db
+            db_table = "environment"
+
+        name = CharField(unique=True)
+
+    class Node(Model):
+        class Meta(object):
+            database = migrator.db
+            db_table = "node"
+
+        name = CharField()
+        environment = ForeignKeyField(Environment, on_delete="RESTRICT")
+
+    class Pool(Model):
+        class Meta(object):
+            database = migrator.db
+            db_table = "pool"
+
+        environment = ForeignKeyField(Environment, on_delete="RESTRICT")
+        name = CharField()
+
+    class Service(Model):
+        class Meta(object):
+            database = migrator.db
+            db_table = "service"
+
+        environment = ForeignKeyField(Environment, on_delete="RESTRICT")
+        service = CharField()
+        config = TextField()
+        pool = ForeignKeyField(Pool, null=True)
+        node = ForeignKeyField(Node)
+
+    if len(Environment.select()) != 0:
+        for env in Environment.select():
+            print("Migrating %s" % env.name)
+            for s in Service.select().where(Service.environment == env.id):
+                if s.service == "login":
+                    conf = yaml.full_load(s.config)
+                    if "session_ttl" in conf:
+                        if "d" not in str(conf["session_ttl"]):
+                            conf["session_ttl"] = str(conf["session_ttl"]) + "d"
+                            s.config = json.dumps(conf, sort_keys=True)
+                            s.save()