hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added bearer token

Browse Source
This commit is contained in:
jeffser
2024-06-28 21:29:36 -06:00
parent 0fbde265ef
commit 527f7b4452
3 changed files with 43 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/connection_handler.py
View File

@@ -2,10 +2,19 @@
import json, requests
url = None
bearer_token = None
def get_headers() -> dict:
headers = {
"Content-Type": "application/json"
}
if bearer_token:
headers["Authorization"] = "Bearer " + bearer_token
return headers
def simple_get(connection_url:str) -> dict:
try:
response = requests.get(connection_url)
response = requests.get(connection_url, headers=get_headers())
if response.status_code == 200:
return {"status": "ok", "text": response.text, "status_code": response.status_code}
else:
@@ -15,10 +24,7 @@ def simple_get(connection_url:str) -> dict:
def simple_post(connection_url:str, data) -> dict:
try:
headers = {
"Content-Type": "application/json"
}
response = requests.post(connection_url, headers=headers, data=data, stream=False)
response = requests.post(connection_url, headers=get_headers(), data=data, stream=False)
if response.status_code == 200:
return {"status": "ok", "text": response.text, "status_code": response.status_code}
else:
@@ -28,7 +34,7 @@ def simple_post(connection_url:str, data) -> dict:
def simple_delete(connection_url:str, data) -> dict:
try:
response = requests.delete(connection_url, json=data)
response = requests.delete(connection_url, headers=get_headers(), json=data)
if response.status_code == 200:
return {"status": "ok", "status_code": response.status_code}
else:
@@ -38,10 +44,7 @@ def simple_delete(connection_url:str, data) -> dict:
def stream_post(connection_url:str, data, callback:callable) -> dict:
try:
headers = {
"Content-Type": "application/json"
}
response = requests.post(connection_url, headers=headers, data=data, stream=True)
response = requests.post(connection_url, headers=get_headers(), data=data, stream=True)
if response.status_code == 200:
for line in response.iter_lines():
if line: