forked from PhongT16/outlook_calendar_report_generator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
OutlookCalendar.py
159 lines (127 loc) · 7.09 KB
/
OutlookCalendar.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#!/usr/bin/python
import SharedCalendar
import argparse
from datetime import datetime
from SimpleEvent import SimpleEvent
from os import path
from datetime import timedelta
import time
import logging
from logging import handlers
import utils
from msal import PublicClientApplication
import IndividualCalendar
import GenerateReport
def process_args():
parser = argparse.ArgumentParser(
prog = 'vacation_calendar_sync',
formatter_class=argparse.RawDescriptionHelpFormatter,
description = 'Updates shared calendar among team members using each member\'s calendar with events marked as status \'away\'',
epilog =
'''
Program is controlled using the following environment variables:
VCS_CONFIG
path to the yaml configuration file
''')
parser.add_argument('-s', '--update_shared_calendar', action='store_true', help='Update shared calendar')
#parser.add_argument('-g', '--generate_report', action='store_true', help='Generate a report of the shared calendar')
parser.add_argument('-d', '--dump_json', action='store_true', help='Dump table data to console as json')
parser.add_argument('-g', '--generate_report', action='store', nargs=3, help="Generate a report to console of members OUT events: "+
"<group_name> <start_date> <end_date> with format YYYY-MM-DD")
parser.add_argument('-m', '--manual_update', action='store', nargs=2, help="Manually update the shared calendar with start and end time "+
"with format YYYY-MM-DD")
args = parser.parse_args()
return args
def sanitize_input(start_date, end_date):
"""
Sanitizes the user arguments to verify their validity
"""
# If the start_date and end_date given by user doesn't fit the format, then the datetime.strptime will
# throw its own error
start_date = datetime.strptime(start_date,"%Y-%m-%d")
end_date = datetime.strptime(end_date,"%Y-%m-%d")
# Check whether start date occurs before end_date
if (end_date - start_date).days < 0:
raise ValueError('start date should start prior to the end date')
return (start_date, end_date)
def retrieve_and_update_calendars(current_date, end_date, group_members, grouping, access_token):
logger.debug(f"{current_date} to {end_date}")
individual_calendars_events = []
# Create a list of lists in chunks of size grouping
for group in [group_members[i : i + grouping] for i in range(0, len(group_members), grouping)]:
individual_calendars = IndividualCalendar.get_individual_calendars(current_date, end_date, group, access_token)
individual_calendars_events.extend(IndividualCalendar.process_individual_calendars(individual_calendars, current_date, end_date))
# Retrieve the shared calendar and process it
shared_calendar_id = SharedCalendar.get_shared_calendar_id(configs['shared_calendar_name'], access_token)
shared_calendar = SharedCalendar.get_shared_calendar(shared_calendar_id, current_date, end_date, access_token)
shared_calendar_events, event_ids = SharedCalendar.process_shared_calendar(shared_calendar, group_members)
SharedCalendar.update_shared_calendar(individual_calendars_events, shared_calendar_events, event_ids, shared_calendar_id, configs['category_name'], configs['category_color'], access_token)
def main(configs):
args = process_args()
start_date = None
end_date = None
days_out = timedelta(days=configs['days_out'])
group_members = None
# Define the msal public client
app = PublicClientApplication(client_id=configs['client_id'], authority=f"https://login.microsoftonline.com/{configs['tenant_id']}")
if args.generate_report:
group_name = args.generate_report[0]
dates = sanitize_input(args.generate_report[1], args.generate_report[2])
start_date = dates[0]
end_date = dates[1]
access_token = utils.acquire_access_token(app, configs['scopes'])
current_date = start_date
emails = utils.get_email_list_from_ldap(group_name)
while (current_date + timedelta(14) <= end_date):
temp_end_date = current_date + timedelta(14)
GenerateReport.generate_report_for_specified_group(emails, current_date, temp_end_date, access_token)
current_date = temp_end_date
if (current_date < end_date):
GenerateReport.generate_report_for_specified_group(emails, current_date, end_date, access_token)
return
count = 0
while True:
#if args.update_shared_calendar or args.generate_report:
if args.update_shared_calendar:
logger.info(f"Updating shared calendar -> count: {count}")
today = datetime.today()
start_date = datetime(year=today.year, month=today.month, day=today.day, hour=0,minute=0)
end_date = start_date + days_out
elif args.manual_update:
logger.info(f"Running manually")
dates = sanitize_input(args.manual_update[0], args.manual_update[1])
start_date = dates[0]
end_date = dates[1]
# Retrieve the group member emails
group_members = utils.get_email_list(configs['group_name'], configs['email_list_update_interval'])
# Get access token
access_token = utils.acquire_access_token(app, configs['scopes'])
# Retrieve the individual calendar and process it
grouping = 10
current_date = start_date
while (current_date + timedelta(14) <= end_date):
temp_end_date = current_date + timedelta(14)
retrieve_and_update_calendars(current_date, temp_end_date, group_members, grouping, access_token)
current_date = temp_end_date
if (current_date < end_date):
retrieve_and_update_calendars(current_date, end_date, group_members, grouping, access_token)
if args.manual_update: break
count = count + 1
time.sleep(configs['update_interval'])
if __name__ == '__main__':
configs = utils.get_configurations()
formater = logging.Formatter('%(name)s:%(asctime)s:%(filename)s:%(levelname)s:%(message)s')
if not path.exists(configs['vcs_directory']):
raise KeyError(f"{configs['vcs_directory']} does not exist.")
debug_file = 'vcs_debug'
rotate_file_handler_info = handlers.RotatingFileHandler(f"{configs['vcs_directory']}{debug_file}.log", mode='a', maxBytes=2000000, backupCount=2)
rotate_file_handler_info .setFormatter(fmt=formater)
rotate_file_handler_info .setLevel(logging.DEBUG)
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
logger.addHandler(rotate_file_handler_info)
stream_handler = logging.StreamHandler()
stream_handler.setLevel(logging.DEBUG)
stream_handler.setFormatter(fmt=logging.Formatter('%(name)s:%(asctime)s:%(filename)s:%(levelname)s:%(message)s'))
logger.addHandler(stream_handler)
main(configs)