49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
|
from os import environ
|
||
|
|
import json
|
||
|
|
import requests
|
||
|
|
from io import BytesIO
|
||
|
|
|
||
|
|
from sauceclient import SauceClient, SauceException
|
||
|
|
|
||
|
|
try:
|
||
|
|
import http.client as http_client
|
||
|
|
from urllib.parse import urlencode
|
||
|
|
except ImportError:
|
||
|
|
import httplib as http_client
|
||
|
|
from urllib import urlencode
|
||
|
|
|
||
|
|
|
||
|
|
sauce_username = environ.get('SAUCE_USERNAME')
|
||
|
|
sauce_access_key = environ.get('SAUCE_ACCESS_KEY')
|
||
|
|
|
||
|
|
sauce = SauceClient(sauce_username, sauce_access_key)
|
||
|
|
apibase = 'eu-central-1.saucelabs.com'
|
||
|
|
|
||
|
|
|
||
|
|
def request(method, url, body=None, content_type='application/json'):
|
||
|
|
"""This is to monkey patch further this method in order to use apibase"""
|
||
|
|
headers = sauce.make_auth_headers(content_type)
|
||
|
|
connection = http_client.HTTPSConnection(apibase)
|
||
|
|
connection.request(method, url, body, headers=headers)
|
||
|
|
response = connection.getresponse()
|
||
|
|
data = response.read()
|
||
|
|
connection.close()
|
||
|
|
if response.status not in [200, 201]:
|
||
|
|
raise SauceException('{}: {}.\nSauce Status NOT OK'.format(
|
||
|
|
response.status, response.reason), response=response)
|
||
|
|
return json.loads(data.decode('utf-8'))
|
||
|
|
|
||
|
|
sauce.request = request
|
||
|
|
|
||
|
|
def upload_from_url(apk_path = str()):
|
||
|
|
response = requests.get(apk_path, stream=True)
|
||
|
|
response.raise_for_status()
|
||
|
|
apk_name = apk_path.split("/")[-1]
|
||
|
|
file = BytesIO(response.content)
|
||
|
|
del response
|
||
|
|
requests.post('https://eu-central-1.saucelabs.com/rest/v1/storage/'
|
||
|
|
+ sauce_username + '/' + apk_name + '?overwrite=true',
|
||
|
|
auth=(sauce_username, sauce_access_key),
|
||
|
|
data=file,
|
||
|
|
headers={'Content-Type': 'application/octet-stream'})
|