def render_package_information(self, package_name: str):
package = Package.find_by_name(package_name)
if package is None:
return abort(404, "package does not exist in repository")
folder = self.package_folder.joinpath(package_name)
payload = {
"name": package_name,
"packages": [],
"use_prompt": package.private and not self.is_pip,
}
if self.is_pip and package.private and not package.is_valid_token(
self.token):
return render_template('package-info.html', **payload)
for path in folder.iterdir():
filename = path.name
if path.is_file() and (filename.endswith('.whl')
or filename.endswith('.tar.gz')):
filepath = path.as_posix()
info = pkginfo.Wheel(filepath) if filename.endswith(
'.whl') else pkginfo.SDist(filepath)
payload["packages"].append({
"link":
f"/simple/{package_name}/{path.name}",
"title":
path.name,
"requires_python":
info.requires_python
})
return render_template('package-info.html', **payload)
def test_update_package(db_session):
pkg = Package.find_by_name(pkg1.name)
pkg.update(allow_override=True, private=True)
assert pkg.to_dict() == {
"name": pkg1.name,
"allow_override": True,
"private": True
}
def send_package(self, package_name: str, filename: str):
package = Package.find_by_name(package_name)
if package.private and not package.is_valid_token(self.token):
abort(
403,
"Not allowed to download private package without valid credentials"
)
return send_file(self.package_folder.joinpath(package_name,
filename).as_posix(),
attachment_filename=filename)
def get_project_details(project):
p = Package.find_by_name(project)
if p is None:
abort(404, f"Package/project '{project}' does not exist")
folder = PackageFolder(project)
return {
"name": project,
"latest_version": folder.latest_version,
"private": p.private,
"projects": folder.version_info_list
}
def upload_package():
account = get_account()
package_name = get_formatted_package_name()
package = Package.find_by_name(package_name)
if package is None:
package = Package(package_name, False, False)
account.packages.append(package)
account.save()
elif package.account != account:
abort(403, "Invalid credentials. Please ensure that you have the right credentials")
folder = current_app.config['PACKAGE_FOLDER']
return PostHandler(folder, package_name).upload_package(package.allow_override)
def test_add_lock():
pkg = Package.find_by_name(pkg1.name)
pkg.add_package_lock("random description for token")
assert len(pkg.to_dict(True)["locks"]) == 1
def test_find_package(db_session, name, is_none):
pkg = Package.find_by_name(name)
if is_none:
assert pkg is None
else:
assert pkg is not None
def test_delete_package_cascades(db_session):
pkg = Package.find_by_name(pkg1.name)
pkg.delete()
assert len(PackageLock.query.all()
) == 0, "all package locks should have been removed"