contentdb/app/logic/screenshots.py

import datetime, json

from flask_babel import lazy_gettext

from app.logic.LogicError import LogicError
from app.logic.uploads import upload_file
from app.models import User, Package, PackageScreenshot, Permission, NotificationType, db, AuditSeverity
from app.utils import addNotification, addAuditLog
from app.utils.image import get_image_size


def do_create_screenshot(user: User, package: Package, title: str, file, is_cover_image: bool, reason: str = None):
	thirty_minutes_ago = datetime.datetime.now() - datetime.timedelta(minutes=30)
	count = package.screenshots.filter(PackageScreenshot.created_at > thirty_minutes_ago).count()
	if count >= 20:
		raise LogicError(429, lazy_gettext("Too many requests, please wait before trying again"))

	uploaded_url, uploaded_path = upload_file(file, "image", lazy_gettext("a PNG or JPG image file"))

	counter = 1
	for screenshot in package.screenshots.all():
		screenshot.order = counter
		counter += 1

	ss = PackageScreenshot()
	ss.package  = package
	ss.title    = title or "Untitled"
	ss.url      = uploaded_url
	ss.approved = package.check_perm(user, Permission.APPROVE_SCREENSHOT)
	ss.order    = counter
	ss.width, ss.height = get_image_size(uploaded_path)

	if ss.is_too_small():
		raise LogicError(429,
				lazy_gettext("Screenshot is too small, it should be at least %(width)s by %(height)s pixels",
						width=PackageScreenshot.HARD_MIN_SIZE[0], height=PackageScreenshot.HARD_MIN_SIZE[1]))

	db.session.add(ss)

	if reason is None:
		msg = "Created screenshot {}".format(ss.title)
	else:
		msg = "Created screenshot {} ({})".format(ss.title, reason)

	addNotification(package.maintainers, user, NotificationType.PACKAGE_EDIT, msg, package.get_url("packages.view"), package)
	addAuditLog(AuditSeverity.NORMAL, user, msg, package.get_url("packages.view"), package)

	db.session.commit()

	if is_cover_image:
		package.cover_image = ss
		db.session.commit()

	return ss


def do_order_screenshots(_user: User, package: Package, order: [any]):
	lookup = {}
	for screenshot in package.screenshots.all():
		lookup[screenshot.id] = screenshot

	counter = 1
	for ss_id in order:
		try:
			lookup[int(ss_id)].order = counter
			counter += 1
		except KeyError as e:
			raise LogicError(400, "Unable to find screenshot with id={}".format(ss_id))
		except (ValueError, TypeError) as e:
			raise LogicError(400, "Invalid id, not a number: {}".format(json.dumps(ss_id)))

	db.session.commit()


def do_set_cover_image(_user: User, package: Package, cover_image):
	try:
		cover_image = int(cover_image)
	except (ValueError, TypeError) as e:
		raise LogicError(400, "Invalid id, not a number: {}".format(json.dumps(cover_image)))

	for screenshot in package.screenshots.all():
		if screenshot.id == cover_image:
			package.cover_image = screenshot
			db.session.commit()
			return

	raise LogicError(400, "Unable to find screenshot")