1 files changed, 63 insertions, 0 deletions
diff --git a/tasks.py b/tasks.py
new file mode 100644
index 0000000..07c252c
--- /dev/null
+++ b/tasks.py
@@ -0,0 +1,63 @@
+import invoke
+from invoke import task
+import os
+from yaksh.settings import SERVER_POOL_PORT
+
+
+SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
+TARGET_CONTAINER_NAME = 'yaksh_code_server'
+SRC_IMAGE_NAME = 'yaksh_image'
+
+def create_dir(path):
+    if not os.path.exists(path):
+            ctx.run("mkdir {0}".format(path))
+
+@task
+def setupdb(ctx):
+    print("** Setting up & migrating database **")
+    ctx.run("python manage.py migrate")
+
+@task(setupdb)
+def run(ctx):
+    print("** Running the Django web server **")
+    ctx.run("python manage.py runserver")
+
+@task
+def clean(ctx):
+    print("** Discarding database **")
+    ctx.run("rm -rf {0}".format(os.path.join(SCRIPT_DIR, 'db.sqlite3')))
+
+@task
+def getimage(ctx, image=SRC_IMAGE_NAME):
+    try:
+        result = ctx.run("sudo docker inspect {0}".format(image), hide=True)
+    except invoke.exceptions.Failure:
+        print("The docker image {0} does not exist locally".format(image))
+        print("\nPulling latest image <{0}> from docker hub".format(image))
+        ctx.run("sudo docker pull {0}".format(image))
+
+@task
+def runcodeserver(ctx, ports=SERVER_POOL_PORT, image=SRC_IMAGE_NAME, unsafe=False):
+    if unsafe:
+        with ctx.cd(SCRIPT_DIR):
+            ctx.run("sudo python -m yaksh.code_server")
+    else:
+        cmd_params = {'ports': ports,
+            'image': SRC_IMAGE_NAME,
+            'name': TARGET_CONTAINER_NAME,
+            'vol_mount_dest': '/src/online_test/',
+            'vol_mount_src': os.path.join(SCRIPT_DIR),
+            'command': 'sh /src/yaksh_script.sh',
+        }
+
+        getimage(ctx, image=SRC_IMAGE_NAME)
+
+        create_dir(os.path.join(SCRIPT_DIR, 'output/'))
+        create_dir(os.path.join(SCRIPT_DIR, 'yaksh/data/'))
+
+        ctx.run(
+            "sudo docker run --privileged \
+            -dp {ports}:{ports} --name={name} \
+            -v {vol_mount_src}:{vol_mount_dest} \
+            {image} {command}".format(**cmd_params)
+        )