From 912b917a47c082fce902065f72d709f330d5153d Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Sat, 30 Jan 2021 19:49:15 +0000 Subject: [PATCH] Fix API packages with no arguments --- app/models/packages.py | 4 ++-- app/querybuilder.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/app/models/packages.py b/app/models/packages.py index 2fbdd50..f51b0a9 100644 --- a/app/models/packages.py +++ b/app/models/packages.py @@ -404,10 +404,10 @@ class Package(db.Model): "type": self.type.toName(), } - def getAsDictionaryShort(self, base_url, version=None, release_id=None): + def getAsDictionaryShort(self, base_url, version=None, release_id=None, no_load=False): tnurl = self.getThumbnailURL(1) - if release_id is None: + if release_id is None and no_load == False: release = self.getDownloadRelease(version=version) release_id = release and release.id diff --git a/app/querybuilder.py b/app/querybuilder.py index 7c049d5..ec754cc 100644 --- a/app/querybuilder.py +++ b/app/querybuilder.py @@ -89,8 +89,8 @@ class QueryBuilder: releases[package_id] = release_id def toJson(package: Package): - release_id = releases[package.id] - return package.getAsDictionaryShort(current_app.config["BASE_URL"], release_id=release_id) + release_id = releases.get(package.id) + return package.getAsDictionaryShort(current_app.config["BASE_URL"], release_id=release_id, no_load=True) return [toJson(pkg) for pkg in packages]