mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
from datetime import datetime, timedelta
|
|
from urllib2 import HTTPError, URLError
|
|
|
|
from django.conf import settings
|
|
from django.utils.timezone import now
|
|
|
|
from celery.task import Task, task
|
|
|
|
from reddit.models import RedditAccount
|
|
from reddit.api import Inbox, LoginError, Flair
|
|
|
|
|
|
class send_reddit_message(Task):
|
|
|
|
default_retry_delay = 5 * 60 # retry in 5 minutes
|
|
ignore_result = True
|
|
|
|
def run(self, to, subject, message, **kwargs):
|
|
logger = self.get_logger(**kwargs)
|
|
|
|
logger.info("Sending Reddit message to %s" % to)
|
|
try:
|
|
ib = Inbox(username=settings.REDDIT_USER, password=settings.REDDIT_PASSWORD)
|
|
ib.send(to, subject, message)
|
|
except (HTTPError, URLError), exc:
|
|
logger.error("Error sending message, queueing for retry")
|
|
send_reddit_message.retry(args=[to, subject, message], kwargs=kwargs, exc=exc)
|
|
pass
|
|
except LoginError, exc:
|
|
logger.error("Error logging into Reddit")
|
|
|
|
|
|
@task(ignore_result=True)
|
|
def process_validations():
|
|
logger = process_validations.get_logger()
|
|
try:
|
|
inbox = Inbox(settings.REDDIT_USER, settings.REDDIT_PASSWORD)
|
|
for msg in inbox:
|
|
if not msg.was_comment:
|
|
try:
|
|
acc = RedditAccount.objects.get(username__iexact=msg.author)
|
|
if not acc.validated and msg.subject == "Validation: %s" % acc.user.username:
|
|
logger.info("Validated %s" % acc.user.username)
|
|
acc.validated = True
|
|
acc.save()
|
|
except RedditAccount.DoesNotExist:
|
|
continue
|
|
except (HTTPError, URLError), exc:
|
|
logger.error("Error with Reddit, aborting.")
|
|
return
|
|
except LoginError, exc:
|
|
logger.error("Error logging into Reddit")
|
|
return
|
|
|
|
|
|
@task(ignore_result=True)
|
|
def update_account(username):
|
|
|
|
logger = process_validations.get_logger()
|
|
|
|
try:
|
|
acc = RedditAccount.objects.get(pk=username)
|
|
acc.api_update()
|
|
except RedditAccount.DoesNotExist:
|
|
pass
|
|
else:
|
|
acc.save()
|
|
|
|
|
|
@task(ignore_result=True, expires=120)
|
|
def queue_account_updates(update_delay=604800, batch_size=50):
|
|
"""
|
|
Updates all Reddit accounts in the system
|
|
"""
|
|
|
|
log = queue_account_updates.get_logger()
|
|
# Update all the eve accounts and related corps
|
|
delta = timedelta(seconds=update_delay)
|
|
log.info("Updating Accounts older than %s" % (now() - delta))
|
|
accounts = RedditAccount.objects.order_by('last_update').filter(last_update__lt=(now() - delta))[:batch_size]
|
|
log.info("%s account(s) to update" % accounts.count())
|
|
for acc in accounts:
|
|
log.debug("Queueing Account %s for update" % acc.username)
|
|
if not acc.user:
|
|
acc.delete()
|
|
continue
|
|
update_account.delay(username=acc.pk)
|
|
|
|
|
|
class update_user_flair(Task):
|
|
"""
|
|
Updates a user's flair on Reddit
|
|
"""
|
|
|
|
default_retry_delay = 5 * 60 # retry in 5 minutes
|
|
ignore_result = True
|
|
|
|
def run(self, username, character_name, **kwargs):
|
|
try:
|
|
ib = Flair(username=settings.REDDIT_USER, password=settings.REDDIT_PASSWORD)
|
|
ib.set_flair(settings.REDDIT_SUBREDDIT, username, character_name, '')
|
|
except (HTTPError, URLError), exc:
|
|
logger.error("Error updating flair, queueing for retry")
|
|
update_user_flair.retry(args=[username, character_name], kwargs=kwargs, exc=exc)
|
|
pass
|
|
except LoginError, exc:
|
|
logger.error("Error logging into Reddit")
|
|
|