learningpal_api/Testing_Parallel.py at master · TomNong/learningpal_api · GitHub

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
import requests, json, time, multiprocessing, sys, signal

def recognize(IMG_NAME):
    url = 'http://api.learningpal.com/math/upload'
    files = {'file': open('./' + IMG_NAME, 'rb')}
    headers = {'content-type': 'application/json'}
    try:
        r = requests.post(url, files=files)
        print r.text
        response = json.loads(r.text)
        task_ID = response['task_ID']
        payload = {'task_ID' : task_ID, 'password' : ""}
        url2 = 'http://api.learningpal.com/math/result'
        while True:
            response = requests.post(url2, data=json.dumps(payload), headers=headers)
            print response.text
            if response.text and "processing" not in response.text:
                return response.text
                break
    except Exception as e :
        print e
        return None

def main(arg):
    print 'Job started...'
    tasks = ['single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
             'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
             'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png',
            'single_equation.png','single_equation.png','single_equation.png','single_equation.png','single_equation.png']
    tic = time.time()
    #result = worker(arg)
    #print result
    pool = multiprocessing.Pool(multiprocessing.cpu_count())
    tasks = tasks[:(int(arg))]
    print '    number of sending images: ', len(tasks)
    results = pool.map(recognize, tasks)
    pool.close()
    pool.join()
    print results
    print '    Done! Time cost: ', str(time.time() - tic), 's'

def quit(signum, frame):
    print 'Stop'
    sys.exit()
if __name__=="__main__":
    signal.signal(signal.SIGINT, quit)
    signal.signal(signal.SIGTERM, quit)
    main(sys.argv[1])