Added optional robots.txt and favicon.ico baking. Fixes #23.

palewire · Feb 12, 2014 · 702b702 · 702b702
1 parent 9783bc5
commit 702b702
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 4 deletions.
diff --git a/bakery/management/commands/build.py b/bakery/management/commands/build.py
@@ -76,12 +76,26 @@ def handle(self, *args, **options):
                 interactive=False,
                 verbosity=0
             )
+            target_dir = os.path.join(self.build_dir, settings.STATIC_URL[1:])
             if os.path.exists(settings.STATIC_ROOT) and settings.STATIC_URL:
-                shutil.copytree(
-                    settings.STATIC_ROOT,
-                    os.path.join(self.build_dir, settings.STATIC_URL[1:])
+                shutil.copytree(settings.STATIC_ROOT, target_dir)
+            # If they exist in the static directory, copy the robots.txt
+            # and favicon.ico files down to the root so they will work
+            # on the live website.
+            robot_src = os.path.join(target_dir, 'robots.txt')
+            favicon_src = os.path.join(target_dir, 'favicon.ico')
+            if os.path.exists(robot_src):
+                shutil.copy(robot_src, os.path.join(
+                    settings.BUILD_DIR,
+                    'robots.txt'
+                    )
+                )
+            if os.path.exists(favicon_src):
+                shutil.copy(favicon_src, os.path.join(
+                    settings.BUILD_DIR,
+                    'favicon.ico',
+                    )
                 )
-
         # Build the media directory
         if not options.get("skip_media"):
             if self.verbosity > 1:

diff --git a/bakery/tests/__init__.py b/bakery/tests/__init__.py
@@ -117,6 +117,10 @@ def test_build_cmd(self):
             'build_dir': settings.BUILD_DIR,
         })
         call_command("build", 'bakery.tests.MockDetailView')
+        robots_path = os.path.join(settings.BUILD_DIR, 'robots.txt')
+        self.assertTrue(os.path.exists(robots_path))
+        favicon_path = os.path.join(settings.BUILD_DIR, 'favicon.ico')
+        self.assertTrue(os.path.exists(favicon_path))
         if django.VERSION >= (1, 5):
             self.assertRaises(
                 CommandError,

diff --git a/bakery/tests/static/favicon.ico b/bakery/tests/static/favicon.ico
diff --git a/bakery/tests/static/robots.txt b/bakery/tests/static/robots.txt
@@ -0,0 +1 @@
+Hello robots!