From edef07a5920b8d80848d8777d7b05ddebf05e6e3 Mon Sep 17 00:00:00 2001
From: Klaus-Uwe Mitterer <git@kumi.email>
Date: Sat, 27 Feb 2021 14:37:26 +0100
Subject: [PATCH] Make TDV steal more resilient

---
 tdvsteal/__init__.py | 31 ++++++++++++++++++-------------
 1 file changed, 18 insertions(+), 13 deletions(-)

diff --git a/tdvsteal/__init__.py b/tdvsteal/__init__.py
index b01baed..fd53bca 100644
--- a/tdvsteal/__init__.py
+++ b/tdvsteal/__init__.py
@@ -6,8 +6,13 @@ import subprocess
 import tempfile
 import pathlib
 import os
+
 from stitching import tiles_to_equirectangular_blender, multistitch
 
+from socket import timeout
+from ssl import SSLError
+from urllib.error import URLError, HTTPError
+
 def tdv_normalize(url):
     '''
     Takes the URL of any image in a tdv panorama and returns a string
@@ -42,16 +47,7 @@ def tdv_getmaxzoom(schema):
     :return: int value of largest available zoom level
     '''
 
-    l = 0
-
-    while True:
-        try:
-            url = schema % ("f", l+1, 0, 0)
-            with urllib.request.urlopen(url) as res:
-                assert res.getcode() == 200
-                l += 1
-        except:
-            return l
+    return 0
 
 def tdv_export(schema):
     '''
@@ -75,9 +71,18 @@ def tdv_export(schema):
 
             while True:
                 try:
-                    res = urllib.request.urlopen(schema % (tile, maxzoom, y, x))
-                    assert res.getcode() == 200
-                    fo = io.BytesIO(res.read())
+                    print("Getting %s" % (schema % (tile, maxzoom, y, x)))
+                    while True:
+                        try:
+                            res = urllib.request.urlopen(schema % (tile, maxzoom, y, x), timeout=10)
+                            assert res.getcode() == 200
+                            fo = io.BytesIO(res.read())
+                            break
+                        except (timeout, SSLError, URLError) as e:
+                            if isinstance(e, HTTPError):
+                                raise
+                            continue
+                    print("Done")
                     img = PIL.Image.open(fo)
                     r_array.append(img)
                     x += 1