Files
postgres-web/pgweb/news/management/commands/twitter_post.py
2019-01-26 16:19:26 +01:00

50 lines
1.8 KiB
Python

#!/usr/bin/env python3
#
# Script to post previosly unposted news to twitter
#
#
from django.core.management.base import BaseCommand, CommandError
from django.db import connection
from django.conf import settings
from datetime import datetime, timedelta
import time
from pgweb.news.models import NewsArticle
import requests_oauthlib
class Command(BaseCommand):
help = 'Post to twitter'
def handle(self, *args, **options):
curs = connection.cursor()
curs.execute("SELECT pg_try_advisory_lock(62387372)")
if not curs.fetchall()[0][0]:
raise CommandError("Failed to get advisory lock, existing twitter_post process stuck?")
articles = list(NewsArticle.objects.filter(tweeted=False, approved=True, date__gt=datetime.now() - timedelta(days=7)).order_by('date'))
if not len(articles):
return
tw = requests_oauthlib.OAuth1Session(settings.TWITTER_CLIENT,
settings.TWITTER_CLIENTSECRET,
settings.TWITTER_TOKEN,
settings.TWITTER_TOKENSECRET)
for a in articles:
# We hardcode 30 chars for the URL shortener. And then 10 to cover the intro and spacing.
statusstr = "News: {0} {1}/about/news/{2}/".format(a.title[:140 - 40], settings.SITE_ROOT, a.id)
r = tw.post('https://api.twitter.com/1.1/statuses/update.json', data={
'status': statusstr,
})
if r.status_code != 200:
print("Failed to post to twitter: %s " % r)
else:
a.tweeted = True
a.save()
# Don't post more often than once / 30 seconds, to not trigger flooding.
time.sleep(30)