campaigns: Decrease verbosity when sending emails

[ipfire.org.git] / src / backend / campaigns.py

#!/usr/bin/python3

import logging

from .decorators import *
from .misc import Object

class Campaigns(Object):
	async def launch_manually(self, name, uid):
		account = self.backend.accounts.get_by_uid(uid)
		if account:
			self.launch(name, account)

	def launch(self, name, account):
		logging.debug("Launching all campaigns for %s" % account)

		# Update old timestamps first
		self.db.execute("UPDATE campaign_templates \
			SET launch_at = launch_at + repeat_after \
			WHERE (launch_at IS NOT NULL AND launch_at <= CURRENT_TIMESTAMP) \
				AND repeat_after IS NOT NULL")

		# Launch all campaigns
		self.db.execute("INSERT INTO campaign_emails(account_uid, template, \
			launch_at, repeat_after) \
			SELECT %s, template, COALESCE(launch_at, CURRENT_TIMESTAMP + launch_after), \
				repeat_after FROM campaign_templates WHERE name = %s",
			account.uid, name)

	def _get_campaign_emails(self, query, *args):
		res = self.db.query(query, *args)

		for row in res:
			yield CampaignEmail(self.backend, row.id, data=row)

	async def run(self):
		with self.db.transaction():
			emails = self._get_campaign_emails("SELECT * FROM campaign_emails \
				WHERE launch_at <= CURRENT_TIMESTAMP ORDER BY launch_at")

			# Send them all
			for email in emails:
				email.send()


class CampaignEmail(Object):
	def init(self, id, data=None):
		self.id   = id
		self.data = data

	@lazy_property
	def account(self):
		return self.backend.accounts.get_by_uid(self.data.account_uid)

	@property
	def template(self):
		return self.data.template

	@property
	def repeat_after(self):
		return self.data.repeat_after

	def send(self):
		# Delete if the account does not exist any more
		if not self.account:
			return self._delete()

		logging.debug("Sending %s to %s" % (self.template, self.account))

		# Generate the email
		self.backend.messages.send_template(self.template, account=self.account)

		# Update this email for the next launch
		self._update()

	def _update(self):
		# If this email repeats, we will update the timestamp
		if self.repeat_after:
			self.db.execute("UPDATE campaign_emails \
				SET launch_at = launch_at + repeat_after \
				WHERE id = %s", self.id)

		# Otherwise we will delete it
		else:
			self._delete()

	def _delete(self):
		"""
			Deletes this email
		"""
		self.db.execute("DELETE FROM campaign_emails WHERE id = %s", self.id)