diff options
Diffstat (limited to 'master/setup-master.py')
-rwxr-xr-x | master/setup-master.py | 31 |
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() |