diff options
Diffstat (limited to 'old/server/app/main/utils.py')
| -rw-r--r-- | old/server/app/main/utils.py | 37 |
1 files changed, 0 insertions, 37 deletions
diff --git a/old/server/app/main/utils.py b/old/server/app/main/utils.py deleted file mode 100644 index 62d9c95f..00000000 --- a/old/server/app/main/utils.py +++ /dev/null @@ -1,37 +0,0 @@ -from flask import current_app as app -from PIL import Image -import numpy as np -import cv2 as cv -import os -from os.path import join - -def ensure_pil(im): - try: - im.verify() - return im - except: - return Image.fromarray(im.astype('uint8'), 'RGB') - -def ensure_np(im): - if type(im) == np.ndarray: - return im - return np.asarray(im, np.uint8) - -def get_recent_uploads(limit=10): - d_uploads = app.config['UPLOADS'] - d_renders = app.config['RENDERS'] - - # list all files in uploads dir - filenames = [s for s in os.listdir(d_uploads) - if os.path.isfile(os.path.join(d_uploads, s))] - # sort uploaded files by date - filenames.sort(key=lambda s: os.path.getmtime(os.path.join(d_uploads, s)),reverse=True) - basenames = [os.path.splitext(os.path.basename(f))[0] for f in filenames] - basenames = basenames[:limit] - filenames = [f for f in basenames if os.path.isfile(join(d_renders,'{}_dull.jpg'.format(f)))] - - # create list for uploads and renders - uploads = [join('/img/uploads',f) for f in filenames] - renders = [join('/img/renders','{}_dull'.format(f)) for f in filenames] - urls = [join('/d',f) for f in basenames] - return uploads, renders, urls |
