karmaworld/apps/notes/tasks.py

   1 #!/usr/bin/env python
   2 # -*- coding:utf8 -*-
   3 # Copyright (C) 2013  FinalsClub Foundation
   4
   5 from celery import task
   6 from karmaworld.apps.notes.models import Note
   7 import twitter
   8 import logging
   9 import gdshortener
  10
  11 logger = logging.getLogger(__name__)
  12
  13
  14 @task(name="tweet_note")
  15 def tweet_note():
  16     """Tweet about a new note."""
  17
  18     try:
  19         import karmaworld.secret.twitter as secrets
  20     except ImportError:
  21         logger.warn("No twitter secrets found, not running tweet_note")
  22         return
  23
  24     api = twitter.Api(consumer_key=secrets.CONSUMER_KEY,
  25                       consumer_secret=secrets.CONSUMER_SECRET,
  26                       access_token_key=secrets.ACCESS_TOKEN_KEY,
  27                       access_token_secret=secrets.ACCESS_TOKEN_SECRET)
  28
  29     newest_notes = Note.objects.all()[:100]
  30     for n in newest_notes:
  31         if not n.tweeted:
  32             update = tweet_string(n)
  33             logger.info("Tweeting: " + update)
  34
  35             # Mark this tweeted before we actually tweet it
  36             # to be extra safe against double tweets
  37             n.tweeted = True
  38             n.save()
  39
  40             api.PostUpdate(tweet_string(n))
  41
  42             break
  43
  44
  45 def tweet_string(note):
  46     # This url will use 12 or less characters
  47     shortener = gdshortener.VGDShortener()
  48     url = "https://www.karmanotes.org" + \
  49         note.get_absolute_url()
  50     short_url = shortener.shorten(url)[0]
  51
  52     # space character
  53
  54     # 16 characters
  55     school = note.course.school.slug
  56     short_school = school[:school.find('-')][:16]
  57
  58     # space character
  59
  60     # 50 characters
  61     short_course = note.course.name[:50]
  62
  63     # space and colon characters
  64
  65     # 57 characters
  66     short_note = note.name[:58]
  67
  68     return short_url + " #" + short_school + " " + \
  69         short_course + ": " + \
  70         short_note