X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/7debd6efcc5face8fa6c3bd4f5f317a1801d14ce..417d32f3718d7c254275b67154a6f529b71face1:/imago.py diff --git a/imago.py b/imago.py index 1e0cbba..4b87f0f 100755 --- a/imago.py +++ b/imago.py @@ -92,7 +92,7 @@ def main(): draw = ImageDraw.Draw(im_g) for l in grid[0] + grid[1]: draw.line(l, fill=(64, 255, 64), width=1) - do_something(im_g) + do_something(im_g, "grid", name="grid") board = intrsc.board(image, lines, show_all, do_something) @@ -120,12 +120,15 @@ class Imsave(): self.saving_dir = saving_dir self.saving_num = 0 - def save(self, image, title=''): - filename = self.saving_dir + "{0:0>2}".format(self.saving_num) + '.jpg' + def save(self, image, title='', name=None): + if name: + filename = self.saving_dir + name + '.jpg' + else: + filename = self.saving_dir + "{0:0>2}".format(self.saving_num) + '.jpg' + self.saving_num += 1 if not os.path.isdir(self.saving_dir): os.makedirs(self.saving_dir) image.save(filename, 'JPEG') - self.saving_num += 1 if __name__ == '__main__': try: