From 9346539c20b977abaee13522eab3763bac72d684 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Z=C3=BCmr=C3=BCd-=C3=BC=20Anka?= <osmandurdag@hotmail.com>
Date: Sun, 27 Dec 2020 20:20:21 +0300
Subject: [PATCH] Added python script files for download_vot.sh and
 unpack_vot.sh

---
 scripts/download_vot.py | 75 +++++++++++++++++++++++++++++++++++++++++
 scripts/unpack_vot.py   | 22 ++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 scripts/download_vot.py
 create mode 100644 scripts/unpack_vot.py

diff --git a/scripts/download_vot.py b/scripts/download_vot.py
new file mode 100644
index 0000000..e93419e
--- /dev/null
+++ b/scripts/download_vot.py
@@ -0,0 +1,75 @@
+import sys
+import os
+import requests
+import json
+import zipfile
+import io
+import urllib.request, shutil
+
+VOT_YEAR = sys.argv[1]
+VOT_CHALLENGE = "main"
+name = "vot" + VOT_YEAR + "_" + VOT_CHALLENGE
+
+dl = f"../dl/{name}"
+
+try:
+    os.makedirs(dl)
+except FileExistsError:
+    print("Loading...")
+
+abspath = os.path.abspath(__file__)
+projectDir = os.path.dirname(abspath)
+
+os.chdir(dl)
+
+base_url = f"https://data.votchallenge.net/vot{VOT_YEAR}/{VOT_CHALLENGE}"
+
+if os.path.exists("description.json"):
+    pass
+else:
+    response = requests.get(base_url + "/description.json", allow_redirects = True)
+    with open("description.json", "wb") as descriptionFile:
+        descriptionFile.write(response.content)
+
+with open("annotations.txt", "w") as annotationsFileOutput, open("color.txt", "w") as colorFileOutput, open("list.txt", "w") as listFileOutput:
+    with open("description.json", "r") as description:
+        data = json.load(description)
+        for sequence in data["sequences"]:
+            annotationsFileOutput.write(sequence["annotations"]["url"] + "\n")
+            colorFileOutput.write(sequence["channels"]["color"]["url"] + "\n")
+            listFileOutput.write(sequence["name"] + "\n")
+
+try:
+    os.makedirs("annotations")
+except FileExistsError:
+    print("Loading...")
+
+try:
+    os.makedirs("color")
+except FileExistsError:
+    print("Loading...")
+
+os.chdir("./annotations")
+with open("../annotations.txt", "r") as annotationsTxtFile:
+    lines = annotationsTxtFile.readlines()
+    for line in lines:
+        url = f"{base_url}/{line}"
+        response = requests.get(url, stream = True)
+        with urllib.request.urlopen(url) as response, open(line.strip(), "wb") as downloadedAnnotation:
+            shutil.copyfileobj(response, downloadedAnnotation)
+            print("Test datas are downloading...")
+
+os.chdir("../color")
+with open("../color.txt", "r") as colorTxtFile:
+    lines = colorTxtFile.readlines()
+    for line in lines:
+        url = f"{base_url}/{line}"
+        response = requests.get(url, stream = True)
+        with urllib.request.urlopen(url) as response, open(line[16:].strip(), "wb") as downloadedColor:
+            shutil.copyfileobj(response, downloadedColor)
+            print("Test datas are downloading...")
+        
+        # response = requests.get(url, allow_redirects = True)
+        # with open(line[16:].strip(), "wb") as downloadedColor:
+        #     downloadedColor.write(response.content)
+        #     print("Test datas are downloading...")
\ No newline at end of file
diff --git a/scripts/unpack_vot.py b/scripts/unpack_vot.py
new file mode 100644
index 0000000..df7082e
--- /dev/null
+++ b/scripts/unpack_vot.py
@@ -0,0 +1,22 @@
+import sys
+import os
+from shutil import copyfile
+VOT_YEAR = sys.argv[1]
+VOT_CHALLENGE = "main"
+
+defaultName = f"vot{VOT_YEAR}_{VOT_CHALLENGE}"
+
+dl = "../" + sys.argv[2]
+data = "../../" + sys.argv[3]
+# scripts="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+try:
+    os.makedirs(data)
+except FileExistsError:
+    print("Loading...")
+
+descriptionFileSrc = f"../dl/{defaultName}/description.json"
+descriptionFileDestination = f"{data}/description.json"
+copyfile(descriptionFileSrc, descriptionFileDestination)
+
+os.system(f"python unzip_vot.py {dl} {data}")
\ No newline at end of file