summaryrefslogtreecommitdiff
path: root/megapixels/app/server/tasks/sleep.py
blob: 9b91cc526a4682c5512754e011ff4f2e72c2ad2b (plain)
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
import time

# from .. import basemodels
# celery = basemodels.celery

from celery.utils.log import get_task_logger
celery_logger = get_task_logger(__name__)

from app.server.tasks import celery

import imutils

@celery.task(bind=True)
def sleep_task(self, uuid_name):
  celery_logger.debug('sleep_task'.format(uuid_name))
  msgs = [ 
    {'msg':'Uploaded OK','time':.1},
    {'msg':'Segmenting Image...','time':2},
    {'msg':'Found: Person, Horse','time':1},
    {'msg':'Creating Pix2Pix','time':2}
  ]
  for i,m in enumerate(msgs):
    percent = int(float(i)/float(len(msgs))*100.0)
    self.update_state(
      state = 'PROCESSING',
      meta = {
        'percent': percent,
        'message': m['msg'],
        'uuid': uuid_name
      })
    celery_logger.debug(m['msg'])
    time.sleep(m['time'])

  return {
    'percent': 100,
    'state': 'complete',
    'uuid': uuid_name
  }