X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/c4599df2f08c95fd317967746a3325c3b168acdd..8a3484bca2af09fad4f912d50c5af5601274d295:/apps/catalogue/helpers.py diff --git a/apps/catalogue/helpers.py b/apps/catalogue/helpers.py index a97a0400..d340b461 100644 --- 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): - assert isinstance(dest_gallery, str) - assert isinstance(src_gallery, str) 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) + @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 @@ -77,8 +80,10 @@ class GalleryMerger(object): return self.dest files = listdir(self.path(self.dest)) + files.sort() 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: @@ -132,8 +137,6 @@ class GalleryMerger(object): 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))