def main():
parser = argparse.ArgumentParser()
parser.add_argument("--send", help="dispatches e-mails to service providers, if not set will just write a report out",
action="store_true")
parser.add_argument("--infolder", type=str, help="folder to load from, defaults to {}".format(INPUT_DIR))
parser.add_argument("--infilename", type=str, help="file to load from, defaults to {}".format(INPUT_FILE_NAME))
parser.add_argument("--outfolder", type=str, help="folder to write to, defaults to {}".format(OUTPUT_DIR))
parser.add_argument("--outfilename", type=str, help="file to write to, defaults to {}".format(OUTPUT_FILE_NAME))
args = parser.parse_args()
if args.send:
print "--send flag has been set so SENDING E-MAILS TO SERVICE PROVIDERS"
# set up for output file if specified
in_folder_name = ""
if args.infolder:
in_folder_name = args.infolder
else:
in_folder_name = INPUT_DIR
in_file_name = ""
if args.infilename:
in_file_name = args.infilename
else:
in_file_name = INPUT_FILE_NAME
infile = os.path.join(in_folder_name, in_file_name)
print "using input data from file {}".format(infile)
# set up for output file if specified
out_folder_name = ""
if args.outfolder:
out_folder_name = args.outfolder
else:
out_folder_name = OUTPUT_DIR
out_file_name = ""
if args.outfilename:
out_file_name = args.outfilename
else:
out_file_name = OUTPUT_FILE_NAME
wb = open_workbook(infile)
for s in wb.sheets():
print 'found worksheet:', s.name
print
# make service provider map
provider_sheet = wb.sheet_by_name('Service providers')
sp_ids = provider_sheet.col_values(0,1)
sp_names = provider_sheet.col_values(1,1)
sp_primary_emails = provider_sheet.col_values(4,1)
providers = {}
i=0
for sp_id in sp_ids:
#print 'Service provider:', sp_id
sp = ServiceProvider()
sp.id = sp_id
sp.name = sp_names[i]
sp.primary_email = sp_primary_emails[i]
i=i+1
providers[sp_id] = sp
# for each activity in activities sheet, attach activity to appropriate provider
activity_sheet = wb.sheet_by_name('Activities')
act_ids = activity_sheet.col_values(0,1)
act_provider_ids = activity_sheet.col_values(1,1)
act_names = activity_sheet.col_values(3,1)
act_intros = activity_sheet.col_values(5,1)
act_start_dates = activity_sheet.col_values(7,1)
act_last_dates = activity_sheet.col_values(8,1)
act_start_times = activity_sheet.col_values(9,1)
act_recurrences = activity_sheet.col_values(11,1)
j=0
for activity_provider_id in act_provider_ids:
provider = providers.get(activity_provider_id)
if provider != None:
ac = Activity()
ac.id = act_ids[j]
ac.name = act_names[j]
ac.introduction = act_intros[j]
ac.time = act_start_times[j]
# work out what day the activity runs on from its start date
try:
start_date = datetime.strptime(act_start_dates[j], "%d/%m/%Y").date()
except TypeError as e:
exc_type, exc_value, exc_traceback = sys.exc_info()
print "Error converting activity start date at row {}. {}".format(j, str(e))
print "Try converting the date column into text and trying again. Also check that the year has century numbers i.e.; it is 2015 rather than 15"
traceback.print_exception(exc_type, exc_value, exc_traceback, limit=2, file=sys.stdout)
ac.day = find_day_of_week(start_date)
ends_on_date = datetime.strptime(act_last_dates[j], "%d/%m/%Y").date()
branch = {0 : generate_one_off_occurence_list,
1 : generate_next_weekly_occurence_list,
2 : generate_next_fortnightly_occurence_list,
3 : generate_next_day_of_month_occurence_list,
4 : generate_next_date_of_month_occurence_list}
next_dates = branch.get(act_recurrences[j], generate_default_occurence)(date.today(), start_date, ends_on_date)
HOW_MANY = 2
ac.next_dates = resize_next_dates(next_dates, HOW_MANY)
provider.addActivity(ac)
else:
print 'Activity id:', act_ids[j], ' has a provider_id of ', activity_provider_id, ' which does not match a service provider id'
j += 1
# if e-mail switch is on then generate and send e-mails, but only if the provider has some activities
if args.send:
for k in providers.keys():
provider = providers.get(k)
if len(provider.activities) > 0:
generate_email_to_provider(provider)
else:
# set up output for report
wb = Workbook()
ws = wb.add_sheet('Kids Connect output', cell_overwrite_ok=True)
# make header row
ws.row(0).write(0,'provider id')
ws.row(0).write(1,'provider name')
ws.row(0).write(2,'e-mail')
ws.row(0).write(3,'activity id')
ws.row(0).write(4,'activity name')
ws.row(0).write(5,'introduction')
ws.row(0).write(6,'day')
ws.row(0).write(7,'time')
ws.row(0).write(8,'next event 1')
ws.row(0).write(9,'next event 2')
row_num = 1
for k in providers.keys():
provider = providers.get(k)
print "processing provider {} {}".format(provider.id, provider.name)
for activity in provider.activities:
ws.row(row_num).write(0,provider.id)
ws.row(row_num).write(1,provider.name)
ws.row(row_num).write(2,provider.primary_email)
ws.row(row_num).write(3,activity.id)
ws.row(row_num).write(4,activity.name)
ws.row(row_num).write(5,activity.introduction)
ws.row(row_num).write(6,activity.day)
ws.row(row_num).write(7,activity.time)
if len(activity.next_dates) > 0:
event_col = 8
for next_event_date in activity.next_dates:
ws.row(row_num).write(event_col, next_event_date.strftime('%d-%m-%Y'))
event_col += 1
row_num += 1
output_file = os.path.join(out_folder_name, out_file_name)
wb.save(output_file)
print "writing output data to file {}".format(output_file)
print "finished!"
