From 5558e3e13f215e03b7789dd8f2354fe868475324 Mon Sep 17 00:00:00 2001
From: Stefano Alberto Russo <stefano.russo@gmail.com>
Date: Thu, 20 Feb 2020 15:35:01 +0100
Subject: [PATCH] Added testuser demo container as Jupyter Notebook and made is
 admin.

---
 .../management/commands/base_app_populate.py  | 28 +++++++++++++++----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/images/webapp/code/rosetta/base_app/management/commands/base_app_populate.py b/images/webapp/code/rosetta/base_app/management/commands/base_app_populate.py
index 16a9d65..746ee31 100644
--- a/images/webapp/code/rosetta/base_app/management/commands/base_app_populate.py
+++ b/images/webapp/code/rosetta/base_app/management/commands/base_app_populate.py
@@ -18,14 +18,22 @@ class Command(BaseCommand):
         
         # Testuser
         try:
-            User.objects.get(username='testuser')
+            testuser = User.objects.get(username='testuser')
             print('Not creating test user as it already exist')
+        
         except User.DoesNotExist:
             print('Creating test user with default password')
             testuser = User.objects.create_user('testuser', 'testuser@rosetta.platform', 'testpass')
+            print('Making testuser admin')
+            testuser.is_staff = True
+            testuser.is_admin=True
+            testuser.is_superuser=True
+            testuser.save() 
+            print('Creating testuser profile')
             Profile.objects.create(user=testuser, authtoken='129aac94-284a-4476-953c-ffa4349b4a50')
+            
 
-        # public containers
+        # Public containers
         public_containers = Container.objects.filter(user=None)
         if public_containers:
             print('Not creating public containers as they already exist')
@@ -47,9 +55,19 @@ class Command(BaseCommand):
                                      service_ports = '8590')
 
 
-
-
-
+        # Public containers
+        testuser_containers = Container.objects.filter(user=testuser)
+        if testuser_containers:
+            print('Not creating testuser containers as they already exist')
+        else:
+            print('Creating testuser containers...')
+            
+            # JuPyter
+            Container.objects.create(user          = testuser,
+                                     image         = 'jupyter/base-notebook',
+                                     type          = 'docker',
+                                     registry      = 'docker_hub',
+                                     service_ports = '8888')
 
 
 
-- 
GitLab