1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
import os
import re
import uuid
import time
import dlib
import simplejson as json
import numpy as np
from flask import Blueprint, request, jsonify
from PIL import Image, ImageOps # todo: try to remove PIL dependency
from celery.result import AsyncResult
from app.server.tasks import celery
from app.server.tasks import task_lookup, list_active_tasks
# from app.models.sql_factory import load_sql_datasets, list_datasets, get_dataset, get_table
api_task = Blueprint('task', __name__)
@api_task.route('/')
def index():
"""List active tasks"""
return jsonify(list_active_tasks)
@api_task.route('/status/<task_name>/<task_id>')
def task_status(task_name, task_id):
"""Return celery image processing status"""
if task_name in task_lookup:
task = task_lookup[task_name]['task'].AsyncResult(task_id)
# task = AsyncResult(task_id, app=celery)
else:
return jsonify({
'state': 'error',
'percent': 100,
'message': 'Unknown task'
})
# app.logger.info('task state: {}'.format(task.state))
if task.state == 'PENDING':
response = {
'state': task.state,
'percent': 0,
'message': 'Pending...'
}
elif task.state != 'FAILURE':
response = {
'state': task.state,
'percent': task.info.get('percent', 0),
'uuid': task.info.get('uuid', 0),
'message': task.info.get('message', '')
}
if 'result' in task.info:
response['result'] = task.info['result']
else:
# something went wrong in the background job
response = {
'state': task.state,
'percent': 100,
'message': str(task.info), # this is the exception raised
}
return jsonify(response)
@api_task.route('/upload/sleep', methods=['GET', 'POST'])
def sleep_test():
"""
Test the Celery system using a task that sleeps.
"""
async_task = task_lookup['sleep']['task'].apply_async(args=['sleep_test'])
task_url = '/task/{}/{}'.format('sleep', async_task.id)
return jsonify({
'result': True,
'task_url': task_url,
})
@api_task.route('/upload/:style', methods=['POST'])
def upload(style):
"""
Process a images in a particular style
"""
print('style: {}'.format(style))
if style in task_lookup:
task = task_lookup[style]['task']
print('task',task)
else:
return jsonify({
'result': False,
'error': 'Unknown task',
})
file = request.files['user_image']
ext = request.form['ext']
if ext is None:
ext = request.files['ext']
uuid_name = str(uuid.uuid4())
app.logger.info('[+] style: {}'.format(style))
app.logger.info('[+] ext: {}'.format(ext))
app.logger.info('[+] uuid_name: {}'.format(uuid_name))
# convert PNG to JPG
print('[+] Resizing image')
im = Image.open(file.stream).convert('RGB')
im = ImageOps.fit(im, (256, 256))
# # Save image to disk
# print('[+] Save image to {}'.format(fpath))
# im.save(fpath, 'JPEG', quality=100)
# im_pil_256 = im.resize((256,256))
# print('[+] ensure_np...')
# im_np = imx.ensure_np(im_pil_256)
celery_result = {
im: im,
}
print('[+] Start celery')
async_task = task.apply_async(args=[uuid_name, celery_result])
task_url = '/task/{}/{}'.format(style, async_task.id)
return jsonify({
'result': True,
'taskURL': task_url,
'uuid': uuid_name
})
|
