From: Lorin Hochstein <lorin.hochstein@sendgrid.com>
Date: Sat, 7 Feb 2015 02:19:31 +0000 (-0500)
Subject: add scripts dirctory
X-Git-Url: https://git.halfball.org/?a=commitdiff_plain;h=a3f68ecf3ab2ab26db53faa01e5e1ea07a8bb786;p=ansiblebook.git

add scripts dirctory
---

diff --git a/ch06/playbooks/scripts/setadmin.py b/ch06/playbooks/scripts/setadmin.py
new file mode 100644
index 0000000..7800302
--- /dev/null
+++ b/ch06/playbooks/scripts/setadmin.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# A script to set the admin credentials
+# Assumes two environment variables
+#
+# PROJECT_DIR: the project directory (e.g., ~/projname)
+# ADMIN_PASSWORD: admin user's password
+
+import os
+import sys
+
+# Add the project directory to system path
+proj_dir = os.path.expanduser(os.environ['PROJECT_DIR'])
+sys.path.append(proj_dir)
+
+os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'
+
+
+from mezzanine.utils.models import get_user_model
+User = get_user_model()
+u, _ = User.objects.get_or_create(username='admin')
+u.is_staff = u.is_superuser = True
+u.set_password(os.environ['ADMIN_PASSWORD'])
+u.save()
diff --git a/ch06/playbooks/scripts/setsite.py b/ch06/playbooks/scripts/setsite.py
new file mode 100644
index 0000000..9455624
--- /dev/null
+++ b/ch06/playbooks/scripts/setsite.py
@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+# A script to set the site domain
+# Assumes two environment variables
+#
+# PROJECT_DIR: the project directory (e.g., ~/projname)
+# WEBSITE_DOMAIN: the domain of the site (e.g., www.example.com)
+
+import os
+import sys
+
+# Add the project directory to system path
+proj_dir = os.path.expanduser(os.environ['PROJECT_DIR'])
+sys.path.append(proj_dir)
+
+os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'
+from django.conf import settings
+from django.contrib.sites.models import Site
+
+domain = os.environ['WEBSITE_DOMAIN']
+Site.objects.filter(id=settings.SITE_ID).update(domain=domain)
+Site.objects.get_or_create(domain=domain)
+