From 4ce9b320d303987c91eb58c6bc1cd3bcdffef0b3 Mon Sep 17 00:00:00 2001
From: Stefal <stephane.peneau@wanadoo.fr>
Date: Wed, 20 Sep 2023 12:21:14 +0200
Subject: [PATCH] get sequence from image id(s)

---
 README.md   | 25 ++++++++++++++++++++++++-
 download.py | 34 +++++++++++++++++++---------------
 2 files changed, 43 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index efd6ed5..248fb1f 100644
--- a/README.md
+++ b/README.md
@@ -4,5 +4,28 @@ Simple code to download images in one or several mapillary sequences.
 ## How to use
 change the access token with your access token and the sequence ids with the ids of the sequences you want to download
 ```Shell
-python download.py --access_token "MLY|xxxx|xxxxxxx" --sequence_ids xxxxxxxxxxx xxxxxxxxxxx
+python download.py "MLY|xxxx|xxxxxxx" --sequence_ids xxxxxxxxxxx xxxxxxxxxxx
+```
+
+## Available arguments
+```Shell
+python download.py -h
+usage: download.py [-h] [--sequence_ids [SEQUENCE_IDS ...]] [--image_ids [IMAGE_IDS ...]] [--destination DESTINATION]
+                   [--image_limit IMAGE_LIMIT] [--overwrite]
+                   access_token
+
+positional arguments:
+  access_token          Your mapillary access token
+
+optional arguments:
+  -h, --help            show this help message and exit
+  --sequence_ids [SEQUENCE_IDS ...]
+                        The mapillary sequence id(s) to download
+  --image_ids [IMAGE_IDS ...]
+                        The mapillary image id(s) to get their sequence id(s)
+  --destination DESTINATION
+                        Path destination for the images
+  --image_limit IMAGE_LIMIT
+                        How many images you want to download
+  --overwrite           overwrite existing images
 ```
diff --git a/download.py b/download.py
index cdf3429..53e42e9 100644
--- a/download.py
+++ b/download.py
@@ -21,15 +21,17 @@ session.mount('https://', HTTPAdapter(max_retries=retries_strategies))
 
 def parse_args(argv =None):
     parser = argparse.ArgumentParser()
-    parser.add_argument('--sequence_ids', type=str, nargs='+', help='The mapillary sequence id(s) to download')
-    parser.add_argument('--access_token', type=str, help='Your mapillary access token')
+    parser.add_argument('access_token', type=str, help='Your mapillary access token')
+    parser.add_argument('--sequence_ids', type=str, nargs='*', help='The mapillary sequence id(s) to download')
+    parser.add_argument('--image_ids', type=int, nargs='*', help='The mapillary image id(s) to get their sequence id(s)')
     parser.add_argument('--destination', type=str, default='data', help='Path destination for the images')
     parser.add_argument('--image_limit', type=int, default=None, help='How many images you want to download')
     parser.add_argument('--overwrite', default=False, action='store_true', help='overwrite existing images')
 
     global args
     args = parser.parse_args(argv)
-    #print(args)
+    if args.sequence_ids is None and args.image_ids is None:
+        parser.error("Please enter at least one sequence id or image id")
 
 def background(f):
     def wrapped(*args, **kwargs):
@@ -47,10 +49,9 @@ def download(url, filepath, metadata=None):
     print("{} downloaded".format(filepath))
 
 def get_single_image_data(image_id, mly_header):
-    req_url = 'https://graph.mapillary.com/{}?fields=thumb_original_url,altitude,camera_type,captured_at,compass_angle,geometry,exif_orientation'.format(image_id)
+    req_url = 'https://graph.mapillary.com/{}?fields=thumb_original_url,altitude,camera_type,captured_at,compass_angle,geometry,exif_orientation,sequence'.format(image_id)
     r = session.get(req_url, headers=mly_header)
     data = r.json()
-    #print(data)
     return data
 
 def get_image_data_from_sequences(sequences_id, mly_header):
@@ -116,30 +117,33 @@ def write_exif(picture, img_metadata):
 if __name__ == '__main__':
     parse_args()
 
-    if args.sequence_ids == None:
-        print('please provide the sequence_id')
-        exit()
-
     if args.access_token == None:
         print('please provide the access_token')
         exit()
 
-    sequence_ids= args.sequence_ids
+    sequence_ids= args.sequence_ids if args.sequence_ids is not None else []
+    images_ids = args.image_ids
     access_token = args.access_token
     images_data = []
     header = {'Authorization' : 'OAuth {}'.format(access_token)}
-    
-    # create the data folder
-    if not os.path.exists('data'):
-            os.makedirs('data')
+
+    if images_ids:
+        for image_id in images_ids:
+            image_data = get_single_image_data(image_id, header)
+            if 'error' in image_data:
+                print("data : ", image_data)
+                print("something wrong happened ! Please check your image id and/or your connection")
+                sys.exit()
+            else:
+                sequence_ids.append(image_data.get('sequence'))
 
     #for i,image_data in enumerate(get_image_data_from_sequences(sequence_ids, header)):
     for i,image_data in enumerate(get_image_data_from_sequences__future(sequence_ids, header)):
         if args.image_limit is not None and i >= args.image_limit:
             break
         if 'error' in image_data:
-            print("something wrong happened ! Please check your token and/or your connection")
             print("data : ", image_data)
+            print("something wrong happened ! Please check your token and/or your connection")
             sys.exit()
         images_data.append(image_data)
     #sys.exit()