Ticket #7358: 003_create_default_site.patch

django/contrib/sites/management.py

@@ -6 +6 @@
 from django.db.models import signals
 from django.contrib.sites.models import Site
 from django.contrib.sites import models as site_app
+import socket
 
-def create_default_site(app, created_models, verbosity):
+def get_default_hostname():
+    try:
+        hostname = socket.gethostname()
+    except socket.error:
+        from django.conf import settings
+        hostname = getattr(settings.DEFAULT_SITE_NAME, 'localhost')
+    return hostname
+
+def create_default_site(app, created_models, verbosity=1):
     if Site in created_models:
-        if verbosity >= 2:
-            print "Creating example.com Site object"
-        s = Site(domain="example.com", name="example.com")
+        # Don't let script-kiddies ruin your efforts by
+        # hardcoding something as stupid as example.com here.
+        hostname = get_default_hostname()
+        if verbosity >= 1:
+            print "Creating default site: <%s>" % repr(hostname)
+        s = Site(domain=hostname, name=hostname)
         s.save()
 
 dispatcher.connect(create_default_site, sender=site_app, signal=signals.post_syncdb)