def setUp(self): self.app = app_from_env() self.mock_import_result = { 'bulkResponse': { 'took': 15, 'errors': False, 'items': [{ 'create': { '_index': 'yvqg1901zmu5bw9q', '_type': '_doc', '_id': 'dd0KZtqyec48n1q1fniqVMV5yllhRRGx', '_version': 1, 'result': 'created', 'forced_refresh': True, '_shards': { 'total': 1, 'successful': 1, 'failed': 0 }, '_seq_no': 0, '_primary_term': 1, 'status': 201 } }] }, 'failed': [], 'created': ['dd0KZtqyec48n1q1fniqVMV5yllhRRGx'], 'jobId': 'ba310246-1f87-1ece-b67c-be3f79a80d11' } # A mock search result used for asset search tests self.mock_search_result = { 'took': 4, 'timed_out': False, 'hits': { 'total': {'value': 2}, 'max_score': 0.2876821, 'hits': [ { '_index': 'litvqrkus86sna2w', '_type': 'asset', '_id': 'dd0KZtqyec48n1q1ffogVMV5yzthRRGx2WKzKLjDphg', '_score': 0.2876821, '_source': { 'source': { 'path': 'https://i.imgur.com/SSN26nN.jpg' } } }, { '_index': 'litvqrkus86sna2w', '_type': 'asset', '_id': 'aabbccddec48n1q1fginVMV5yllhRRGx2WKyKLjDphg', '_score': 0.2876821, '_source': { 'source': { 'path': 'https://i.imgur.com/foo.jpg' } } } ] } }
</video> Usage: python main.py -f <ANALYSIS FILE TO SAMPLE> -s <API SERVER URL> -k <PATH TO API KEY> ''' parser = ArgumentParser() parser.add_argument("-k", "--key", dest="key", help="Path to APIKEY json file") parser.add_argument("-s", "--server", dest="server", help="Server path, defaults to https://api.zvi.zorroa.com", default="https://api.zvi.zorroa.com") parser.add_argument("-f", "--file", dest="file", help="File ID. Found in Visualizer > Info > Files. Example: assets/l5azb3Pp-lWU7F3J4yIgokvnSbGBH5pc/timeline/gcp-video-label-detection-timeline.json.gz", default=None) args = parser.parse_args() md_key = args.key md_server = args.server md_file = args.file app = app_from_env() apikey = {} with open(md_key) as json_file: apikey = json.load(json_file) app.client.apikey = apikey app.client.server = md_server b = app.assets.download_file(stored_file=md_file) def _sort_start_clip(clip): return float(clip["start"]) def _generate_webvtt(name, clips): print(name)
def setUp(self): self.app = app_from_env() self.mock_search_result = mock_search_result