from flask_sslify import SSLify
from flask import Flask, request, jsonify, Response
from flask_cors import CORS
-from camera_driver.capture_frame import VideoCamera, VideoFile
-from influxdb import InfluxDBClient
+# from camera_driver.capture_frame import VideoCamera, VideoFile
+# from capture_frame import VideoCamera, VideoFile
+# from influxdb import InfluxDBClient
import json
import time
import requests
+import os
+import cv2
app = Flask(__name__)
app.config['CORS_SUPPORTS_CREDENTIALS'] = True
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg'])
+ALLOWED_VIDEO_EXTENSIONS = {'mp4'}
count = 0
listOfMsgs = []
listOfCameras = []
listOfVideos = []
+mock_func = 1
-class inventry_info:
+class inventory_info:
"""
Store the data and manage multiple input video feeds
"""
def __init__(self, current_count=0, total_count=0, time=0):
- self.type = "Shelf_INV"
+ self.type = "Shelf_INV1"
self.labels = "Bottles"
self.current_count = current_count
self.total_count = total_count
return self.time
-def store_data(inventry_info):
+# temporary copied capture_frame file to this due to docker issue for module
+# import
+
+class VideoCamera(object):
+ """
+ opneCV to capture frame from a camera
+ """
+ def __init__(self, url):
+ self.video = cv2.VideoCapture(url)
+
+ def delete(self):
+ self.video.release()
+
+ def get_frame(self):
+ """
+ get a frame from camera url
+ """
+ success, image = self.video.read()
+ return success, image
+
+
+class VideoFile(object):
+ """
+ opneCV to capture frame from a video stream
+ """
+ def __init__(self, video_name):
+ self.video = cv2.VideoCapture("./test/resources/" + video_name)
+
+ def delete(self):
+ self.video.release()
+
+ def get_frame(self):
+ """
+ get a frane from stream
+ """
+ success, image = self.video.read()
+ return success, image
+
+
+def store_data(inventory_info):
"""
store time series data in influx db
"""
# TODO config, schema table, DB, fill data set
create_database()
- store_info_db(inventry_info)
+ store_info_db(inventory_info)
-def shelf_inventry(video_capture, camera_info):
+def mock_table(inven_info):
+ current_count = 3
+ labels = "Bottles"
+ total_count = 6
+ inven_info.setcurrentcount(current_count)
+ inven_info.settotalcount(total_count)
+ inven_info.setlabel(labels)
+ inven_info.utime = time.time()
+ # store_data(inven_info)
+ local_store(inven_info)
+
+
+def shelf_inventory(video_capture, camera_info, true=None):
"""
- 人脸识别
+ shelf_inventory
"""
global count
+ global mock_func
+
labels = "bottles"
process_this_frame = 0
- while True:
- success, frame = video_capture.get_frame()
- if not success:
- break
- if process_this_frame == 0:
- url = config.detection_url + "/v1/obj_detection/detect"
- # info1 = cv2.imencode(".jpg", rgb_small_frame)[1].tobytes()
- data = json.loads(requests.post(url, data=frame,
- verify=config.ssl_cacertpath).text)
- inven_info = inventry_info()
+ if mock_func == 1:
+ inven_info = inventory_info()
+ mock_table(inven_info)
+ else:
+ while True:
+ success, frame = video_capture.get_frame()
+ if not success:
+ break
+ if process_this_frame == 0:
+ url = config.detection_url + "/v1/obj_detection/detect"
+ # info1 = cv2.imencode(".jpg", rgb_small_frame)[1].tobytes()
+ data = json.loads(requests.post
+ (url, data=frame,
+ verify=config.ssl_cacertpath).text)
+ inven_info = inventory_info()
current_count = data[count]
labels = data[labels]
total_count = inven_info.current_count + inven_info.total_count
inven_info.settotalcount(total_count)
inven_info.setlabel(labels)
inven_info.utime = time.time()
- store_data(inven_info)
+ # store_data(inven_info)
+ local_store(inven_info)
+
+
+def local_store(inven_info):
+ """
+ store "shelf" data to array
+
+ :param inven_info: Inventry object
+ :return: None
+ """
+ if len(listOfMsgs) >= 100:
+ listOfMsgs.pop()
+ newdict = {"shelfName": inven_info.type, "ObjType": inven_info.labels,
+ "currentCount": inven_info.current_count,
+ "totalCount": inven_info.total_count,
+ "time": time.time()}
+ listOfMsgs.insert(0, newdict)
def store_info_db(inven_info):
:return: None
"""
global db_client
+# proxy = {"http": "http://{}:{}".format(config.IPADDRESS, config.PORT)}
+# db_client = InfluxDBClient(host=config.IPADDRESS, port=config.PORT,
+# proxies=proxy, database=config.DATABASE_NAME)
+# db_client.create_database(config.DATABASE_NAME)
- proxy = {"http": "http://{}:{}".format(config.IPADDRESS, config.PORT)}
- db_client = InfluxDBClient(host=config.IPADDRESS, port=config.PORT,
- proxies=proxy, database=config.DATABASE_NAME)
- db_client.create_database(config.DATABASE_NAME)
+
+@app.route('/v1/inventry/table', methods=['GET'])
+def inventry_table():
+ """
+ return inventry table
+
+ :return: inventry table
+ """
+ return jsonify(listOfMsgs)
+
+
+@app.route('/v1/inventry/image', methods=['GET'])
+def detected_image():
+ """
+ return inventry table
+
+ :return: inventry table
+ """
+ return jsonify(listOfMsgs)
+
+
+def allowed_videofile(filename):
+ """
+ File types to upload:mp4
+ param: filename:
+ """
+ return '.' in filename and \
+ filename.rsplit('.', 1)[1].lower() in ALLOWED_VIDEO_EXTENSIONS
+
+
+@app.route('/v1/monitor/video', methods=['POST'])
+def upload_video():
+ app.logger.info("Received message from ClientIP [" + request.remote_addr
+ + "] Operation [" + request.method + "]" +
+ " Resource [" + request.url + "]")
+ if 'file' in request.files:
+ files = request.files.getlist("file")
+ for file in files:
+ if allowed_videofile(file.filename):
+ file.save(os.path.join(app.config['VIDEO_PATH'],
+ file.filename))
+ else:
+ raise IOError('video format error')
+ return Response("success")
@app.route('/v1/monitor/cameras', methods=['POST'])
video_file = VideoFile(camera_info["rtspurl"])
video_dict = {camera_info["name"]: video_file}
listOfVideos.append(video_dict)
- return Response(shelf_inventry(video_file, camera_info["name"]),
+ return Response(shelf_inventory(video_file, camera_info["name"]),
mimetype='multipart/x-mixed-replace; boundary=frame')
else:
video_file = VideoCamera(camera_info["rtspurl"])
video_dict = {camera_info["name"]: video_file}
listOfVideos.append(video_dict)
- return Response(shelf_inventry(video_file, camera_info["name"]),
+ return Response(shelf_inventory(video_file, camera_info["name"]),
mimetype='multipart/x-mixed-replace; boundary=frame')
+ "] Operation [" + request.method + "]" +
" Resource [" + request.url + "]")
return jsonify(listOfCameras)
- return Response("success")
+
+
+@app.route('/', methods=['GET'])
+def hello_world():
+ app.logger.info("Received message from ClientIP [" + request.remote_addr
+ + "] Operation [" + request.method + "]" +
+ " Resource [" + request.url + "]")
+ return Response("Hello MEC Developer")
def start_server(handler):
Code Review / ealt-edge.git / blobdiff