updates

1 files changed, 38 insertions, 36 deletions

diff --git a/reupload.py b/reupload.py
index 94fbbe3..5d2c0a0 100755
--- a/reupload.py
+++ b/reupload.py
@@ -97,48 +97,50 @@ def reupload_post(args):
     else:
         image_count = 1
 
-    if image_count != 1:
-        raise NotImplementedError()
+    session = requests.Session()
+    (login_token, csrf_token) = login(session)
 
-    jpeg_path = base_path + ".jpg"
-    date_part = base_path.split('/')[1].split('_')[0].replace('-', '')
-    time_part = base_path.split('/')[1].split('_')[1].replace('-', '')[:4]
-    remote_name = f"CEQD_{date_part}_{time_part}_1.jpg"
-    page_text = f"""[[Category:Center for Extraordinary Quarantine Dining]]
+    for image_num in range(1,image_count+1):
+
+        if image_count == 1:
+            jpeg_path = base_path + ".jpg"
+        else:
+            jpeg_path = base_path + f"_{image_num}.jpg"
+        date_part = base_path.split('/')[1].split('_')[0].replace('-', '')
+        time_part = base_path.split('/')[1].split('_')[1].replace('-', '')[:4]
+        remote_name = f"CEQD-{date_part}-{time_part}-{image_num}.jpg"
+        page_text = f"""[[Category:Center for Extraordinary Quarantine Dining]]
 
 == Summary ==
 {caption}
 
 Imported from ELS instagram: https://www.instagram.com/p/{shortcode}/
-    """
-
-    session = requests.Session()
-    (login_token, csrf_token) = login(session)
-
-    # First, check if file already exists
-    existing = get_imageinfo(remote_name, session)
-    if existing:
-        print(json.dumps(existing, sort_keys=True, indent=2))
-        sys.exit(-1)
-
-    # If it doesn't, upload it!
-    print(f"Uploading {remote_name}")
-    resp = session.post(
-        BASE_URL,
-        data={
-            "action": "upload",
-            "filename": remote_name,
-            "text": page_text,
-            "format": "json",
-            "token": csrf_token,
-            "ignorewarnings": 1
-        },
-        files={
-            'file': ('filename.jpg', open(jpeg_path, 'rb'), 'multipart/form-data'),
-        },
-    )
-    resp.raise_for_status()
-    print(json.dumps(resp.json(), sort_keys=True, indent=2)
+        """
+
+        # First, check if file already exists
+        existing = get_imageinfo(remote_name, session)
+        if existing:
+            print(f"Already exists: {remote_name}")
+            continue
+
+        # If it doesn't, upload it!
+        print(f"Uploading: {remote_name}")
+        resp = session.post(
+            BASE_URL,
+            data={
+                "action": "upload",
+                "filename": remote_name,
+                "text": page_text,
+                "format": "json",
+                "token": csrf_token,
+                "ignorewarnings": 1
+            },
+            files={
+                'file': ('filename.jpg', open(jpeg_path, 'rb'), 'multipart/form-data'),
+            },
+        )
+        resp.raise_for_status()
+        #print(json.dumps(resp.json(), sort_keys=True, indent=2)
 
 def main():
     parser = argparse.ArgumentParser(