added builder function for private galleries

2 files changed, 34 insertions, 1 deletions

diff --git a/app/photos/build.py b/app/photos/build.py
new file mode 100644
index 0000000..15a649e
--- /dev/null
+++ b/app/photos/build.py
@@ -0,0 +1,30 @@
+import os
+from django.core.urlresolvers import reverse
+from builder.base import BuildNew
+
+
+class BuildLuxPhotos(BuildNew):
+
+    def build(self):
+        self.build_detail_view()
+
+    def get_model_queryset(self):
+        return self.model.objects.all()
+
+    def build_detail_view(self):
+        '''
+        write out all the expenses for each trip
+        '''
+        for obj in self.get_model_queryset():
+            url = obj.get_absolute_url()
+            path, slug = os.path.split(url)
+            path = '%s/' % path
+            # write html
+            response = self.client.get(url)
+            print(path, slug)
+            self.write_file(path, response.content, filename=slug)
+
+
+def builder():
+    j = BuildLuxPhotos("photos", "LuxGallery")
+    j.build()
diff --git a/app/photos/models.py b/app/photos/models.py
index ada9069..c2d77e4 100644
--- a/app/photos/models.py
+++ b/app/photos/models.py
@@ -99,7 +99,10 @@ class LuxGallery(models.Model):
         return "%sgallery_thumbs/%s.jpg" % (settings.IMAGES_URL, self.id)
 
     def get_absolute_url(self):
-        return "/photos/galleries/%s/" % (self.slug)
+        if self.is_public:
+            return "/photos/galleries/%s" % (self.slug)
+        else:
+            return "/photos/galleries/private/%s" % (self.slug)
 
 
 class Photo(models.Model):