Advertisement
xah

webtoon stitch v0.0.1

xah
Feb 27th, 2017 (edited)
153
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.18 KB | None | 0 0
  1. from PIL import Image
  2. import os
  3.  
  4. def getFolders(dir):
  5.     return [name for name in os.listdir(dir)
  6.             if os.path.isdir(os.path.join(dir, name))]
  7.  
  8. def getFiles(dir):
  9.     files = []
  10.     for c in os.listdir(dir):
  11.         _, ext = os.path.splitext(c)
  12.  
  13.         if ext == '.png':
  14.             files.append(c)
  15.  
  16.     return files
  17.  
  18. def merge_images(dir, file):
  19.     image = Image.open(os.path.join(dir, file))
  20.     base = Image.open(out_file)
  21.  
  22.     (width1, height1) = image.size
  23.     (width2, height2) = base.size
  24.  
  25.     result_width = max(width1, width2)
  26.     result_height = height1 + height2
  27.  
  28.     out = Image.new('RGB', (result_width, result_height))
  29.     out.paste(im=base, box=(0, 0))
  30.     out.paste(im=image, box=(0, height2))
  31.  
  32.     out.save(out_file,"PNG")
  33.     return out
  34.  
  35. def createTarget():
  36.     result = Image.new('RGB', (1, 1))
  37.     result.save(out_file,"PNG")
  38.    
  39. dir = '/Users/user/Documents/C5'
  40. dir1 = dir
  41. folder = getFolders(dir)
  42.  
  43. for i in range(len(folder)):
  44.     dir = dir1 + folder[i]
  45.     ff = getFiles(dir)
  46.     out_file = folder[i] + '.png'
  47.     createTarget()
  48.     for f in ff:
  49.         merge_images(dir, f)
  50.     print(folder[i] + ': Done.')
  51. print('All done!')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement