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
|
import os
import re
import uuid
import time
import dlib
import tempfile
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)
if task_name not in task_lookup or task.info is None:
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...',
'data': task.info,
}
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', ''),
'data': task.info,
}
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
'data': task.info,
}
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/status/{}/{}'.format('sleep', async_task.id)
return jsonify({
'result': True,
'task_url': task_url,
})
@api_task.route('/upload/blur', methods=['POST'])
def upload():
return process('blur')
@api_task.route('/upload/demo', methods=['POST'])
def demo():
return process('demo')
def process(style):
"""
Process an image 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',
})
print('get file...')
file = request.files['query_img']
uuid_str = str(uuid.uuid4())
print('[+] style: {}'.format(style))
print('[+] uuid_name: {}'.format(uuid_str))
im = Image.open(file.stream).convert('RGB')
im = ImageOps.fit(im, (256, 256,))
tmpfile = tempfile.NamedTemporaryFile(delete=False)
# Save image to disk
print('[+] Save image to temporary file')
im.save(tmpfile, 'JPEG', quality=80)
print('[+] Start celery')
async_task = task.apply_async(args=[uuid_str, tmpfile.name])
task_url = '/task/status/{}/{}'.format(style, async_task.id)
return jsonify({
'result': True,
'taskURL': task_url,
'uuid': uuid_str
})
|
