Fix tests.

[redakcja.git] / apps / catalogue / helpers.py

diff --git a/apps/catalogue/helpers.py b/apps/catalogue/helpers.py
index a97a040..d340b46 100644 (file)
--- a/apps/catalogue/helpers.py
+++ b/apps/catalogue/helpers.py
@@ -46,8 +46,6 @@ def parse_isodate(isodate):
 
 class GalleryMerger(object):
     def __init__(self, dest_gallery, src_gallery):
 
 class GalleryMerger(object):
     def __init__(self, dest_gallery, src_gallery):

-        assert isinstance(dest_gallery, str)
-        assert isinstance(src_gallery, str)

         self.dest = dest_gallery
         self.src = src_gallery
         self.dest_size = None
         self.dest = dest_gallery
         self.src = src_gallery
         self.dest_size = None


@@ -70,6 +68,11 @@ class GalleryMerger(object):
         m = not always and re.match(r"^([0-9])-", name)
         return "%1d-%s" % (prefix, m and name[2:] or name)
 
         m = not always and re.match(r"^([0-9])-", name)
         return "%1d-%s" % (prefix, m and name[2:] or name)
 

+    @property
+    def was_merged(self):
+        "Check if we have gallery size recorded"
+        return self.dest_size is not None
+

     def merge(self):
         if not self.dest:
             return self.src
     def merge(self):
         if not self.dest:
             return self.src


@@ -77,8 +80,10 @@ class GalleryMerger(object):
             return self.dest
 
         files = listdir(self.path(self.dest))
             return self.dest
 
         files = listdir(self.path(self.dest))

+        files.sort()

         self.dest_size = len(files)
         files_other = listdir(self.path(self.src))
         self.dest_size = len(files)
         files_other = listdir(self.path(self.src))

+        files_other.sort()

         self.src_size = len(files_other)
 
         if files and files_other:
         self.src_size = len(files_other)
 
         if files and files_other:


@@ -132,8 +137,6 @@ class GalleryMerger(object):
                 renamed_files_other[f] = self.set_prefix(f, 1, True)
 
         # finally, move / rename files.
                 renamed_files_other[f] = self.set_prefix(f, 1, True)
 
         # finally, move / rename files.

-        from nose.tools import set_trace
-        #        set_trace()

         for frm, to in renamed_files.items():
             move(join(self.path(self.dest), frm),
                         join(self.path(self.dest), to))
         for frm, to in renamed_files.items():
             move(join(self.path(self.dest), frm),
                         join(self.path(self.dest), to))