1 files changed, 27 insertions, 4 deletions
diff --git a/master/setup-master.py b/master/setup-master.py
index 896f680..3b2f3b1 100755
--- a/master/setup-master.py
+++ b/master/setup-master.py
@@ -66,11 +66,7 @@ def install_master():
 
 def syncdb_master():
     """Initialize the database"""
-    import settings
-    from django.core.management import setup_environ
-    setup_environ(settings)
     from db_defaults import providers, archs, stages, releases, mirrors
-    from master.models import StageProvider, Arch, Stage, Release, Mirror
     import copy
 
     management.call_command('syncdb')
@@ -111,6 +107,26 @@ def syncdb_master():
             serverobj.prefix = server[1]
             serverobj.save()
 
+def setup_sample_job():
+    from sample_data import sample_job
+    job = Job()
+    for i in ['name', 'jobtagerev', 'atoms']:
+        setattr(job, i, sample_job[i])
+    maintainers = User.objects.filter(username=sample_job['maintainer'][0])
+    if maintainers:
+        job.maintainer = maintainers[0]
+    else:
+        job.maintainer = User.objects.create_user(*sample_job['maintainer'])
+    job.provider = StageProvider.objects.get(name=sample_job['provider'])
+    job.stage = Stage.objects.get(name=sample_job['stage'], 
+                                  provider=job.provider)
+    job.arch = Arch.objects.get(specific=sample_job['arch'],
+                                provider=job.provider)
+    job.release = Release.objects.get(name=sample_job['release'],
+                                      provider=job.provider)
+    print job.atoms
+    job.save()
+
 if len(sys.argv) < 3:
     print_help()
     sys.exit(1)
@@ -134,7 +150,14 @@ if sys.argv[1] == 'install':
 Now you need to edit the database settings in %(dest)s/settings.py
 and run `./setup-master.py syncdb %(dest)s`""" % { 'dest': os.path.join(sys.argv[2], DESTDIR) }
 elif sys.argv[1] == 'syncdb':
+    # Import stuff
+    import settings
+    from django.core.management import setup_environ
+    setup_environ(settings)
+    from master.models import User, StageProvider, Arch, Stage, Release, Mirror, Job
+    # Start stuff
     syncdb_master()
+    setup_sample_job()
     print "All done! Now you can start the master with `python manage.py runserver`"
 else:
     print_help()