74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
from celery import task, shared_task
|
|
from celery.utils.log import get_task_logger
|
|
|
|
import os
|
|
import sys
|
|
import datetime
|
|
import fileinput
|
|
import smtplib
|
|
from pathlib import Path
|
|
|
|
from email.feedparser import FeedParser
|
|
from email.message import EmailMessage
|
|
|
|
import django
|
|
from django.conf import settings
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'procat2.settings')
|
|
django.setup()
|
|
|
|
from .utils import clean_path, ensure_dir, set_file_perms, WORKDIR
|
|
from .email import reply, reply_missing
|
|
from .matching import find_marked_products
|
|
from .spreadsheet import write_spreadsheet
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
@shared_task
|
|
def process_message(path):
|
|
parser = FeedParser()
|
|
with open(path) as f:
|
|
for line in f:
|
|
parser.feed(line)
|
|
msg = parser.close()
|
|
|
|
found_pdf = False
|
|
if msg.is_multipart():
|
|
for attach in msg.walk():
|
|
if attach.get_content_type() == 'application/pdf':
|
|
process_attachment(msg['From'], msg['Subject'], attach)
|
|
found_pdf = True
|
|
|
|
if not found_pdf:
|
|
reply_missing(msg['From'], msg['Subject'])
|
|
|
|
|
|
def process_attachment(from_address, subject, attachment):
|
|
# write out pdf
|
|
pdf_name = attachment.get_filename()
|
|
pdf_base = Path(pdf_name).stem
|
|
|
|
workdir = os.path.join(WORKDIR, clean_path(from_address), pdf_base)
|
|
ensure_dir(workdir)
|
|
pdf_path = os.path.join(workdir, pdf_name)
|
|
print(f'saving pdf to {pdf_path}')
|
|
with open(pdf_path, 'wb') as att:
|
|
att.write(attachment.get_payload(decode=True))
|
|
set_file_perms(pdf_path)
|
|
|
|
# find matches
|
|
matches = find_marked_products(pdf_path, workdir, debug=0)
|
|
print(f'{len(matches)} product matches')
|
|
|
|
# write spreadsheet
|
|
xls_path = write_spreadsheet(matches, workdir, pdf_base)
|
|
|
|
if xls_path:
|
|
# send reply
|
|
print(f'wrote spreadsheet: {xls_path}')
|
|
reply(from_address, subject, xls_path, pdf_path)
|
|
else:
|
|
# send error
|
|
print(f'error creating spreadsheet')
|