1 files changed, 44 insertions, 0 deletions
diff --git a/django_pyscss/extension/django.py b/django_pyscss/extension/django.py
new file mode 100644
index 0000000..c127270
--- /dev/null
+++ b/django_pyscss/extension/django.py
@@ -0,0 +1,44 @@
+from __future__ import absolute_import, unicode_literals
+
+from itertools import product
+from pathlib import PurePath
+
+from scss.extension.core import CoreExtension
+from scss.source import SourceFile
+
+from ..utils import get_file_and_storage
+
+
+class DjangoExtension(CoreExtension):
+    name = 'django'
+
+    def handle_import(self, name, compilation, rule):
+        """
+        Re-implementation of the core Sass import mechanism, which looks for
+        files using the staticfiles storage and staticfiles finders.
+        """
+        original_path = PurePath(name)
+
+        if original_path.suffix:
+            search_exts = [original_path.suffix]
+        else:
+            search_exts = compilation.compiler.dynamic_extensions
+
+        if original_path.is_absolute():
+            # Remove the beginning slash
+            search_path = original_path.relative_to('/').parent
+        elif rule.source_file.origin:
+            search_path = rule.source_file.origin
+        else:
+            search_path = original_path.parent
+
+        basename = original_path.stem
+
+        for prefix, suffix in product(('_', ''), search_exts):
+            filename = PurePath(prefix + basename + suffix)
+
+            full_filename, storage = get_file_and_storage(str(search_path / filename))
+
+            if full_filename:
+                with storage.open(full_filename) as f:
+                    return SourceFile.from_file(f, origin=search_path, relpath=filename)