mirror of
https://github.com/postgres/pgweb.git
synced 2025-07-25 16:02:27 +00:00
50 lines
1.8 KiB
Python
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)
|