Mercurial > hg > gcmultimerge
annotate multimerge.py @ 52:b87dbc887f63
Simplify.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Tue, 05 Jul 2016 19:26:42 +0300 |
parents | 54644b29a9a3 |
children | ea62e0ed05ae |
rev | line source |
---|---|
1 | 1 #!/usr/bin/python |
3 | 2 # coding=utf-8 |
2 | 3 ### |
36 | 4 ### Google Calendar MultiMerge |
2 | 5 ### (C) 2016 Matti 'ccr' Hamalainen <ccr@tnsp.org> |
6 ### | |
7 ### Python 2.7 <= x < 3 required! Please refer to | |
8 ### README.txt for information on other depencies. | |
9 ### | |
1 | 10 import os |
11 import sys | |
12 import signal | |
13 import re | |
45
035be8a9e982
Force reading of configuration in Unicode UTF-8.
Matti Hamalainen <ccr@tnsp.org>
parents:
44
diff
changeset
|
14 import codecs |
29
50711871fd1e
Comment out currently unnecessary imports.
Matti Hamalainen <ccr@tnsp.org>
parents:
28
diff
changeset
|
15 #import time |
50711871fd1e
Comment out currently unnecessary imports.
Matti Hamalainen <ccr@tnsp.org>
parents:
28
diff
changeset
|
16 #import datetime |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
17 |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
18 import smtplib |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
19 from email.mime.text import MIMEText |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
20 |
1 | 21 import httplib2 |
22 import ConfigParser | |
20 | 23 |
1 | 24 import oauth2client |
25 from oauth2client import client | |
26 from oauth2client import tools | |
4 | 27 from oauth2client import file |
1 | 28 from googleapiclient import discovery |
29 | |
30 | |
31 ### | |
32 ### Misc. helper functions | |
33 ### | |
34 | |
35 ## Wrapper for print() that does not break when redirecting stdin/out | |
36 ## because of piped output not having a defined encoding. We default | |
37 ## to UTF-8 encoding in output here. | |
38 def gcm_print(smsg): | |
47 | 39 gcm_msgbuf.append(smsg) |
1 | 40 if sys.stdout.encoding != None: |
41 print(smsg.encode(sys.stdout.encoding)) | |
42 else: | |
43 print(smsg.encode("UTF-8")) | |
44 | |
45 | |
46 ## Fatal errors | |
47 def gcm_fatal(smsg): | |
48 gcm_print(u"ERROR: "+ smsg) | |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
49 if cfg.email_ok and cfg.email: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
50 ## If e-mail is set, send e-mail |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
51 msg = MIMEText(("\n".join(gcm_msgbuf)).encode("UTF-8"), "plain") |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
52 msg.set_charset("UTF-8") |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
53 msg["Subject"] = cfg.email_subject |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
54 msg["From"] = cfg.email_sender |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
55 msg["To"] = ",".join(cfg.email_to) |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
56 try: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
57 smtpH = smtplib.SMTP('localhost') |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
58 smtpH.sendmail(cfg.email_sender, cfg.email_to, msg.as_string()) |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
59 smtpH.quit() |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
60 except: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
61 gcm_print("FATAL: Oh crap, e-mail sending failed.") |
1 | 62 sys.exit(1) |
63 | |
64 | |
65 ## Debug messages | |
66 def gcm_debug(smsg): | |
67 if cfg.debug: | |
68 gcm_print(u"DBG: "+ smsg) | |
69 else: | |
47 | 70 gcm_msgbuf.append(u"DBG: {0}".format(smsg)) |
1 | 71 |
72 | |
24 | 73 ## Handler for SIGINT signals |
1 | 74 def gcm_signal_handler(signal, frame): |
75 gcm_print("\nQuitting due to SIGINT / Ctrl+C!") | |
76 sys.exit(0) | |
77 | |
78 | |
24 | 79 ## Function for handling Google API credentials |
1 | 80 def gcm_get_credentials(mcfg): |
42 | 81 try: |
82 store = oauth2client.file.Storage(mcfg.credential_file) | |
83 except Exception as e: | |
84 gcm_fatal("Failed to read credential file:\n{0}\n\nERROR: {1}\n".format(mcfg.credential_file, str(e))) | |
85 | |
1 | 86 credentials = store.get() |
87 if not credentials or credentials.invalid: | |
42 | 88 try: |
89 flow = client.flow_from_clientsecrets(mcfg.secret_file, mcfg.scope) | |
90 except Exception as e: | |
91 gcm_fatal("Failed to fetch client secret:\n{0}\n\nERROR: {1}\n".format(mcfg.secret_file, str(e))) | |
92 | |
1 | 93 flow.user_agent = mcfg.app_name |
94 credentials = tools.run_flow(flow, store, mcfg) | |
95 if not credentials or credentials.invalid: | |
96 gcm_fatal("Failed to authenticate / invalid credentials.") | |
97 return credentials | |
98 | |
99 | |
100 def gcm_dump_events(events): | |
101 for event in events: | |
102 ev_start = event["start"].get("dateTime", event["start"].get("date")) | |
103 ev_end = event["end"].get("dateTime", event["end"].get("date")) | |
104 gcm_print(u"{0:25} - {1:25} : {2}".format(ev_start, ev_end, event["summary"])) | |
105 | |
26
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
106 def gcm_generate_ids(events, calendar_id): |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
107 if not events: |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
108 return events |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
109 |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
110 for ev in events: |
28 | 111 ev["gcm_cal_id"] = calendar_id |
30 | 112 ev["gcm_id"] = re.sub("[^a-v0-9]", "0", calendar_id.lower()) + ev["id"] |
26
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
113 |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
114 return events |
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
115 |
1 | 116 |
24 | 117 ## |
118 ## Class for handling configuration / settings | |
119 ## | |
1 | 120 class GCMSettings(dict): |
121 def __init__(self): | |
122 self.m_data = {} | |
123 self.m_saveable = {} | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
124 self.m_validate = {} |
1 | 125 self.m_translate = {} |
126 | |
127 def __getattr__(self, name): | |
128 if name in self.m_data: | |
129 return self.m_data[name] | |
130 else: | |
131 gcm_fatal("GCMSettings.__getattr__(): No such attribute '"+ name +"'.") | |
132 | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
133 def mvalidate(self, name, value): |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
134 if name in self.m_validate and self.m_validate[name]: |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
135 if not self.m_validate[name](value): |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
136 gcm_fatal("GCMSettings.mvalidate(): Invalid value for attribute '{0}': {1}".format(name, value)) |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
137 |
1 | 138 def mtranslate(self, name, value): |
139 if name in self.m_translate and self.m_translate[name]: | |
140 return self.m_translate[name](value) | |
141 else: | |
142 return value | |
143 | |
144 def mdef(self, name, saveable, validate, translate, value): | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
145 self.mvalidate(name, value) |
1 | 146 self.m_saveable[name] = saveable |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
147 self.m_validate[name] = validate |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
148 self.m_translate[name] = translate |
1 | 149 self.m_data[name] = self.mtranslate(name, value) |
150 | |
151 def mset(self, name, value): | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
152 self.mvalidate(name, value) |
1 | 153 if name in self.m_data: |
154 self.m_data[name] = self.mtranslate(name, value) | |
155 else: | |
43 | 156 gcm_fatal(u"GCMSettings.mset(): No such attribute '"+ name +"'.") |
1 | 157 |
158 def mget(self, name): | |
159 if name in self.m_data: | |
160 return self.m_data[name] | |
161 else: | |
162 return None | |
163 | |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
164 def mread(self, cfgparser, sect): |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
165 for name in self.m_saveable: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
166 if cfgparser.has_option(sect, name): |
46
51c87abe5a1e
This Unicode conversion here is probably redundant.
Matti Hamalainen <ccr@tnsp.org>
parents:
45
diff
changeset
|
167 value = cfgparser.get(sect, name) |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
168 self.mset(name, value) |
43 | 169 gcm_debug(u"{0} -> '{1}' == {2}".format(name, value, self.mget(name))) |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
170 |
23 | 171 def is_str(self, mvalue): |
172 return isinstance(mvalue, basestring) | |
173 | |
174 def is_string(self, mvalue): | |
175 return mvalue == None or self.is_str(mvalue) | |
176 | |
177 def is_log_level(self, mvalue): | |
178 if not self.is_str(mvalue): | |
179 return False | |
180 else: | |
181 return mvalue.upper() in ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"] | |
182 | |
183 def trans_log_level(self, mvalue): | |
184 return mvalue.upper() | |
185 | |
186 def is_filename(self, mvalue): | |
187 if not self.is_str(mvalue): | |
188 return False | |
189 else: | |
190 return re.match("^[a-z0-9][a-z0-9\.\_\-]+$", mvalue, flags=re.IGNORECASE) | |
191 | |
192 def trans_bool(self, mvalue): | |
193 if self.is_str(mvalue): | |
194 if re.match("^\s*(true|1|on|yes)\s*$", mvalue, re.IGNORECASE): | |
195 mvalue = True | |
196 elif re.match("^\s*(false|0|off|no)\s*$", mvalue, re.IGNORECASE): | |
197 mvalue = False | |
198 else: | |
199 return None | |
200 return mvalue | |
201 | |
202 def is_bool(self, mvalue): | |
203 mval = self.trans_bool(mvalue) | |
204 if not isinstance(mval, bool): | |
205 gcm_fatal("GCMSettings.is_bool(): Invalid boolean value '{0}', should be true|false|1|0|on|off|yes|no.".format(mvalue)) | |
206 else: | |
207 return True | |
208 | |
209 def trans_list(self, mvalue): | |
210 morig = mvalue | |
211 if self.is_str(mvalue): | |
212 mvalue = re.split("\s*,\s*", mvalue, flags=re.IGNORECASE) | |
213 if not isinstance(mvalue, list): | |
214 gcm_fatal("GCMSettings.trans_list(): Could not parse list '{0}'.".format(mvalue)) | |
215 elif not isinstance(mvalue, list): | |
216 gcm_fatal("GCMSettings.trans_list(): Invalid value '{0}'.".format(mvalue)) | |
217 return mvalue | |
218 | |
219 def is_list(self, mvalue): | |
220 return self.trans_list(mvalue) | |
221 | |
222 def is_email(self, mvalue): | |
223 if not self.is_string(mvalue): | |
224 return False | |
225 else: | |
226 return re.match("^.*?\s+<[a-z0-9]+[a-z0-9\.\+\-]*\@[a-z0-9]+[a-z0-9\.\-]+>\s*$|[a-z0-9]+[a-z0-9\.\+\-]*\@[a-z0-9]+[a-z0-9\.\-]+", mvalue, flags=re.IGNORECASE) | |
227 | |
228 def trans_email_list(self, mvalue): | |
229 if mvalue == None: | |
230 return mvalue | |
231 else: | |
232 return self.trans_list(mvalue.strip()) | |
233 | |
234 def is_email_list(self, mvalue): | |
235 mvalue = self.trans_email_list(mvalue) | |
236 if mvalue != None: | |
237 for email in mvalue: | |
238 if not self.is_email(email): | |
239 gcm_fatal("Invalid e-mail address '{0}' in list {1}.".format(email, ", ".join(mvalue))) | |
240 return True | |
241 | |
1 | 242 |
243 ### | |
244 ### Main program starts | |
245 ### | |
246 gcm_msgbuf = [] | |
247 signal.signal(signal.SIGINT, gcm_signal_handler) | |
248 | |
249 | |
250 ## Settings | |
251 cfg = GCMSettings() | |
252 | |
23 | 253 cfg.mdef("debug", True, cfg.is_bool, cfg.trans_bool, False) |
1 | 254 |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
255 cfg.mdef("email_ok", False, None, None, False) |
23 | 256 cfg.mdef("email", True, cfg.is_bool, cfg.trans_bool, False) |
257 cfg.mdef("email_to", True, cfg.is_email_list, cfg.trans_email_list, None) | |
258 cfg.mdef("email_sender", True, cfg.is_email, None, None) | |
49 | 259 cfg.mdef("email_subject", True, cfg.is_string, None, u"Google Calendar MultiMerge status") |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
260 |
49 | 261 cfg.mdef("source_regex", True, cfg.is_string, None, u"^R:\s*(.*?)\s*\(\s*(.+?)\s*\)\s*$") |
23 | 262 cfg.mdef("source_regmap", False, cfg.is_list, cfg.trans_list, [1, 2]) |
1 | 263 cfg.mdef("source_regmap_len", False, None, None, len(cfg.source_regmap)) |
264 | |
23 | 265 cfg.mdef("dest_name", True, cfg.is_string, None, u"Raahen kansainvälisyystoiminta") |
266 cfg.mdef("dest_id", True, cfg.is_string, None, None) | |
1 | 267 |
268 cfg.mdef("noauth_local_webserver", False, None, None, True) | |
269 #cfg.mdef("auth_host_name", False, None, None, "localhost") | |
270 #cfg.mdef("auth_host_port", False, None, None, [8080, 8090]) | |
23 | 271 cfg.mdef("logging_level", True, cfg.is_log_level, cfg.trans_log_level, "ERROR") |
1 | 272 |
273 # No need to touch these | |
274 cfg.mdef("app_name", False, None, None, "Google Calendar MultiMerge") | |
275 cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar") | |
276 #cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar.readonly") | |
23 | 277 cfg.mdef("secret_file", True, cfg.is_filename, None, "client_secret.json") |
278 cfg.mdef("credential_file", True, cfg.is_filename, None, "client_credentials.json") | |
1 | 279 |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
280 |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
281 ## Read, parse and validate configuration file |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
282 if len(sys.argv) > 1: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
283 gcm_debug("Reading configuration from '{0}'.".format(sys.argv[1])) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
284 try: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
285 cfgparser = ConfigParser.RawConfigParser() |
45
035be8a9e982
Force reading of configuration in Unicode UTF-8.
Matti Hamalainen <ccr@tnsp.org>
parents:
44
diff
changeset
|
286 cfgparser.readfp(codecs.open(sys.argv[1], "r", "UTF-8")) |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
287 except Exception as e: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
288 gcm_fatal("Failed to read configuration file '{0}': {1}".format(sys.argv[1], str(e))) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
289 |
7
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
290 # Check that the required section exists |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
291 section = "gcm" |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
292 if not cfgparser.has_section(section): |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
293 gcm_fatal("Invalid configuration, missing '{0}' section.".format(section)) |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
294 |
10
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
295 # Debug setting is a special case, we need to get it |
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
296 # set before everything else, so do it here .. |
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
297 if cfgparser.has_option(section, "debug"): |
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
298 cfg.mset("debug", cfgparser.get(section, "debug")) |
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
299 |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
300 # Parse the settings and validate |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
301 cfg.mread(cfgparser, section) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
302 |
8 | 303 |
304 ## Validate settings | |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
305 if cfg.email: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
306 if cfg.email_subject == None or len(cfg.email_subject) == 0: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
307 gcm_fatal("E-mail enabled but email_subject not set.") |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
308 elif cfg.email_sender == None: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
309 gcm_fatal("E-mail enabled but email_sender not set.") |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
310 elif cfg.email_to == None: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
311 gcm_fatal("E-mail enabled but email_to not set.") |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
312 else: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
313 cfg.mset("email_ok", True) |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
314 |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
315 |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
316 if len(cfg.source_regmap) != cfg.source_regmap_len: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
317 gcm_fatal("Setting source_regmap list must be {0} items.".format(cfg.source_regmap_len)) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
318 else: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
319 # Force to integers |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
320 try: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
321 cfg.source_regmap = map(lambda x: int(x), cfg.source_regmap) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
322 except Exception as e: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
323 gcm_fatal("Invalid source_regmap: {0}".format(str(e))) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
324 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
325 |
8 | 326 if not cfg.dest_name and not cfg.dest_id: |
327 gcm_fatal("Target calendar ID or name required, but not set.") | |
328 | |
329 | |
330 if cfg.dest_name: | |
331 cfg.mset("dest_name", cfg.mget("dest_name").strip()) | |
332 | |
333 | |
1 | 334 ## Initialize and authorize API connection |
335 credentials = gcm_get_credentials(cfg) | |
336 http = credentials.authorize(httplib2.Http()) | |
337 service = discovery.build("calendar", "v3", http=http) | |
338 | |
339 | |
340 ## Fetch complete calendar list | |
341 gcm_debug("Fetching available calendars ..") | |
342 calendars = [] | |
19 | 343 cal_token = None |
1 | 344 while True: |
345 # We want everything except deleted and hidden calendars | |
17 | 346 result = service.calendarList().list( |
1 | 347 showHidden=False, |
348 showDeleted=False, | |
19 | 349 pageToken=cal_token |
1 | 350 ).execute() |
351 | |
17 | 352 calendars.extend(result.get("items", [])) |
19 | 353 cal_token = result.get("nextPageToken") |
354 if not cal_token: | |
1 | 355 break |
356 | |
357 if len(calendars) == 0: | |
358 gcm_fatal("No calendars found?") | |
359 | |
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
360 gcm_debug(u"{0} calendars total found.".format(len(calendars))) |
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
361 |
1 | 362 |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
363 ## Filter desired SOURCE calendars based on specified regexp |
49 | 364 src_re = re.compile(cfg.source_regex, re.UNICODE) |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
365 src_calendars = [] |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
366 for calendar in calendars: |
51
54644b29a9a3
Match also summaryOverride attribute against the source calendar regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
50
diff
changeset
|
367 if u"summary" in calendar: |
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
368 # Find destination calendar ID if not set |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
369 if not cfg.dest_id and cfg.dest_name == calendar["summary"].strip(): |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
370 cfg.mset("dest_id", calendar["id"]) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
371 |
51
54644b29a9a3
Match also summaryOverride attribute against the source calendar regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
50
diff
changeset
|
372 # If summary or summaryOverride match the regexp, add calendar |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
373 mre = src_re.match(calendar["summary"]) |
52 | 374 if not mre and u"summaryOverride" in calendar: |
375 mre = src_re.match(calendar[u"summaryOverride"]) | |
376 | |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
377 if mre: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
378 calendar["gcm_title"] = mre.group(cfg.source_regmap[0]) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
379 calendar["gcm_id"] = mre.group(cfg.source_regmap[1]) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
380 src_calendars.append(calendar) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
381 |
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
382 gcm_debug(u"{0} source calendars found.".format(len(src_calendars))) |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
383 |
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
384 |
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
385 ## Check if we have destination calendar ID |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
386 if not cfg.dest_id: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
387 gcm_fatal(u"Could not find target/destination calendar ID for '"+ cfg.dest_name +"'.") |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
388 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
389 |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
390 ## Now, we fetch and collect events |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
391 gcm_debug(u"Fetching calendar events .. ") |
19 | 392 src_events = [] |
31
6becdaa5c45f
Set event colors (just sequentially now) based on which source calendar they belong to.
Matti Hamalainen <ccr@tnsp.org>
parents:
30
diff
changeset
|
393 color_id = 0 |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
394 for calendar in src_calendars: |
31
6becdaa5c45f
Set event colors (just sequentially now) based on which source calendar they belong to.
Matti Hamalainen <ccr@tnsp.org>
parents:
30
diff
changeset
|
395 color_id = color_id + 1 |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
396 gcm_debug("- "+calendar["id"]) |
17 | 397 result = service.events().list( |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
398 timeZone="EEST", |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
399 calendarId=calendar["id"], |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
400 singleEvents=True, |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
401 showDeleted=False, |
17 | 402 # orderBy="startTime", |
403 ).execute() | |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
404 |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
405 # Add events, if any, to main list |
26
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
406 events = gcm_generate_ids(result.get("items", []), calendar["id"]) |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
407 if events: |
31
6becdaa5c45f
Set event colors (just sequentially now) based on which source calendar they belong to.
Matti Hamalainen <ccr@tnsp.org>
parents:
30
diff
changeset
|
408 for event in events: |
6becdaa5c45f
Set event colors (just sequentially now) based on which source calendar they belong to.
Matti Hamalainen <ccr@tnsp.org>
parents:
30
diff
changeset
|
409 event["colorId"] = color_id |
35
be1e798cc60a
Change event's summary to contain source calendar name-id in target calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
34
diff
changeset
|
410 event["summary"] = u"{0} [{1}]".format(event["summary"], calendar["gcm_id"]) |
21
f392d495c5b6
We need to extend a list, not append to it.
Matti Hamalainen <ccr@tnsp.org>
parents:
20
diff
changeset
|
411 src_events.extend(events) |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
412 if cfg.debug: |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
413 gcm_dump_events(events) |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
414 |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
415 |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
416 ## Get current events |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
417 gcm_debug(u"Fetching current target calendar events {0}".format(cfg.dest_id)) |
17 | 418 result = service.events().list( |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
419 calendarId=cfg.dest_id, |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
420 singleEvents=True, |
32 | 421 showDeleted=True, |
422 ).execute() | |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
423 |
33
d58a0a1f23fa
We should not doubly add destination calendar ID here ..
Matti Hamalainen <ccr@tnsp.org>
parents:
32
diff
changeset
|
424 dst_events = gcm_generate_ids(result.get("items", []), "") |
22 | 425 if dst_events: |
426 gcm_debug(u"Found {0} event(s).".format(len(dst_events))) | |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
427 else: |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
428 gcm_debug(u"No current events.") |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
429 |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
430 |
27 | 431 ## Start merging events .. |
432 gcm_debug("Re-merging events to target calendar ..") | |
34 | 433 dst_gcm_ids = frozenset(map(lambda x: x["gcm_id"], dst_events)) |
434 src_ids = frozenset(map(lambda x: x["id"], src_events)) | |
435 dst_ids = frozenset(map(lambda x: x["id"], dst_events)) | |
27 | 436 |
437 for event in src_events: | |
438 # Does the event exist already in the target? | |
439 if event["gcm_id"] in dst_gcm_ids: | |
34 | 440 ## Yes. Thus, we just update the event. |
41 | 441 #print "IS in dst_gcm_ids: "+ event["id"] +" : "+ event["gcm_id"] |
39
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
442 # Check if event NEEDS updating .. aka compare data |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
443 #d_event = gcm_get_event_by_gcm_id(dst_events, event["gcm_id"]) |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
444 #if d_event and gcm_compare_events(event, d_event): |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
445 if True: |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
446 try: |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
447 new_event = service.events().update(calendarId=cfg.dest_id, eventId=event["id"], body=event).execute() |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
448 except Exception as e: |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
449 gcm_fatal("Failed to update event:\n{0}\n\nERROR: {1}\n".format(event, str(e))) |
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
450 else: |
40 | 451 gcm_debug("No need to update event {0}.".format(event["gcm_id"])) |
27 | 452 else: |
34 | 453 ## Event does not seem to exist. Insert new event. |
454 event.pop("iCalUID", None) # Remove the iCalUID, having it conflicts with event ID | |
455 event["id"] = event["gcm_id"] # Replace Google generated ID with our own | |
456 try: | |
457 new_event = service.events().insert(calendarId=cfg.dest_id, body=event).execute() | |
458 except Exception as e: | |
459 gcm_fatal("Failed to insert new event:\n{0}\n\nERROR: {1}\n".format(event, str(e))) | |
460 | |
461 ## Remove "stale" events | |
38
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
462 for event in dst_events: |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
463 if not event["id"] in src_ids: |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
464 try: |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
465 service.events().delete(calendarId=cfg.dest_id, eventId=event["id"]).execute() |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
466 except Exception as e: |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
467 gcm_fatal("Failed to delete stale event:\n{0}\n\nERROR: {1}\n".format(event, str(e))) |
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
468 |
27 | 469 |
470 gcm_debug("Finished.") |