Complete and tested email processing
Processed successfully individual files, zip files, unrecognized files, zips containing unrecognized files
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user