start rework with autodiscovery

scaffolding/__init__.py

@@ -1,129 +1,6 @@
 # -*- coding: utf-8 -*-
-import os
-import random
-import gc
-from scaffolding import library
-from library.lorem_ipsum import LOREM_IPSUM
-import urllib
-from django.core.files import File
 
-class Tube(object):
-    """ The base class for scaffolding objects.
-    """
-    def __init__(self, **kwargs):
-        self.count = kwargs.get('count')
-        self.cls = kwargs.get('cls')
-
-    def __iter__(self):
-        return self
-
-    def next(self):
-        raise NotImplementedError('You need to implement your own next method.')
-
-#---------- custom classes -----------------
-
-class Name(Tube):
-    """ Generates a random name. <gender> can be 'male', 'female', 'm' or 'f'.
-    """
-    def __init__(self, max_length=30, gender=None, **kwargs):
-        super(Name, self).__init__(**kwargs)
-        self.max_length = max_length
-        self.first_names = library.FirstNames(gender=gender)
-        self.last_names = library.LastNames()
-
-    def next(self):
-        return '%s %s'[:self.max_length] % (self.first_names.next(), self.last_names.next())
-
-
-class LoremIpsum(Tube):
-    """ Generates a Lorem Ipsum Text. The number of paragraphs is defined in paragraphs.
-    """
-    def __init__(self, paragraphs=7, max_length=None, text=LOREM_IPSUM, **kwargs):
-        super(LoremIpsum, self).__init__(**kwargs)
-        self.text = text
-        self.max_length = max_length
-        self.paragraphs = paragraphs
-        #  TODO: Loop paragraphs.
-        if self.paragraphs > len(self.text):
-            raise AttributeError('The Text %s only has %s paragraphs' %(text, len(text)))
-
-    def next(self):
-        if self.paragraphs < len(self.text):
-            late_start = len(self.text) - self.paragraphs - 1
-            start = random.randint(0, late_start)
-        else:
-            start = 0
-        text = u'\n\n'.join(self.text[start:(start+self.paragraphs)])
-        if self.max_length:
-            return text[:self.max_length]
-        return text
-
-
-class RandInt(Tube):
-    """ Generates a random integer between min and max """
-    def __init__(self, min, max, **kwargs):
-        super(RandInt, self).__init__(**kwargs)
-        self.min = min
-        self.max = max
-
-    def next(self):
-        return random.randint(self.min, self.max)
-
-
-class Contrib(object):
-    """ Crates a Custom Object. The backend class is the first parameter.
-        The backend class has to inherit from Tube.
-    """
-    def __init__(self, backend, **kwargs):
-        self.backend = backend(**kwargs)
-
-    def __iter__(self):
-        return self.backend
-
-    def next(self):
-        return self.backend.next()
-
-
-class AlwaysTrue(Tube):
-    """ Always returns True.
-    """
-    def next(self):
-        return True
-
-
-class AlwaysFalse(Tube):
-    """ Always returns False.
-    """
-    def next(self):
-        return False
-
-
-class RandomInternetImage(Tube):
-    """ Creates a random image for an ImageField using an internet source.
-    """
-    def __init__(self, backend, **kwargs):
-        super(RandomInternetImage, self).__init__(**kwargs)
-        self.backend = backend(**kwargs)
-
-    def next(self):
-        # returns a filename and File object, ready to be fed to the image.save() method.
-        url = self.backend.next()
-        temp_image = urllib.urlretrieve(url)
-        return os.path.basename(url), File(open(temp_image[0]))
-
-class ForeignKey(Tube):
-    """ Creates a foreign key assigning the queryset.
-    """
-    def __init__(self, queryset, chunksize=100, wrap=True, **kwargs):
-        super(ForeignKey, self).__init__(**kwargs)
-        self.queryset = queryset.order_by('pk')[:chunksize]
-        self.wrap = wrap
-        self.length = len(self.queryset)
-        self.i = 0
-
-    def next(self):
-        if self.i == (self.length-1):
-            self.i = 0
-            return self.queryset[0]
-        self.i += 1
-        return self.queryset[self.i]
+from tubes import Tube, Name, LoremIpsum, RandInt, Contrib, AlwaysTrue, \
+                AlwaysFalse, RandomInternetImage, ForeignKey
+
+#                http://charlesleifer.com/blog/looking-registration-patterns-django/

scaffolding/management/commands/scaffold.py

@@ -3,9 +3,35 @@
 from django.db import models
 from datetime import datetime
 from django.conf import settings
+import imp
+
 import logging
 logger = logging.getLogger(__name__)
 
+def generic_autodiscover(module_name):
+    """
+    I have copy/pasted this code too many times...Dynamically autodiscover a
+    particular module_name in a django project's INSTALLED_APPS directories,
+    a-la django admin's autodiscover() method.
+    
+    Usage:
+        generic_autodiscover('commands') <-- find all commands.py and load 'em
+    """
+
+    for app in settings.INSTALLED_APPS:
+        try:
+            import_module(app)
+            app_path = sys.modules[app].__path__
+        except AttributeError:
+            continue
+        try:
+            imp.find_module(module_name, app_path)
+        except ImportError:
+            continue
+        import_module('%s.%s' % (app, module_name))
+        app_path = sys.modules['%s.%s' % (app, module_name)]
+
+
 class Command(BaseCommand):
     args = '<app.model> <count>'
     help = 'Creates placeholder data for your models.'

scaffolding/tubes.py

@@ -0,0 +1,128 @@
+import os
+import random
+import gc
+from scaffolding import library
+from library.lorem_ipsum import LOREM_IPSUM
+import urllib
+from django.core.files import File
+
+class Tube(object):
+    """ The base class for scaffolding objects.
+    """
+    def __init__(self, **kwargs):
+        self.count = kwargs.get('count')
+        self.cls = kwargs.get('cls')
+
+    def __iter__(self):
+        return self
+
+    def next(self):
+        raise NotImplementedError('You need to implement your own next method.')
+
+#---------- custom classes -----------------
+
+class Name(Tube):
+    """ Generates a random name. <gender> can be 'male', 'female', 'm' or 'f'.
+    """
+    def __init__(self, max_length=30, gender=None, **kwargs):
+        super(Name, self).__init__(**kwargs)
+        self.max_length = max_length
+        self.first_names = library.FirstNames(gender=gender)
+        self.last_names = library.LastNames()
+
+    def next(self):
+        return '%s %s'[:self.max_length] % (self.first_names.next(), self.last_names.next())
+
+
+class LoremIpsum(Tube):
+    """ Generates a Lorem Ipsum Text. The number of paragraphs is defined in paragraphs.
+    """
+    def __init__(self, paragraphs=7, max_length=None, text=LOREM_IPSUM, **kwargs):
+        super(LoremIpsum, self).__init__(**kwargs)
+        self.text = text
+        self.max_length = max_length
+        self.paragraphs = paragraphs
+        #  TODO: Loop paragraphs.
+        if self.paragraphs > len(self.text):
+            raise AttributeError('The Text %s only has %s paragraphs' %(text, len(text)))
+
+    def next(self):
+        if self.paragraphs < len(self.text):
+            late_start = len(self.text) - self.paragraphs - 1
+            start = random.randint(0, late_start)
+        else:
+            start = 0
+        text = u'\n\n'.join(self.text[start:(start+self.paragraphs)])
+        if self.max_length:
+            return text[:self.max_length]
+        return text
+
+
+class RandInt(Tube):
+    """ Generates a random integer between min and max """
+    def __init__(self, min, max, **kwargs):
+        super(RandInt, self).__init__(**kwargs)
+        self.min = min
+        self.max = max
+
+    def next(self):
+        return random.randint(self.min, self.max)
+
+
+class Contrib(object):
+    """ Crates a Custom Object. The backend class is the first parameter.
+        The backend class has to inherit from Tube.
+    """
+    def __init__(self, backend, **kwargs):
+        self.backend = backend(**kwargs)
+
+    def __iter__(self):
+        return self.backend
+
+    def next(self):
+        return self.backend.next()
+
+
+class AlwaysTrue(Tube):
+    """ Always returns True.
+    """
+    def next(self):
+        return True
+
+
+class AlwaysFalse(Tube):
+    """ Always returns False.
+    """
+    def next(self):
+        return False
+
+
+class RandomInternetImage(Tube):
+    """ Creates a random image for an ImageField using an internet source.
+    """
+    def __init__(self, backend, **kwargs):
+        super(RandomInternetImage, self).__init__(**kwargs)
+        self.backend = backend(**kwargs)
+
+    def next(self):
+        # returns a filename and File object, ready to be fed to the image.save() method.
+        url = self.backend.next()
+        temp_image = urllib.urlretrieve(url)
+        return os.path.basename(url), File(open(temp_image[0]))
+
+class ForeignKey(Tube):
+    """ Creates a foreign key assigning the queryset.
+    """
+    def __init__(self, queryset, chunksize=100, wrap=True, **kwargs):
+        super(ForeignKey, self).__init__(**kwargs)
+        self.queryset = queryset.order_by('pk')[:chunksize]
+        self.wrap = wrap
+        self.length = len(self.queryset)
+        self.i = 0
+
+    def next(self):
+        if self.i == (self.length-1):
+            self.i = 0
+            return self.queryset[0]
+        self.i += 1
+        return self.queryset[self.i]