minuku_androidServer/server.py at master · minuku/minuku_androidServer · 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
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
from flask import Flask, request, jsonify, json
from bson.objectid import ObjectId
from flask_pymongo import PyMongo
import datetime
from datetime import datetime
import dateutil.parser
from bson import json_util
import codecs
import ast
import logging

app = Flask(__name__)

app.config['MONGO_DBNAME'] = 'LabelingStudy'
app.config['MONGO_URI'] = 'mongodb://localhost/LabelingStudy'

mongo = PyMongo(app)

TripDataKey = ["_id", "device_id", "condition", "createdTime", "startTime", "endTime", "startTimeString", "endTimeString", "sessionid", "annotations"]


@app.route('/')
def test():
    return 'OK'


@app.route('/find_latest_and_insert', methods=['POST'])
def find_latest_and_insert():
    if request.method == 'POST':
        collection = str(request.args['collection'])
        action = str(request.args['action'])
        user_id = request.args['id']

        if collection == 'dump':
            user = mongo.db.dump
        elif collection == 'trip':
            user = mongo.db.trip
        elif collection == 'isAlive':
            user = mongo.db.isAlive
        if action == 'search':
            data = user.find({'device_id': user_id})
            res = data.sort('startTime', -1).limit(1)

            returnStartTime = 0
            for item in res:
                returnStartTime = item['startTime']
            message = json.dumps({'startTime':returnStartTime})
            return message

        json_request = request.get_json(force=True, silent=True)
        if action == 'insert' and collection == 'trip':
            print (json_request)

            missing_key = False
            try:
                if '_id' in json_request and 'createdTime' in json_request:
                    data = dict()
                    for key in TripDataKey:
                        if key in json_request and key != '_id':
                            data[key] = json_request[key]
                        elif key not in json_request:
                            missing_key = True
                    if missing_key:
                        file = open('MissingKeyData.txt', 'a')
                        file.write(str(json_request)+'\n')

                    user.update({'_id': json_request['_id']}, {'$set': data}, upsert=True, multi=True)
            except Exception as e:
                print (e)
            else:  # if try successfully, then execute else
                # unique data
                user_data = user.find({'_id': json_request['_id']})
                returnCreatedTime = 0
                for item in user_data:
                    returnCreatedTime = item['createdTime']
                message = json.dumps({'createdTime': returnCreatedTime})
                return message
        elif action=='insert' and collection=='dump':
            try:
                user.insert(json_request)
                user_data = user.find({'device_id': user_id})
                res = user_data.sort('endTime', -1).limit(1)
                returnEndTime = 0
                for item in res:
                    returnEndTime = item['endTime']
                message = json.dumps({'endTime': returnEndTime})
                return message
            except Exception as e:
                print (e)
        else:
            user.insert(json_request)
            return 'insert OK!'


@app.route('/time_interval', methods=['POST'])
def time_interval():
    jsonquery = request.get_json(force=True, silent=True)  # type: unico$
    query = json.loads(jsonquery)
    d_id = query['device_id']
    collection = query['collection']
    start = query['query_start_time']
    end = query['query_end_time']

    if collection == 'dump':
        col = mongo.db.dump
    elif collection == 'trip':
        col = mongo.db.trip
    elif collection == 'isAlive':
        col = mongo.db.isAlive

    json_docs = []
    number = col.find({'device_id': d_id, 'startTime': {'$gte': str(start["$date"])}, 'endTime': {'$lt': str(end["$date"])}}).count()
    in_time_range = col.find({'device_id': d_id, 'startTime': {'$gte': str(start["$date"])}, 'endTime': {'$lt': str(end["$date"])}})
    for doc in in_time_range:
        json_docs.append(doc)
    print (json_docs)
    return json.dumps({'device_id': d_id, "number": number})

if __name__ == '__main__':
    app.run(host="172.31.3.66", threaded=True)