forked from forks/microblog.pub
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
|
import os
|
||
|
|
import logging
|
||
|
|
import random
|
||
|
|
|
||
|
|
import requests
|
||
|
|
from celery import Celery
|
||
|
|
from requests.exceptions import HTTPError
|
||
|
|
|
||
|
|
from config import HEADERS
|
||
|
|
from config import ID
|
||
|
|
from config import DB
|
||
|
|
from config import KEY
|
||
|
|
from config import USER_AGENT
|
||
|
|
from utils.httpsig import HTTPSigAuth
|
||
|
|
from utils.opengraph import fetch_og_metadata
|
||
|
|
|
||
|
|
|
||
|
|
log = logging.getLogger()
|
||
|
|
app = Celery('tasks', broker=os.getenv('MICROBLOGPUB_AMQP_BROKER', 'pyamqp://guest@localhost//'))
|
||
|
|
# app = Celery('tasks', broker='pyamqp://guest@rabbitmq//')
|
||
|
|
SigAuth = HTTPSigAuth(ID+'#main-key', KEY.privkey)
|
||
|
|
|
||
|
|
|
||
|
|
@app.task(bind=True, max_retries=12)
|
||
|
|
def post_to_inbox(self, payload, to):
|
||
|
|
try:
|
||
|
|
log.info('payload=%s', payload)
|
||
|
|
log.info('to=%s', to)
|
||
|
|
resp = requests.post(to, data=payload, auth=SigAuth, headers={
|
||
|
|
'Content-Type': HEADERS[1],
|
||
|
|
'Accept': HEADERS[1],
|
||
|
|
'User-Agent': USER_AGENT,
|
||
|
|
})
|
||
|
|
print(resp)
|
||
|
|
log.info('resp=%s', resp)
|
||
|
|
log.info('resp_body=%s', resp.text)
|
||
|
|
resp.raise_for_status()
|
||
|
|
except HTTPError as err:
|
||
|
|
log.exception('request failed')
|
||
|
|
if 400 >= err.response.status_code >= 499:
|
||
|
|
log.info('client error, no retry')
|
||
|
|
return
|
||
|
|
self.retry(exc=err, countdown=int(random.uniform(2, 4) ** self.request.retries))
|
||
|
|
|
||
|
|
|
||
|
|
@app.task(bind=True, max_retries=12)
|
||
|
|
def fetch_og(self, col, remote_id):
|
||
|
|
try:
|
||
|
|
log.info('fetch_og_meta remote_id=%s col=%s', remote_id, col)
|
||
|
|
if col == 'INBOX':
|
||
|
|
log.info('%d links saved', fetch_og_metadata(USER_AGENT, DB.inbox, remote_id))
|
||
|
|
elif col == 'OUTBOX':
|
||
|
|
log.info('%d links saved', fetch_og_metadata(USER_AGENT, DB.outbox, remote_id))
|
||
|
|
except Exception as err:
|
||
|
|
self.log.exception('failed')
|
||
|
|
self.retry(exc=err, countdown=int(random.uniform(2, 4) ** self.request.retries))
|