def snap_screenshots(snap_id, session, data=None, files=None):
method = "GET"
files_array = None
headers = get_authorization_header(session)
headers["Accept"] = "application/json"
if data:
method = "PUT"
files_array = []
if files:
for f in files:
files_array.append(
(f.filename, (f.filename, f.stream, f.mimetype))
)
else:
# API requires a multipart request, but we have no files to push
# https://github.com/requests/requests/issues/1081
files_array = {"info": ("", data["info"])}
data = None
screenshot_response = api_session.request(
method=method,
url=SCREENSHOTS_QUERY_URL.format(snap_id=snap_id),
headers=headers,
data=data,
files=files_array,
)
if authentication.is_macaroon_expired(screenshot_response.headers):
raise MacaroonRefreshRequired
return process_response(screenshot_response)
def post_register_name(
session, snap_name, registrant_comment=None, is_private=False, store=None
):
json = {"snap_name": snap_name}
if registrant_comment:
json["registrant_comment"] = registrant_comment
if is_private:
json["is_private"] = is_private
if store:
json["store"] = store
response = api_session.post(
url=REGISTER_NAME_URL,
headers=get_authorization_header(session),
json=json,
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_agreement(session):
headers = get_authorization_header(session)
agreement_response = api_session.get(url=AGREEMENT_URL, headers=headers)
if authentication.is_macaroon_expired(agreement_response.headers):
raise MacaroonRefreshRequired
return agreement_response.json()
def post_snap_release(session, snap_name, json):
response = api_session.post(url=SNAP_RELEASE,
headers=get_authorization_header(session),
json=json)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def post_snap_release(session, snap_name, json):
response = api_session.post(
url=SNAP_RELEASE, headers=get_authorization_header(session), json=json
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_account(session):
headers = get_authorization_header(session)
response = api_session.get(url=ACCOUNT_URL, headers=headers)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_account(session):
headers = get_authorization_header(session)
response = api_session.get(url=ACCOUNT_URL, headers=headers)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_agreement(session):
headers = get_authorization_header(session)
agreement_response = api_session.get(url=AGREEMENT_URL, headers=headers)
if authentication.is_macaroon_expired(agreement_response.headers):
raise MacaroonRefreshRequired
return agreement_response.json()
def snap_release_history(session, snap_name):
response = api_session.get(
url=SNAP_RELEASE_HISTORY_URL.format(snap_name=snap_name),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def post_agreement(session, agreed):
headers = get_authorization_header(session)
json = {'latest_tos_accepted': agreed}
agreement_response = cache.get(AGREEMENT_URL, headers, json)
if authentication.is_macaroon_expired(agreement_response.headers):
raise MacaroonRefreshRequired()
return process_response(agreement_response)
def get_snap_info(snap_name, session):
response = api_session.get(
url=SNAP_INFO_URL.format(snap_name=snap_name),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_snap_info(snap_name, session):
response = api_session.get(
url=SNAP_INFO_URL.format(snap_name=snap_name),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def snap_channel_map(session, snap_name):
response = api_session.get(
url=SNAP_CHANNEL_MAP_URL.format(snap_name=snap_name),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def snap_revision_history(session, snap_id):
response = api_session.get(
url=REVISION_HISTORY_URL.format(snap_id=snap_id),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def snap_release_history(session, snap_name, page=1):
response = api_session.get(
url=SNAP_RELEASE_HISTORY_URL.format(snap_name=snap_name, page=page),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def snap_revision_history(session, snap_id):
response = api_session.get(
url=REVISION_HISTORY_URL.format(snap_id=snap_id),
headers=get_authorization_header(session),
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def post_close_channel(session, snap_id, json):
url = CLOSE_CHANNEL.format(snap_id=snap_id)
response = api_session.post(url=url,
headers=get_authorization_header(session),
json=json)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def post_agreement(session, agreed):
headers = get_authorization_header(session)
json = {"latest_tos_accepted": agreed}
agreement_response = api_session.post(
url=AGREEMENT_URL, headers=headers, json=json
)
if authentication.is_macaroon_expired(agreement_response.headers):
raise MacaroonRefreshRequired
return process_response(agreement_response)
def snap_metadata(snap_id, session, json=None):
method = "PUT" if json is not None else None
metadata_response = cache.get(METADATA_QUERY_URL.format(snap_id=snap_id),
headers=get_authorization_header(session),
json=json,
method=method)
if authentication.is_macaroon_expired(metadata_response.headers):
raise MacaroonRefreshRequired()
return process_response(metadata_response)
def post_agreement(session, agreed):
headers = get_authorization_header(session)
json = {"latest_tos_accepted": agreed}
agreement_response = api_session.post(url=AGREEMENT_URL,
headers=headers,
json=json)
if authentication.is_macaroon_expired(agreement_response.headers):
raise MacaroonRefreshRequired
return process_response(agreement_response)
def post_register_name_dispute(session, snap_name, claim_comment):
json = {"snap_name": snap_name, "comment": claim_comment}
response = api_session.post(
url=REGISTER_NAME_DISPUTE_URL,
headers=get_authorization_header(session),
json=json,
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def get_publisher_metrics(session, json):
authed_metrics_headers = PUB_METRICS_QUERY_HEADERS.copy()
auth_header = get_authorization_header(session)["Authorization"]
authed_metrics_headers["Authorization"] = auth_header
metrics_response = api_session.post(url=SNAP_PUB_METRICS_URL,
headers=authed_metrics_headers,
json=json)
if authentication.is_macaroon_expired(metrics_response.headers):
raise MacaroonRefreshRequired
return process_response(metrics_response)
def get_publisher_metrics(session, json):
authed_metrics_headers = PUB_METRICS_QUERY_HEADERS.copy()
auth_header = get_authorization_header(session)["Authorization"]
authed_metrics_headers["Authorization"] = auth_header
metrics_response = api_session.post(
url=SNAP_PUB_METRICS_URL, headers=authed_metrics_headers, json=json
)
if authentication.is_macaroon_expired(metrics_response.headers):
raise MacaroonRefreshRequired
return process_response(metrics_response)
def snap_metadata(snap_id, session, json=None):
method = "PUT" if json is not None else None
metadata_response = api_session.request(
method=method,
url=METADATA_QUERY_URL.format(snap_id=snap_id),
headers=get_authorization_header(session),
json=json,
)
if authentication.is_macaroon_expired(metadata_response.headers):
raise MacaroonRefreshRequired
return process_response(metadata_response)
def post_username(session, username):
headers = get_authorization_header(session)
json = {"short_namespace": username}
username_response = api_session.patch(
url=ACCOUNT_URL, headers=headers, json=json
)
if authentication.is_macaroon_expired(username_response.headers):
raise MacaroonRefreshRequired
if username_response.status_code == 204:
return {}
else:
return process_response(username_response)
def post_username(session, username):
headers = get_authorization_header(session)
json = {"short_namespace": username}
username_response = api_session.patch(url=ACCOUNT_URL,
headers=headers,
json=json)
if authentication.is_macaroon_expired(username_response.headers):
raise MacaroonRefreshRequired
if username_response.status_code == 204:
return {}
else:
return process_response(username_response)
def get_package_upload_macaroon(session, snap_name, channels):
json = {
"packages": [{"name": snap_name, "series": "16"}],
"permissions": ["package_upload"],
"channels": channels,
}
response = api_session.post(
url=ACL_URL, headers=get_authorization_header(session), json=json,
)
if authentication.is_macaroon_expired(response.headers):
raise MacaroonRefreshRequired
return process_response(response)
def post_username(session, username):
headers = get_authorization_header(session)
json = {'short_namespace': username}
username_response = cache.get(url=ACCOUNT_URL,
headers=headers,
json=json,
method='PATCH')
if authentication.is_macaroon_expired(username_response.headers):
raise MacaroonRefreshRequired()
if username_response.status_code == 204:
return {}
else:
return process_response(username_response)
