view core/functions.py @ 816:97f8fab9b1a3

Tweaking the quoting algorithm a bit. Trying to reduce trailing lines that are just ">".
author Brian Neal <bgneal@gmail.com>
date Wed, 10 Sep 2014 19:21:59 -0500
parents e888d627928f
children 79a71b9d0a2a
line wrap: on
line source
"""This file houses various core utility functions"""
from contextlib import contextmanager
import datetime
import logging
import os
import re

from django.contrib.sites.models import Site
from django.conf import settings
import django.core.mail

import core.tasks


@contextmanager
def temp_open(path, mode):
    """A context manager for closing and removing temporary files."""
    fp = open(path, mode)
    try:
        yield fp
    finally:
        fp.close()
        os.remove(path)


def send_mail(subject, message, from_email, recipient_list, defer=True, **kwargs):
    """
    The main send email function. Use this function to send email from the
    site. All applications should use this function instead of calling
    Django's directly.
    If defer is True, the email will be sent to a Celery task to actually send
    the email. Otherwise it is sent on the caller's thread. In any event, the
    email will be logged at the DEBUG level.

    """
    # Guard against empty email addresses
    recipient_list = [dest for dest in recipient_list if dest]
    if not recipient_list:
        logging.warning("Empty recipient_list in send_mail")
        return

    logging.debug('EMAIL:\nFrom: %s\nTo: %s\nSubject: %s\nMessage:\n%s',
        from_email, str(recipient_list), subject, message)

    if defer:
        core.tasks.send_mail.delay(subject, message, from_email, recipient_list,
                **kwargs)
    else:
        django.core.mail.send_mail(subject, message, from_email, recipient_list,
                **kwargs)


def email_admins(subject, message):
    """Emails the site admins. Goes through the site send_mail function."""
    site = Site.objects.get_current()
    subject = '[%s] %s' % (site.name, subject)
    send_mail(subject,
            message,
            '%s@%s' % (settings.GPP_NO_REPLY_EMAIL, site.domain),
            [mail_tuple[1] for mail_tuple in settings.ADMINS])


def email_managers(subject, message):
    """Emails the site managers. Goes through the site send_mail function."""
    site = Site.objects.get_current()
    subject = '[%s] %s' % (site.name, subject)
    send_mail(subject,
            message,
            '%s@%s' % (settings.GPP_NO_REPLY_EMAIL, site.domain),
            [mail_tuple[1] for mail_tuple in settings.MANAGERS])


def get_full_name(user):
    """Returns the user's full name if available, otherwise falls back
    to the username."""
    full_name = user.get_full_name()
    if full_name:
        return full_name
    return user.username


BASE_YEAR = 2010

def copyright_str():
    curr_year = datetime.datetime.now().year
    if curr_year == BASE_YEAR:
        year_range = str(BASE_YEAR)
    else:
        year_range = "%d - %d" % (BASE_YEAR, curr_year)

    return 'Copyright (C) %s, SurfGuitar101.com' % year_range


IP_PAT = re.compile('(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})')

def get_ip(request):
    """Returns the IP from the request or None if it cannot be retrieved."""
    ip = request.META.get('HTTP_X_FORWARDED_FOR',
            request.META.get('REMOTE_ADDR'))

    if ip:
        match = IP_PAT.match(ip)
        ip = match.group(1) if match else None

    return ip


def get_page(qdict):
    """Attempts to retrieve the value for "page" from the given query dict and
    return it as an integer. If the key cannot be found or converted to an
    integer, 1 is returned.
    """
    n = qdict.get('page', 1)
    try:
        n = int(n)
    except ValueError:
        n = 1
    return n


def quote_message(who, message):
    """
    Builds a message reply by quoting the existing message in a
    typical email-like fashion. The quoting is compatible with Markdown.
    """
    msg = "> %s" % message.rstrip().replace('\n', '\n> ')
    if msg.endswith('\n> '):
        msg = msg[:-2]

    return "*%s wrote:*\n\n%s\n\n" % (who, msg)