def get_weekly_stats_update_projects():
"""Return email jobs with weekly stats update for project owner."""
from sqlalchemy.sql import text
from pybossa.core import db
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
only_pros = feature_handler.only_for_pro('project_weekly_report')
only_pros_sql = 'AND "user".pro=true' if only_pros else ''
send_emails_date = current_app.config.get('WEEKLY_UPDATE_STATS')
today = datetime.today().strftime('%A').lower()
timeout = current_app.config.get('TIMEOUT')
if today.lower() == send_emails_date.lower():
sql = text('''
SELECT project.id
FROM project, "user", task
WHERE "user".id=project.owner_id %s
AND "user".subscribed=true
AND task.project_id=project.id
AND task.state!='completed'
UNION
SELECT project.id
FROM project
WHERE project.featured=true;
''' % only_pros_sql)
results = db.slave_session.execute(sql)
for row in results:
job = dict(name=send_weekly_stats_project,
args=[row.id],
kwargs={},
timeout=timeout,
queue='low')
yield job
def notify_blog_users(blog_id, project_id, queue='high'):
"""Send email with new blog post."""
from sqlalchemy.sql import text
from pybossa.core import db
from pybossa.core import blog_repo
from pybossa.pro_features import ProFeatureHandler
blog = blog_repo.get(blog_id)
users = 0
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
only_pros = feature_handler.only_for_pro('notify_blog_updates')
timeout = current_app.config.get('TIMEOUT')
if blog.project.featured or (blog.project.owner.pro or not only_pros):
sql = text('''
SELECT email_addr, name from "user", task_run
WHERE task_run.project_id=:project_id
AND task_run.user_id="user".id
AND "user".subscribed=true
AND "user".restrict=false
GROUP BY email_addr, name, subscribed;
''')
results = db.slave_session.execute(sql, dict(project_id=project_id))
for row in results:
subject = "Project Update: %s by %s" % (blog.project.name,
blog.project.owner.fullname)
body = render_template('/account/email/blogupdate.md',
user_name=row.name,
blog=blog,
config=current_app.config)
html = render_template('/account/email/blogupdate.html',
user_name=row.name,
blog=blog,
config=current_app.config)
mail_dict = dict(recipients=[row.email_addr],
subject=subject,
body=body,
html=html)
job = dict(name=send_mail,
args=[mail_dict],
kwargs={},
timeout=timeout,
queue=queue)
enqueue_job(job)
users += 1
msg = "%s users notified by email" % users
return msg
def notify_blog_users(blog_id, project_id, queue='high'):
"""Send email with new blog post."""
from sqlalchemy.sql import text
from pybossa.core import db
from pybossa.core import blog_repo
from pybossa.pro_features import ProFeatureHandler
blog = blog_repo.get(blog_id)
users = 0
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
only_pros = feature_handler.only_for_pro('notify_blog_updates')
if blog.project.featured or (blog.project.owner.pro or not only_pros):
sql = text('''
SELECT email_addr, name from "user", task_run
WHERE task_run.project_id=:project_id
AND task_run.user_id="user".id
AND "user".subscribed=true
GROUP BY email_addr, name, subscribed;
''')
results = db.slave_session.execute(sql, dict(project_id=project_id))
for row in results:
subject = "Project Update: %s by %s" % (
blog.project.name, blog.project.owner.fullname)
body = render_template('/account/email/blogupdate.md',
user_name=row.name,
blog=blog,
config=current_app.config)
html = render_template('/account/email/blogupdate.html',
user_name=row.name,
blog=blog,
config=current_app.config)
mail_dict = dict(recipients=[row.email_addr],
subject=subject,
body=body,
html=html)
job = dict(name=send_mail,
args=[mail_dict],
kwargs={},
timeout=(10 * MINUTE),
queue=queue)
enqueue_job(job)
users += 1
msg = "%s users notified by email" % users
return msg
def get_export_task_jobs(queue):
"""Export tasks to zip."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get("PRO_FEATURES"))
if feature_handler.only_for_pro("updated_exports"):
if queue == "high":
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.get_all() if p.owner.pro is False)
else:
projects = (p.dictize() for p in project_repo.get_all())
for project in projects:
project_id = project.get("id")
job = dict(name=project_export, args=[project_id], kwargs={}, timeout=(10 * MINUTE), queue=queue)
yield job
def get_autoimport_jobs(queue='low'):
"""Get autoimport jobs."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
if feature_handler.only_for_pro('autoimporter'):
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.get_all())
for project_dict in projects:
project = project_repo.get(project_dict['id'])
if project.has_autoimporter():
job = dict(name=import_tasks,
args=[project.id],
kwargs=project.get_autoimporter(),
timeout=IMPORT_TASKS_TIMEOUT,
queue=queue)
yield job
def get_export_task_jobs(queue):
"""Export tasks to zip."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
if feature_handler.only_for_pro('updated_exports'):
if queue == 'high':
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.get_all()
if p.owner.pro is False)
else:
projects = (p.dictize() for p in project_repo.get_all())
for project in projects:
project_id = project.get('id')
job = dict(name=project_export,
args=[project_id], kwargs={},
timeout=(10 * MINUTE),
queue=queue)
yield job
def get_autoimport_jobs(queue='low'):
"""Get autoimport jobs."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
timeout = current_app.config.get('TIMEOUT')
if feature_handler.only_for_pro('autoimporter'):
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.get_all())
for project_dict in projects:
project = project_repo.get(project_dict['id'])
if project.has_autoimporter():
job = dict(name=import_tasks,
args=[project.id, True],
kwargs=project.get_autoimporter(),
timeout=timeout,
queue=queue)
yield job
def get_export_task_jobs(queue):
"""Export tasks to zip."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get('PRO_FEATURES'))
timeout = current_app.config.get('TIMEOUT')
if feature_handler.only_for_pro('updated_exports'):
if queue == 'high':
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.filter_by(published=True)
if p.owner.pro is False)
else:
projects = (p.dictize() for p in project_repo.filter_by(published=True))
for project in projects:
project_id = project.get('id')
job = dict(name=project_export,
args=[project_id], kwargs={},
timeout=timeout,
queue=queue)
yield job
def get_autoimport_jobs(queue="low"):
"""Get autoimport jobs."""
from pybossa.core import project_repo
import pybossa.cache.projects as cached_projects
from pybossa.pro_features import ProFeatureHandler
feature_handler = ProFeatureHandler(current_app.config.get("PRO_FEATURES"))
if feature_handler.only_for_pro("autoimporter"):
projects = cached_projects.get_from_pro_user()
else:
projects = (p.dictize() for p in project_repo.get_all())
for project_dict in projects:
project = project_repo.get(project_dict["id"])
if project.has_autoimporter():
job = dict(
name=import_tasks,
args=[project.id],
kwargs=project.get_autoimporter(),
timeout=IMPORT_TASKS_TIMEOUT,
queue=queue,
)
yield job
def test_only_for_pro_returns_False_if_feature_is_for_everyone(self):
pro_disabled_handler = ProFeatureHandler(self.config_disabled)
assert pro_disabled_handler.only_for_pro('auditlog') is False
def test_only_for_pro_returns_True_if_feature_is_only_for_pro(self):
pro_enabled_handler = ProFeatureHandler(self.config_enabled)
assert pro_enabled_handler.only_for_pro('auditlog') is True
def test_only_for_pro_returns_False_if_feature_is_for_everyone(self):
pro_disabled_handler = ProFeatureHandler(self.config_disabled)
assert pro_disabled_handler.only_for_pro('auditlog') is False
def test_only_for_pro_returns_True_if_feature_is_only_for_pro(self):
pro_enabled_handler = ProFeatureHandler(self.config_enabled)
assert pro_enabled_handler.only_for_pro('auditlog') is True
