refactored, not tested
This commit is contained in:
Sander Roosendaal
@@ -1,24 +1,19 @@
|
||||
# Processes emails sent to workouts@rowsandall.com
|
||||
""" Processes emails sent to workouts@rowsandall.com """
|
||||
import shutil
|
||||
import time
|
||||
from django.conf import settings
|
||||
from rowers.tasks import handle_sendemail_unrecognized
|
||||
from django_mailbox.models import Mailbox, Message, MessageAttachment
|
||||
from rowers.models import Workout, User, Rower, WorkoutForm, RowerForm, GraphImage, AdvancedWorkoutForm
|
||||
from django.core.files.base import ContentFile
|
||||
from django.core.mail import send_mail, BadHeaderError, EmailMessage
|
||||
from rowsandall_app.settings import BASE_DIR
|
||||
from django_mailbox.models import Message, MessageAttachment
|
||||
from rowers.models import User, Rower, RowerForm
|
||||
|
||||
from django.core.mail import EmailMessage
|
||||
|
||||
from rowingdata import rower as rrower
|
||||
from rowingdata import main as rmain
|
||||
from rowingdata import rowingdata as rrdata
|
||||
from rowingdata import TCXParser, RowProParser, ErgDataParser
|
||||
from rowingdata import MysteryParser, BoatCoachParser
|
||||
from rowingdata import painsledDesktopParser, speedcoachParser, ErgStickParser
|
||||
from rowingdata import SpeedCoach2Parser, FITParser, fitsummarydata
|
||||
from rowingdata import make_cumvalues
|
||||
from rowingdata import summarydata, get_file_type
|
||||
|
||||
from scipy.signal import savgol_filter
|
||||
from rowingdata import rowingdata as rrdata
|
||||
|
||||
from rowingdata import get_file_type
|
||||
|
||||
import zipfile
|
||||
import os
|
||||
@@ -33,6 +28,7 @@ queuehigh = django_rq.get_queue('default')
|
||||
|
||||
|
||||
def send_confirm(u, name, link, options):
|
||||
""" Send confirmation email to user when email has been processed """
|
||||
fullemail = u.email
|
||||
subject = 'Workout added: ' + name
|
||||
message = 'Dear ' + u.first_name + ',\n\n'
|
||||
@@ -55,6 +51,7 @@ def send_confirm(u, name, link, options):
|
||||
|
||||
|
||||
def rdata(file, rower=rrower()):
|
||||
""" Reads rowingdata data or returns 0 on Error """
|
||||
try:
|
||||
res = rrdata(file, rower=rower)
|
||||
except IOError:
|
||||
@@ -65,119 +62,10 @@ def rdata(file, rower=rrower()):
|
||||
|
||||
return res
|
||||
|
||||
# Some error protection around process attachments
|
||||
|
||||
|
||||
def safeprocessattachments():
|
||||
try:
|
||||
return processattachments()
|
||||
except:
|
||||
return [0]
|
||||
|
||||
# This is duplicated in management/commands/processemail
|
||||
# Need to double check the code there, update here, and only
|
||||
# use the code here.
|
||||
|
||||
|
||||
def processattachments():
|
||||
# in res, we store the ids of the new workouts
|
||||
res = []
|
||||
attachments = MessageAttachment.objects.all()
|
||||
for a in attachments:
|
||||
donotdelete = 0
|
||||
m = Message.objects.get(id=a.message_id)
|
||||
from_address = m.from_address[0]
|
||||
name = m.subject
|
||||
|
||||
# get a list of users
|
||||
theusers = User.objects.filter(email=from_address)
|
||||
for u in theusers:
|
||||
try:
|
||||
rr = Rower.objects.get(user=u.id)
|
||||
# move attachment and make workout
|
||||
try:
|
||||
wid = [make_new_workout_from_email(rr, a.document, name)]
|
||||
res += wid
|
||||
link = 'https://rowsandall.com/rowers/workout/' + \
|
||||
str(wid[0]) + '/edit'
|
||||
if wid != 1:
|
||||
dd = send_confirm(u, name, link)
|
||||
except:
|
||||
# replace with code to process error
|
||||
res += ['fail: ' + name]
|
||||
donotdelete = 1
|
||||
except Rower.DoesNotExist:
|
||||
pass
|
||||
|
||||
# remove attachment
|
||||
if donotdelete == 0:
|
||||
a.delete()
|
||||
|
||||
if m.attachments.exists() == False:
|
||||
# no attachments, so can be deleted
|
||||
m.delete()
|
||||
|
||||
# Delete remaining messages (which should not have attachments)
|
||||
mm = Message.objects.all()
|
||||
for m in mm:
|
||||
if m.attachments.exists() == False:
|
||||
m.delete()
|
||||
|
||||
return res
|
||||
|
||||
# As above, but with some print commands for debugging purposes
|
||||
|
||||
|
||||
def processattachments_debug():
|
||||
res = []
|
||||
attachments = MessageAttachment.objects.all()
|
||||
for a in attachments:
|
||||
donotdelete = 1
|
||||
m = Message.objects.get(id=a.message_id)
|
||||
from_address = m.from_address[0]
|
||||
name = m.subject
|
||||
|
||||
# get a list of users
|
||||
theusers = User.objects.filter(email=from_address)
|
||||
print theusers
|
||||
for u in theusers:
|
||||
try:
|
||||
rr = Rower.objects.get(user=u.id)
|
||||
doorgaan = 1
|
||||
except:
|
||||
doorgaan = 0
|
||||
if doorgaan:
|
||||
# move attachment and make workout
|
||||
print a.document
|
||||
print name
|
||||
wid = [make_new_workout_from_email(rr, a.document, name)]
|
||||
res += wid
|
||||
link = 'https://rowsandall.com/rowers/workout/' + \
|
||||
str(wid[0]) + '/edit'
|
||||
if wid != 1:
|
||||
dd = send_confirm(u, name, link)
|
||||
|
||||
# remove attachment
|
||||
if donotdelete == 0:
|
||||
a.delete()
|
||||
|
||||
if m.attachments.exists() == False:
|
||||
# no attachments, so can be deleted
|
||||
m.delete()
|
||||
|
||||
mm = Message.objects.all()
|
||||
for m in mm:
|
||||
if m.attachments.exists() == False:
|
||||
m.delete()
|
||||
|
||||
return res
|
||||
|
||||
# Process the attachment file, create new workout
|
||||
# The code here is duplication of the code in views.py (workout_upload_view)
|
||||
# Need to move the code to a subroutine used both in views.py and here
|
||||
|
||||
|
||||
def make_new_workout_from_email(rr, f2, name, cntr=0):
|
||||
""" This one is used in processemail """
|
||||
workouttype = 'rower'
|
||||
|
||||
try:
|
||||
@@ -196,15 +84,18 @@ def make_new_workout_from_email(rr, f2, name, cntr=0):
|
||||
fileformat = fileformat[2]
|
||||
|
||||
if fileformat == 'unknown':
|
||||
fcopy = "copy_of_"+f2
|
||||
with open(f2, 'r') as f_in, open(fcopy, 'w') as f_out:
|
||||
shutil.copyfileobj(f_in,f_out)
|
||||
if settings.DEBUG:
|
||||
res = handle_sendemail_unrecognized.delay(f2,
|
||||
res = handle_sendemail_unrecognized.delay(fcopy,
|
||||
"roosendaalsander@gmail.com")
|
||||
|
||||
else:
|
||||
res = queuehigh.enqueue(handle_sendemail_unrecognized,
|
||||
f2, "roosendaalsander@gmail.com")
|
||||
fcopy, "roosendaalsander@gmail.com")
|
||||
|
||||
return 1
|
||||
return 0
|
||||
|
||||
summary = ''
|
||||
# handle non-Painsled
|
||||
@@ -216,11 +107,7 @@ def make_new_workout_from_email(rr, f2, name, cntr=0):
|
||||
inboard = 0.88
|
||||
oarlength = 2.89
|
||||
|
||||
# make workout and put in database
|
||||
# r = rrower(hrmax=rr.max,hrut2=rr.ut2,
|
||||
# hrut1=rr.ut1,hrat=rr.at,
|
||||
# hrtr=rr.tr,hran=rr.an,ftp=r.ftp)
|
||||
row = rdata(f3) # ,rower=r)
|
||||
row = rdata(f3)
|
||||
if row == 0:
|
||||
return 0
|
||||
|
||||
|
||||
Reference in New Issue
Block a user