Mercurial > hg > gcmultimerge
annotate multimerge.py @ 25:d32e4d4ef163
Add calendar data in events.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Mon, 04 Jul 2016 15:19:53 +0300 |
parents | 1bce0c6a673c |
children | 1267d61f6224 |
rev | line source |
---|---|
1 | 1 #!/usr/bin/python |
3 | 2 # coding=utf-8 |
2 | 3 ### |
4 ### Google Calendar MultiMerge v0.000001 | |
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 | |
14 import time | |
15 import datetime | |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
16 |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
17 import smtplib |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
18 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
|
19 |
1 | 20 import httplib2 |
21 import ConfigParser | |
20 | 22 |
1 | 23 import oauth2client |
24 from oauth2client import client | |
25 from oauth2client import tools | |
4 | 26 from oauth2client import file |
1 | 27 from googleapiclient import discovery |
28 | |
29 | |
30 ### | |
31 ### Misc. helper functions | |
32 ### | |
33 | |
34 ## Wrapper for print() that does not break when redirecting stdin/out | |
35 ## because of piped output not having a defined encoding. We default | |
36 ## to UTF-8 encoding in output here. | |
37 def gcm_print(smsg): | |
38 gcm_msgbuf.append(smsg.encode("UTF-8")) | |
39 if sys.stdout.encoding != None: | |
40 print(smsg.encode(sys.stdout.encoding)) | |
41 else: | |
42 print(smsg.encode("UTF-8")) | |
43 | |
44 | |
45 ## Fatal errors | |
46 def gcm_fatal(smsg): | |
47 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
|
48 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
|
49 ## 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
|
50 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
|
51 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
|
52 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
|
53 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
|
54 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
|
55 try: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
56 smtpH = smtplib.SMTP('localhost') |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
57 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
|
58 smtpH.quit() |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
59 except: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
60 gcm_print("FATAL: Oh crap, e-mail sending failed.") |
1 | 61 sys.exit(1) |
62 | |
63 | |
64 ## Debug messages | |
65 def gcm_debug(smsg): | |
66 if cfg.debug: | |
67 gcm_print(u"DBG: "+ smsg) | |
68 else: | |
69 gcm_msgbuf.append(u"DBG: "+ smsg.encode("UTF-8")) | |
70 | |
71 | |
24 | 72 ## Handler for SIGINT signals |
1 | 73 def gcm_signal_handler(signal, frame): |
74 gcm_print("\nQuitting due to SIGINT / Ctrl+C!") | |
75 sys.exit(0) | |
76 | |
77 | |
24 | 78 ## Function for handling Google API credentials |
1 | 79 def gcm_get_credentials(mcfg): |
80 store = oauth2client.file.Storage(mcfg.credential_file) | |
81 credentials = store.get() | |
82 if not credentials or credentials.invalid: | |
83 flow = client.flow_from_clientsecrets(mcfg.secret_file, mcfg.scope) | |
84 flow.user_agent = mcfg.app_name | |
85 credentials = tools.run_flow(flow, store, mcfg) | |
86 if not credentials or credentials.invalid: | |
87 gcm_fatal("Failed to authenticate / invalid credentials.") | |
88 return credentials | |
89 | |
90 | |
91 def gcm_dump_events(events): | |
92 for event in events: | |
93 ev_start = event["start"].get("dateTime", event["start"].get("date")) | |
94 ev_end = event["end"].get("dateTime", event["end"].get("date")) | |
95 gcm_print(u"{0:25} - {1:25} : {2}".format(ev_start, ev_end, event["summary"])) | |
96 | |
97 | |
24 | 98 ## |
99 ## Class for handling configuration / settings | |
100 ## | |
1 | 101 class GCMSettings(dict): |
102 def __init__(self): | |
103 self.m_data = {} | |
104 self.m_saveable = {} | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
105 self.m_validate = {} |
1 | 106 self.m_translate = {} |
107 | |
108 def __getattr__(self, name): | |
109 if name in self.m_data: | |
110 return self.m_data[name] | |
111 else: | |
112 gcm_fatal("GCMSettings.__getattr__(): No such attribute '"+ name +"'.") | |
113 | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
114 def mvalidate(self, name, value): |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
115 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
|
116 if not self.m_validate[name](value): |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
117 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
|
118 |
1 | 119 def mtranslate(self, name, value): |
120 if name in self.m_translate and self.m_translate[name]: | |
121 return self.m_translate[name](value) | |
122 else: | |
123 return value | |
124 | |
125 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
|
126 self.mvalidate(name, value) |
1 | 127 self.m_saveable[name] = saveable |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
128 self.m_validate[name] = validate |
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
129 self.m_translate[name] = translate |
1 | 130 self.m_data[name] = self.mtranslate(name, value) |
131 | |
132 def mset(self, name, value): | |
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
133 self.mvalidate(name, value) |
1 | 134 if name in self.m_data: |
135 self.m_data[name] = self.mtranslate(name, value) | |
136 else: | |
137 gcm_fatal("GCMSettings.mset(): No such attribute '"+ name +"'.") | |
138 | |
139 def mget(self, name): | |
140 if name in self.m_data: | |
141 return self.m_data[name] | |
142 else: | |
143 return None | |
144 | |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
145 def mread(self, cfgparser, sect): |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
146 for name in self.m_saveable: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
147 if cfgparser.has_option(sect, name): |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
148 value = cfgparser.get(sect, name) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
149 self.mset(name, value) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
150 gcm_debug("{0} -> '{1}' == {2}".format(name, value, self.mget(name))) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
151 |
23 | 152 def is_str(self, mvalue): |
153 return isinstance(mvalue, basestring) | |
154 | |
155 def is_string(self, mvalue): | |
156 return mvalue == None or self.is_str(mvalue) | |
157 | |
158 def is_log_level(self, mvalue): | |
159 if not self.is_str(mvalue): | |
160 return False | |
161 else: | |
162 return mvalue.upper() in ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"] | |
163 | |
164 def trans_log_level(self, mvalue): | |
165 return mvalue.upper() | |
166 | |
167 def is_filename(self, mvalue): | |
168 if not self.is_str(mvalue): | |
169 return False | |
170 else: | |
171 return re.match("^[a-z0-9][a-z0-9\.\_\-]+$", mvalue, flags=re.IGNORECASE) | |
172 | |
173 def trans_bool(self, mvalue): | |
174 if self.is_str(mvalue): | |
175 if re.match("^\s*(true|1|on|yes)\s*$", mvalue, re.IGNORECASE): | |
176 mvalue = True | |
177 elif re.match("^\s*(false|0|off|no)\s*$", mvalue, re.IGNORECASE): | |
178 mvalue = False | |
179 else: | |
180 return None | |
181 return mvalue | |
182 | |
183 def is_bool(self, mvalue): | |
184 mval = self.trans_bool(mvalue) | |
185 if not isinstance(mval, bool): | |
186 gcm_fatal("GCMSettings.is_bool(): Invalid boolean value '{0}', should be true|false|1|0|on|off|yes|no.".format(mvalue)) | |
187 else: | |
188 return True | |
189 | |
190 def trans_list(self, mvalue): | |
191 morig = mvalue | |
192 if self.is_str(mvalue): | |
193 mvalue = re.split("\s*,\s*", mvalue, flags=re.IGNORECASE) | |
194 if not isinstance(mvalue, list): | |
195 gcm_fatal("GCMSettings.trans_list(): Could not parse list '{0}'.".format(mvalue)) | |
196 elif not isinstance(mvalue, list): | |
197 gcm_fatal("GCMSettings.trans_list(): Invalid value '{0}'.".format(mvalue)) | |
198 return mvalue | |
199 | |
200 def is_list(self, mvalue): | |
201 return self.trans_list(mvalue) | |
202 | |
203 def is_email(self, mvalue): | |
204 if not self.is_string(mvalue): | |
205 return False | |
206 else: | |
207 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) | |
208 | |
209 def trans_email_list(self, mvalue): | |
210 if mvalue == None: | |
211 return mvalue | |
212 else: | |
213 return self.trans_list(mvalue.strip()) | |
214 | |
215 def is_email_list(self, mvalue): | |
216 mvalue = self.trans_email_list(mvalue) | |
217 if mvalue != None: | |
218 for email in mvalue: | |
219 if not self.is_email(email): | |
220 gcm_fatal("Invalid e-mail address '{0}' in list {1}.".format(email, ", ".join(mvalue))) | |
221 return True | |
222 | |
1 | 223 |
224 ### | |
225 ### Main program starts | |
226 ### | |
227 gcm_msgbuf = [] | |
228 signal.signal(signal.SIGINT, gcm_signal_handler) | |
229 | |
230 | |
231 ## Settings | |
232 cfg = GCMSettings() | |
233 | |
23 | 234 cfg.mdef("debug", True, cfg.is_bool, cfg.trans_bool, False) |
1 | 235 |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
236 cfg.mdef("email_ok", False, None, None, False) |
23 | 237 cfg.mdef("email", True, cfg.is_bool, cfg.trans_bool, False) |
238 cfg.mdef("email_to", True, cfg.is_email_list, cfg.trans_email_list, None) | |
239 cfg.mdef("email_sender", True, cfg.is_email, None, None) | |
240 cfg.mdef("email_subject", True, cfg.is_string, None, "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
|
241 |
23 | 242 cfg.mdef("source_regex", True, cfg.is_string, None, "^R:\s*(.*?)\s*\(\s*(.+?)\s*\)\s*$") |
243 cfg.mdef("source_regmap", False, cfg.is_list, cfg.trans_list, [1, 2]) | |
1 | 244 cfg.mdef("source_regmap_len", False, None, None, len(cfg.source_regmap)) |
245 | |
23 | 246 cfg.mdef("dest_name", True, cfg.is_string, None, u"Raahen kansainvälisyystoiminta") |
247 cfg.mdef("dest_id", True, cfg.is_string, None, None) | |
1 | 248 |
249 cfg.mdef("noauth_local_webserver", False, None, None, True) | |
250 #cfg.mdef("auth_host_name", False, None, None, "localhost") | |
251 #cfg.mdef("auth_host_port", False, None, None, [8080, 8090]) | |
23 | 252 cfg.mdef("logging_level", True, cfg.is_log_level, cfg.trans_log_level, "ERROR") |
1 | 253 |
254 # No need to touch these | |
255 cfg.mdef("app_name", False, None, None, "Google Calendar MultiMerge") | |
256 cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar") | |
257 #cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar.readonly") | |
23 | 258 cfg.mdef("secret_file", True, cfg.is_filename, None, "client_secret.json") |
259 cfg.mdef("credential_file", True, cfg.is_filename, None, "client_credentials.json") | |
1 | 260 |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
261 |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
262 ## Read, parse and validate configuration file |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
263 if len(sys.argv) > 1: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
264 gcm_debug("Reading configuration from '{0}'.".format(sys.argv[1])) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
265 try: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
266 cfgparser = ConfigParser.RawConfigParser() |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
267 cfgparser.read(sys.argv[1]) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
268 except Exception as e: |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
269 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
|
270 |
7
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
271 # Check that the required section exists |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
272 section = "gcm" |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
273 if not cfgparser.has_section(section): |
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
274 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
|
275 |
10
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
276 # 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
|
277 # 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
|
278 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
|
279 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
|
280 |
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
281 # Parse the settings and validate |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
282 cfg.mread(cfgparser, section) |
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
283 |
8 | 284 |
285 ## Validate settings | |
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
286 if cfg.email: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
287 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
|
288 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
|
289 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
|
290 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
|
291 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
|
292 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
|
293 else: |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
294 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
|
295 |
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
296 |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
297 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
|
298 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
|
299 else: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
300 # Force to integers |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
301 try: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
302 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
|
303 except Exception as e: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
304 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
|
305 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
306 |
8 | 307 if not cfg.dest_name and not cfg.dest_id: |
308 gcm_fatal("Target calendar ID or name required, but not set.") | |
309 | |
310 | |
311 if cfg.dest_name: | |
312 cfg.mset("dest_name", cfg.mget("dest_name").strip()) | |
313 | |
314 | |
1 | 315 ## Initialize and authorize API connection |
316 credentials = gcm_get_credentials(cfg) | |
317 http = credentials.authorize(httplib2.Http()) | |
318 service = discovery.build("calendar", "v3", http=http) | |
319 | |
320 | |
321 ## Fetch complete calendar list | |
322 gcm_debug("Fetching available calendars ..") | |
323 calendars = [] | |
19 | 324 cal_token = None |
1 | 325 while True: |
326 # We want everything except deleted and hidden calendars | |
17 | 327 result = service.calendarList().list( |
1 | 328 showHidden=False, |
329 showDeleted=False, | |
19 | 330 pageToken=cal_token |
1 | 331 ).execute() |
332 | |
17 | 333 calendars.extend(result.get("items", [])) |
19 | 334 cal_token = result.get("nextPageToken") |
335 if not cal_token: | |
1 | 336 break |
337 | |
338 if len(calendars) == 0: | |
339 gcm_fatal("No calendars found?") | |
340 | |
341 | |
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
342 ## Filter desired SOURCE calendars based on specified regexp |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
343 src_re = re.compile(cfg.source_regex) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
344 src_calendars = [] |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
345 for calendar in calendars: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
346 if "summary" in calendar: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
347 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
|
348 cfg.mset("dest_id", calendar["id"]) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
349 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
350 mre = src_re.match(calendar["summary"]) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
351 if mre: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
352 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
|
353 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
|
354 src_calendars.append(calendar) |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
355 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
356 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
357 ## Check if we have target ID |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
358 if not cfg.dest_id: |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
359 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
|
360 |
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
361 |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
362 ## Now, we fetch and collect events |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
363 gcm_debug(u"Fetching calendar events .. ") |
19 | 364 src_events = [] |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
365 for calendar in src_calendars: |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
366 gcm_debug("- "+calendar["id"]) |
17 | 367 result = service.events().list( |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
368 timeZone="EEST", |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
369 calendarId=calendar["id"], |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
370 singleEvents=True, |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
371 showDeleted=False, |
17 | 372 # orderBy="startTime", |
373 ).execute() | |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
374 |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
375 # Add events, if any, to main list |
17 | 376 events = result.get("items", []) |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
377 if events: |
25
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
378 # Add calendar data to events |
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
379 for ev in events: |
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
380 ev["cal_id"] = calendar["id"] |
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
381 ev["gcm_id"] = "{0}_{1}".format(calendar["id"], ev["id"]) |
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
382 |
21
f392d495c5b6
We need to extend a list, not append to it.
Matti Hamalainen <ccr@tnsp.org>
parents:
20
diff
changeset
|
383 src_events.extend(events) |
25
d32e4d4ef163
Add calendar data in events.
Matti Hamalainen <ccr@tnsp.org>
parents:
24
diff
changeset
|
384 |
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
385 if cfg.debug: |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
386 gcm_dump_events(events) |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
387 |
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
388 |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
389 ## Get current events |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
390 gcm_debug(u"Fetching current target calendar events {0}".format(cfg.dest_id)) |
17 | 391 result = service.events().list( |
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
392 calendarId=cfg.dest_id, |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
393 singleEvents=True, |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
394 showDeleted=True).execute() |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
395 |
22 | 396 dst_events = result.get("items", []) |
397 if dst_events: | |
398 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
|
399 else: |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
400 gcm_debug(u"No current events.") |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
401 |
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
402 |