add: points data

backend_refactor/service/video_analyzer.py

@@ -84,6 +84,7 @@ class BaseVideoAnalyzer:
     def __init__(self, data: dict):
         self.video_uuid = data['video_uuid']
         self.results = None
+        self.points = None
 
     def _do_analyze(self):
         assert False, "Internal error: You should override this with a subclass!"
@@ -96,6 +97,7 @@ class BaseVideoAnalyzer:
 
             if not os.path.exists(PICKLE_DUMP_PATH):  # analyze
                 self.results = run_openpose_for_npy_and_video(self.video_uuid, callback)
+                self.points = self.results
                 pickle.dump(self.results, open(PICKLE_DUMP_PATH, "wb"))
             else:  # read cache
                 self.results = pickle.load(open(PICKLE_DUMP_PATH, "rb"))
@@ -105,11 +107,11 @@ class BaseVideoAnalyzer:
             try:
                 for i in result['data']:
                     result['data'][i]['file'] = FILE_URL + self.video_uuid + "." + result['data'][i]['index'] + ".jpg"
-                result['data']['points'] = self.results
+                result['data']['points'] = self.points
             except Exception as e:
                 logging.warning("file index in data not found, %s" % e)
 
-            print(result)
+            print(result['data']['points'])
             callback("FINISHED", result=result)
         except Exception as e:
             callback("ERROR", error=e)