tubearchivist/tubearchivist-server
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist-server.git synced 2024-11-22 03:40:12 +00:00
Code Issues Projects Releases Wiki Activity

add requests timeout

Browse Source
This commit is contained in:
simon 2022-12-05 14:59:24 +07:00
parent 97d566bf59
commit 420702e7cf
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tubearchivist/web/src/webhook_github.py
View File

@ -308,7 +308,7 @@ class GithubBackup:
def ingest_build_line(self): def ingest_build_line(self):
"""ingest latest release into postgres""" """ingest latest release into postgres"""
response = requests.get(self.URL + self.tag) response = requests.get(self.URL + self.tag, timeout=10)
if not response.ok: if not response.ok:
print(response.text) print(response.text)
raise ValueError raise ValueError
@ -424,7 +424,7 @@ class RoadmapHook:
user = self.repo_conf.get("gh_user") user = self.repo_conf.get("gh_user")
repo = self.repo_conf.get("gh_repo") repo = self.repo_conf.get("gh_repo")
url = f"https://api.github.com/repos/{user}/{repo}/contents/README.md" url = f"https://api.github.com/repos/{user}/{repo}/contents/README.md"
response = requests.get(url).json() response = requests.get(url, timeout=10).json()
content = base64.b64decode(response["content"]).decode() content = base64.b64decode(response["content"]).decode()
paragraphs = [i.strip() for i in content.split("##")] paragraphs = [i.strip() for i in content.split("##")]
for paragraph in paragraphs: for paragraph in paragraphs:
@ -452,7 +452,7 @@ class RoadmapHook:
def delete_webhook(self, message_id): def delete_webhook(self, message_id):
"""delete old message""" """delete old message"""
url = f"{self.hook_url}/messages/{message_id}" url = f"{self.hook_url}/messages/{message_id}"
response = requests.delete(url) response = requests.delete(url, timeout=10)
print(response) print(response)
def send_message(self): def send_message(self):
@ -468,7 +468,9 @@ class RoadmapHook:
"color": 10555 "color": 10555
}] }]
} }
response = requests.post(f"{self.hook_url}?wait=true", json=data) response = requests.post(
f"{self.hook_url}?wait=true", json=data, timeout=10
)
print(response) print(response)
print(response.text) print(response.text)
@ -521,14 +523,14 @@ class EsVersionSync:
def get_expected(self): def get_expected(self):
"""get expected es version from readme""" """get expected es version from readme"""
response = requests.get(self.COMPOSE).json() response = requests.get(self.COMPOSE, timeout=10).json()
content = base64.b64decode(response["content"]).decode() content = base64.b64decode(response["content"]).decode()
line = [i for i in content.split("\n") if self.IMAGE in i][0] line = [i for i in content.split("\n") if self.IMAGE in i][0]
self.expected = line.split()[-1] self.expected = line.split()[-1]
def get_current(self): def get_current(self):
"""get current version from docker hub""" """get current version from docker hub"""
response = requests.get(self.TAGS).json() response = requests.get(self.TAGS, timeout=10).json()
all_tags = [i.get("name") for i in response["results"]] all_tags = [i.get("name") for i in response["results"]]
all_tags.pop(0) all_tags.pop(0)
all_tags.sort() all_tags.sort()