Private
Public Access
1
0

Complete and tested email processing

Processed successfully individual files, zip files, unrecognized files,
zips containing unrecognized files
This commit is contained in:
Sander Roosendaal
2017-10-22 16:04:03 +02:00
parent f36911d73b
commit 973bbb9f0b
4 changed files with 78 additions and 51 deletions

View File

@@ -27,11 +27,11 @@ queuehigh = django_rq.get_queue('default')
# Sends a confirmation with a link to the workout
def send_confirm(u, name, link, options):
def send_confirm(user, name, link, options):
""" Send confirmation email to user when email has been processed """
fullemail = u.email
fullemail = user.email
subject = 'Workout added: ' + name
message = 'Dear ' + u.first_name + ',\n\n'
message = 'Dear ' + user.first_name + ',\n\n'
message += "Your workout has been added to Rowsandall.com.\n"
message += "Link to workout: " + link + "\n\n"
message += "Best Regards, the Rowsandall Team"
@@ -53,68 +53,73 @@ 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)
result = rrdata(file, rower=rower)
except IOError:
try:
res = rrdata(file + '.gz', rower=rower)
result = rrdata(file + '.gz', rower=rower)
except IOError:
res = 0
result = 0
return res
return result
def make_new_workout_from_email(rr, f2, name, cntr=0):
def make_new_workout_from_email(rower, datafile, name, cntr=0):
""" This one is used in processemail """
workouttype = 'rower'
try:
f2 = f2.name
fileformat = get_file_type('media/' + f2)
datafilename = datafile.name
fileformat = get_file_type('media/' + datafilename)
except IOError:
f2 = f2.name + '.gz'
fileformat = get_file_type('media/' + f2)
datafilename = datafile.name + '.gz'
fileformat = get_file_type('media/' + datafilename)
except AttributeError:
fileformat = get_file_type('media/' + f2)
datafilename = datafile
fileformat = get_file_type('media/' + datafile)
if len(fileformat) == 3 and fileformat[0] == 'zip':
f_to_be_deleted = f2
with zipfile.ZipFile('media/' + f2) as z:
f2 = z.extract(z.namelist()[0], path='media/')[6:]
with zipfile.ZipFile('media/' + datafilename) as zip_file:
datafilename = zip_file.extract(
zip_file.namelist()[0],
path='media/')[6:]
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)
# extension = datafilename[-4:].lower()
# fcopy = "media/"+datafilename[:-4]+"_copy"+extension
# with open('media/'+datafilename, 'r') as f_in, open(fcopy, 'w') as f_out:
# shutil.copyfileobj(f_in,f_out)
fcopy = "media/"+datafilename
if settings.DEBUG:
res = handle_sendemail_unrecognized.delay(fcopy,
"roosendaalsander@gmail.com")
rower.user.email)
else:
res = queuehigh.enqueue(handle_sendemail_unrecognized,
fcopy, "roosendaalsander@gmail.com")
fcopy,
rower.user.email)
return 0
summary = ''
# handle non-Painsled
if fileformat != 'csv':
f3, summary, oarlength, inboard = dataprep.handle_nonpainsled(
'media/' + f2, fileformat, summary)
filename_mediadir, summary, oarlength, inboard = dataprep.handle_nonpainsled(
'media/' + datafilename, fileformat, summary)
else:
f3 = 'media/' + f2
filename_mediadir = 'media/' + datafilename
inboard = 0.88
oarlength = 2.89
row = rdata(f3)
row = rdata(filename_mediadir)
if row == 0:
return 0
# change filename
if f2[:5] != 'media':
if datafilename[:5] != 'media':
timestr = time.strftime("%Y%m%d-%H%M%S")
f2 = 'media/' + timestr + str(cntr) + 'o.csv'
datafilename = 'media/' + timestr + str(cntr) + 'o.csv'
try:
avglat = row.df[' latitude'].mean()
@@ -124,19 +129,21 @@ def make_new_workout_from_email(rr, f2, name, cntr=0):
except KeyError:
pass
row.write_csv(f2, gzip=True)
row.write_csv(datafilename, gzip=True)
dosummary = (fileformat != 'fit')
if name == '':
name = 'imported through email'
id, message = dataprep.save_workout_database(f2, rr,
workouttype=workouttype,
dosummary=dosummary,
inboard=inboard,
oarlength=oarlength,
title=name,
workoutsource=fileformat,
notes='imported through email')
id, message = dataprep.save_workout_database(
datafilename, rower,
workouttype=workouttype,
dosummary=dosummary,
inboard=inboard,
oarlength=oarlength,
title=name,
workoutsource=fileformat,
notes='imported through email'
)
return id