MIRRORS/yocto-autobuilder2
1
0
Fork 0
mirror of git://git.yoctoproject.org/yocto-autobuilder2.git synced 2025-07-19 12:49:03 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

reporters/swatbot: If we see a 302 response, attempt a new login

Browse Source 
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
This commit is contained in:
Richard Purdie 2021-04-10 16:04:49 +01:00
parent 1f411279ec
commit f027b0e006
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
reporters/swatbot.py
View File

@ -40,6 +40,15 @@ class SwatBotURI(object):
csrftoken = req2.cookies['csrftoken']
self.headers = {'Content-type':'application/vnd.api+json', 'X-CSRFToken': csrftoken}
def query_with_login(self, query):
req = self.client.get(query, timeout=self.TIMEOUT, headers=self.headers, allow_redirects=False)
# A 302 status means we probably need to renew the login
if req.status_code == requests.codes.found and "login" in req.headers['Location']:
log.err("SwatBot: Attempting to renew login")
self.login()
return self.client.get(query, timeout=self.TIMEOUT, headers=self.headers, allow_redirects=False)
return req
def find_build_collection(self, build):
dbid = None
@ -49,7 +58,7 @@ class SwatBotURI(object):
if build['buildset']['parent_buildid']:
collection_build_id = build['buildset']['parent_buildid']
req = self.client.get(url + "?buildid=" + str(collection_build_id), timeout=self.TIMEOUT, headers=self.headers)
req = self.query_with_login(url + "?buildid=" + str(collection_build_id))
if req.status_code == requests.codes.ok:
try:
data = req.json()['data']
@ -109,7 +118,7 @@ class SwatBotURI(object):
log.err("SwatBot: Couldn't find BuildCollection database ID")
return False
req = self.client.get(url + "?buildid=" + str(build['buildid']), timeout=self.TIMEOUT, headers=self.headers)
req = self.query_with_login(url + "?buildid=" + str(build['buildid']))
if req.status_code == requests.codes.ok:
data = req.json()['data']
if len(data) > 1:
@ -144,7 +153,7 @@ class SwatBotURI(object):
@defer.inlineCallbacks
def update_build(self, build, master):
req = self.client.get(self.uri + "rest/build/?buildid=" + str(build['buildid']), timeout=self.TIMEOUT, headers=self.headers)
req = self.query_with_login(self.uri + "rest/build/?buildid=" + str(build['buildid']))
if req.status_code != requests.codes.ok:
log.err("SwatBot: Couldn't find build to update: %s (%s)" % (str(req.status_code), str(req.json())))
return False