def store_category(category):
status_code = 200
error_info = {}
snaps_results = []
try:
snaps_results = api.get_category_items(category=category,
size=10,
page=1)["results"]
except (StoreApiError, ApiError) as api_error:
status_code, error_info = _handle_error(api_error)
# if the first snap (banner snap) doesn't have an icon, remove the last
# snap from the list to avoid a hanging snap (grid of 9)
if len(snaps_results) == 10 and snaps_results[0]["icon_url"] == "":
snaps_results = snaps_results[:-1]
for index in range(len(snaps_results)):
snaps_results[index] = logic.get_snap_banner_url(
snaps_results[index])
context = {
"category": category,
"has_featured": True,
"snaps": snaps_results,
"error_info": error_info,
}
return (
flask.render_template("store/_category-partial.html", **context),
status_code,
)
def test_get_snap_banner_no_banner(self):
snap_with_banner = {
"media": [
{"type": "icon", "url": "icon.png"},
{"type": "screenshot", "url": "screenshot.png"},
]
}
result = logic.get_snap_banner_url(snap_with_banner)
self.assertEqual(result.get("banner_url"), None)
def store_view():
error_info = {}
status_code = 200
try:
categories_results = api.get_categories()
except ApiError as api_error:
categories_results = []
status_code, error_info = _handle_errors(api_error)
categories = logic.get_categories(categories_results)
try:
featured_snaps_results = api.get_searched_snaps(
snap_searched="", category="featured", size=10, page=1)
except ApiError:
featured_snaps_results = []
featured_snaps = logic.get_searched_snaps(featured_snaps_results)
# if the first snap (banner snap) doesn't have an icon, remove the last
# snap from the list to avoid a hanging snap (grid of 9)
if len(featured_snaps) == 10 and featured_snaps[0]["icon_url"] == "":
featured_snaps = featured_snaps[:-1]
featured_banners_enabled = flask.current_app.config.get(
"FEATURED_BANNERS_ENABLED")
if featured_banners_enabled:
for index in range(len(featured_snaps)):
featured_snaps[index] = logic.get_snap_banner_url(
featured_snaps[index])
livestream = snapcraft_logic.get_livestreams()
return (
flask.render_template(
"store/store.html",
categories=categories,
has_featured=True,
featured_snaps=featured_snaps,
error_info=error_info,
livestream=livestream,
),
status_code,
)
def store_view():
error_info = {}
status_code = 200
try:
categories_results = api.get_categories()
except StoreApiError:
categories_results = []
categories = logic.get_categories(categories_results)
try:
featured_snaps_results = api.get_featured_items()
except (StoreApiError, ApiError) as api_error:
status_code, error_info = _handle_error(api_error)
return flask.abort(status_code)
featured_snaps = logic.get_searched_snaps(featured_snaps_results)
if not featured_snaps:
return flask.abort(503)
# if the first snap (banner snap) doesn't have an icon, remove the last
# snap from the list to avoid a hanging snap (grid of 9)
if len(featured_snaps) == 10 and featured_snaps[0]["icon_url"] == "":
featured_snaps = featured_snaps[:-1]
for index in range(len(featured_snaps)):
featured_snaps[index] = logic.get_snap_banner_url(
featured_snaps[index]
)
livestream = snapcraft_logic.get_livestreams()
return (
flask.render_template(
"store/store.html",
categories=categories,
has_featured=True,
featured_snaps=featured_snaps,
error_info=error_info,
livestream=livestream,
),
status_code,
)
def store_category(category):
status_code = 200
error_info = {}
category_results = []
try:
category_results = api.get_searched_snaps(snap_searched="",
category=category,
size=10,
page=1)
except ApiError as api_error:
status_code, error_info = _handle_errors(api_error)
snaps_results = logic.get_searched_snaps(category_results)
# if the first snap (banner snap) doesn't have an icon, remove the last
# snap from the list to avoid a hanging snap (grid of 9)
if len(snaps_results) == 10 and snaps_results[0]["icon_url"] == "":
snaps_results = snaps_results[:-1]
featured_banners_enabled = flask.current_app.config.get(
"FEATURED_BANNERS_ENABLED")
if featured_banners_enabled:
for index in range(len(snaps_results)):
snaps_results[index] = logic.get_snap_banner_url(
snaps_results[index])
context = {
"category": category,
"has_featured": True,
"snaps": snaps_results,
"error_info": error_info,
}
return (
flask.render_template("store/_category-partial.html", **context),
status_code,
)
def search_snap():
snap_searched = flask.request.args.get("q", default="", type=str)
snap_category = flask.request.args.get("category",
default="",
type=str)
page = flask.request.args.get("page", default=1, type=int)
if snap_category:
snap_category_display = snap_category.capitalize().replace(
"-", " ")
else:
snap_category_display = None
if not snap_searched and not snap_category:
return flask.redirect(flask.url_for(".homepage"))
# The default size is 44
# (11 rows of 4) - on search results pages
# or (1 + 5 rows of 3 + 7 rows of 4) - on category page 1
size = 44
try:
searched_results = api.search(
snap_searched,
category=snap_category,
size=size,
page=page,
)
except (StoreApiError, ApiError) as api_error:
status_code, error_info = _handle_error(api_error)
return flask.abort(status_code)
total_pages = None
if "total" in searched_results:
total_results_count = searched_results["total"]
total_pages = ceil(total_results_count / size)
else:
total_results_count = None
snaps_results = searched_results["results"]
links = {}
if page > 1:
links["first"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=1,
)
links["prev"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=page - 1,
)
if not total_pages or page < total_pages:
links["next"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=page + 1,
)
if total_pages:
links["last"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=total_pages,
)
featured_snaps = []
# These are the hand-selected "featured snaps" in each category.
# We don't have this information on the API, so it's hardcoded.
number_of_featured_snaps = 16
if snap_category_display and page == 1:
if snaps_results and snaps_results[0]:
if snaps_results[0]["icon_url"] == "":
snaps_results = logic.promote_snap_with_icon(snaps_results)
snaps_results[0] = logic.get_snap_banner_url(snaps_results[0])
if (snap_category == "featured"
or len(snaps_results) < number_of_featured_snaps):
featured_snaps = snaps_results
snaps_results = []
else:
featured_snaps = snaps_results[:number_of_featured_snaps]
snaps_results = snaps_results[number_of_featured_snaps:]
context = {
"query": snap_searched,
"category": snap_category,
"category_display": snap_category_display,
"searched_snaps": snaps_results,
"featured_snaps": featured_snaps,
"total": total_results_count,
"links": links,
"page": page,
}
return flask.render_template("store/search.html", **context)
def search_snap():
status_code = 200
snap_searched = flask.request.args.get("q", default="", type=str)
snap_category = flask.request.args.get(
"category", default="", type=str
)
page = flask.request.args.get("page", default=1, type=int)
if snap_category:
snap_category_display = snap_category.capitalize().replace(
"-", " "
)
else:
snap_category_display = None
if not snap_searched and not snap_category:
return flask.redirect(flask.url_for(".homepage"))
# The default size should be 44 (rows of 4)
# it's important that this is smaller than the category page 1 size
# below otherwise snaps can be missed out of results
size = 44
# Page 1 has a snap at the top, and a few rows of 3, followed by rows
# of 4 - so we need to offset to ensure there's no hanging snap
if snap_category and page == 1:
size = 47
error_info = {}
searched_results = []
try:
searched_results = api.get_searched_snaps(
quote_plus(snap_searched),
category=snap_category,
size=size,
page=page,
)
except ApiError as api_error:
status_code, error_info = _handle_errors(api_error)
total_pages = None
if "total" in searched_results:
total_results_count = searched_results["total"]
total_pages = ceil(total_results_count / size)
else:
total_results_count = None
snaps_results = logic.get_searched_snaps(searched_results)
links = {}
if page > 1:
links["first"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=1,
)
links["prev"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=page - 1,
)
if not total_pages or page < total_pages:
links["next"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=page + 1,
)
if total_pages:
links["last"] = logic.build_pagination_link(
snap_searched=snap_searched,
snap_category=snap_category,
page=total_pages,
)
featured_snaps = []
# These are the hand-selected "featured snaps" in each category.
# We don't have this information on the API, so it's hardcoded.
number_of_featured_snaps = 19
if snap_category_display and page == 1:
if snaps_results and snaps_results[0]:
if snaps_results[0]["icon_url"] == "":
snaps_results = logic.promote_snap_with_icon(snaps_results)
snaps_results[0] = logic.get_snap_banner_url(snaps_results[0])
if (
snap_category == "featured"
or len(snaps_results) < number_of_featured_snaps
):
featured_snaps = snaps_results
snaps_results = []
else:
featured_snaps = snaps_results[:number_of_featured_snaps]
snaps_results = snaps_results[number_of_featured_snaps:]
context = {
"query": snap_searched,
"category": snap_category,
"category_display": snap_category_display,
"searched_snaps": snaps_results,
"featured_snaps": featured_snaps,
"total": total_results_count,
"links": links,
"page": page,
"error_info": error_info,
}
return (
flask.render_template("store/search.html", **context),
status_code,
)
