updates csrf token when invalid

opencultureconsulting · Jan 15, 2022 · 0563b54 · 0563b54
1 parent fa3e352
commit 0563b54
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/google/refine/refine.py b/google/refine/refine.py
@@ -97,7 +97,12 @@ def urlopen(self, command, data=None, params=None, project_id=None):
         try:
             response = urllib2.urlopen(req)
         except urllib2.HTTPError as e:
-            raise Exception('HTTP %d "%s" for %s\n\t%s' % (e.code, e.msg, e.geturl(), data))
+            if 'Missing or invalid csrf_token parameter' == e.msg:
+                self.get_csrf_token()
+                params['csrf_token'] = self.token
+                response = urllib2.urlopen(req)
+            else:
+                raise Exception('HTTP %d "%s" for %s\n\t%s' % (e.code, e.msg, e.geturl(), data))
         except urllib2.URLError as e:
             raise urllib2.URLError(
                 '%s for %s. No Refine server reachable/running; ENV set?' %