gistfile1.py

from __future__ import print_function, unicode_literals

from optparse import make_option

#Import DB to close
from django import db

from django.core.management.base import NoArgsCommand

from mezzanine.twitter.models import Query, TwitterQueryException


class Command(NoArgsCommand):
    """
    Polls the Twitter API for tweets associated to the queries in templates.
    """

    option_list = NoArgsCommand.option_list + (
        make_option("--force", default=False, action="store_true"),
    )

    def handle_noargs(self, **options):
        queries = Query.objects.all()
        if not options["force"]:
            queries = queries.filter(interested=True)
        for query in queries:
            try:
                query.run()
            except TwitterQueryException as e:
                print("Twitter query error [%s]: %s" % (query, e))
        #Explicitly close DB connection.
        db.close_connection()