Mercurial > hg > gcmultimerge
annotate multimerge.py @ 75:ee05430f1a4d
Update comment.
| author | Matti Hamalainen <ccr@tnsp.org> |
|---|---|
| date | Tue, 12 Jul 2016 14:30:19 +0300 |
| parents | b3f8621f1a25 |
| children | a63cc3633adb |
| 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 |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
15 import math |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
16 import time |
|
29
50711871fd1e
Comment out currently unnecessary imports.
Matti Hamalainen <ccr@tnsp.org>
parents:
28
diff
changeset
|
17 #import datetime |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
18 |
|
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
19 import smtplib |
|
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
20 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
|
21 |
| 1 | 22 import httplib2 |
| 23 import ConfigParser | |
| 20 | 24 |
| 1 | 25 import oauth2client |
| 26 from oauth2client import client | |
| 27 from oauth2client import tools | |
| 4 | 28 from oauth2client import file |
| 1 | 29 from googleapiclient import discovery |
| 30 | |
| 31 | |
| 32 ### | |
| 33 ### Misc. helper functions | |
| 34 ### | |
|
66
b08d159e35c7
Sanitize gcm_no_compare_fields[]
Matti Hamalainen <ccr@tnsp.org>
parents:
65
diff
changeset
|
35 gcm_no_compare_fields = [ |
|
b08d159e35c7
Sanitize gcm_no_compare_fields[]
Matti Hamalainen <ccr@tnsp.org>
parents:
65
diff
changeset
|
36 "id", "iCalUID", "etag", "sequence", "gcm_cal_id", |
|
72
1ab40033bb87
Add 'creator' to event equality comparision excluded fields.
Matti Hamalainen <ccr@tnsp.org>
parents:
71
diff
changeset
|
37 "created", "updated", "htmlLink", "organizer", "creator", |
|
66
b08d159e35c7
Sanitize gcm_no_compare_fields[]
Matti Hamalainen <ccr@tnsp.org>
parents:
65
diff
changeset
|
38 ] |
|
b08d159e35c7
Sanitize gcm_no_compare_fields[]
Matti Hamalainen <ccr@tnsp.org>
parents:
65
diff
changeset
|
39 |
| 1 | 40 |
| 41 ## Wrapper for print() that does not break when redirecting stdin/out | |
| 42 ## because of piped output not having a defined encoding. We default | |
| 43 ## to UTF-8 encoding in output here. | |
| 44 def gcm_print(smsg): | |
| 47 | 45 gcm_msgbuf.append(smsg) |
| 1 | 46 if sys.stdout.encoding != None: |
| 47 print(smsg.encode(sys.stdout.encoding)) | |
| 48 else: | |
| 49 print(smsg.encode("UTF-8")) | |
| 50 | |
| 51 | |
| 52 ## Fatal errors | |
| 53 def gcm_fatal(smsg): | |
| 54 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
|
55 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
|
56 ## 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
|
57 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
|
58 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
|
59 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
|
60 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
|
61 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
|
62 try: |
|
71
41881ba09ba8
Allow setting SMTP server .. needs more work tho.
Matti Hamalainen <ccr@tnsp.org>
parents:
70
diff
changeset
|
63 smtpH = smtplib.SMTP(cfg.email_server) |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
64 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
|
65 smtpH.quit() |
|
70
d5e3ca4b609d
Improve error handling when e-mail sending fails.
Matti Hamalainen <ccr@tnsp.org>
parents:
68
diff
changeset
|
66 except Exception as e: |
|
d5e3ca4b609d
Improve error handling when e-mail sending fails.
Matti Hamalainen <ccr@tnsp.org>
parents:
68
diff
changeset
|
67 gcm_print(u"FATAL: Oh crap, e-mail sending failed: {0}".format(str(e))) |
| 1 | 68 sys.exit(1) |
| 69 | |
| 70 | |
| 71 ## Debug messages | |
| 72 def gcm_debug(smsg): | |
| 73 if cfg.debug: | |
| 74 gcm_print(u"DBG: "+ smsg) | |
| 75 else: | |
| 47 | 76 gcm_msgbuf.append(u"DBG: {0}".format(smsg)) |
| 1 | 77 |
| 78 | |
| 24 | 79 ## Handler for SIGINT signals |
| 1 | 80 def gcm_signal_handler(signal, frame): |
| 54 | 81 gcm_print(u"\nQuitting due to SIGINT / Ctrl+C!") |
| 1 | 82 sys.exit(0) |
| 83 | |
| 84 | |
| 24 | 85 ## Function for handling Google API credentials |
| 1 | 86 def gcm_get_credentials(mcfg): |
| 42 | 87 try: |
| 88 store = oauth2client.file.Storage(mcfg.credential_file) | |
| 89 except Exception as e: | |
| 65 | 90 gcm_fatal(u"Failed to read credential file:\n{0}\n\nERROR: {1}\n".format(mcfg.credential_file, str(e))) |
| 42 | 91 |
| 1 | 92 credentials = store.get() |
| 93 if not credentials or credentials.invalid: | |
| 42 | 94 try: |
| 95 flow = client.flow_from_clientsecrets(mcfg.secret_file, mcfg.scope) | |
| 96 except Exception as e: | |
| 65 | 97 gcm_fatal(u"Failed to fetch client secret:\n{0}\n\nERROR: {1}\n".format(mcfg.secret_file, str(e))) |
| 42 | 98 |
| 1 | 99 flow.user_agent = mcfg.app_name |
| 100 credentials = tools.run_flow(flow, store, mcfg) | |
| 101 if not credentials or credentials.invalid: | |
| 65 | 102 gcm_fatal(u"Failed to authenticate / invalid credentials.") |
| 1 | 103 return credentials |
| 104 | |
| 105 | |
| 106 def gcm_dump_events(events): | |
| 107 for event in events: | |
| 108 ev_start = event["start"].get("dateTime", event["start"].get("date")) | |
| 109 ev_end = event["end"].get("dateTime", event["end"].get("date")) | |
| 110 gcm_print(u"{0:25} - {1:25} : {2}".format(ev_start, ev_end, event["summary"])) | |
| 111 | |
| 56 | 112 |
| 113 ## Generate gcm IDs for given list of events | |
| 63 | 114 def gcm_generate_ids(events, calendar_id, sep): |
|
26
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
115 if not events: |
|
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
116 return events |
|
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
117 |
|
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
118 for ev in events: |
| 28 | 119 ev["gcm_cal_id"] = calendar_id |
| 63 | 120 ev["gcm_id"] = calendar_id + sep + ev["iCalUID"] |
|
26
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
121 |
|
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
122 return events |
|
1267d61f6224
Add function for generating unique internal IDs.
Matti Hamalainen <ccr@tnsp.org>
parents:
25
diff
changeset
|
123 |
| 1 | 124 |
|
53
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
125 ## Find event by its gcm_id from given list or return None if not found |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
126 def gcm_get_event_by_gcm_id(list, id): |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
127 for event in list: |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
128 if event["gcm_id"] == id: |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
129 return event |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
130 return None |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
131 |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
132 |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
133 |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
134 def gcm_compare_events(ev1, ev2): |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
135 for field in ev1: |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
136 if not field in gcm_no_compare_fields and ev1[field] != ev2[field]: |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
137 return False |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
138 return True |
|
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
139 |
| 63 | 140 |
| 67 | 141 ### |
| 142 ### Class for parsing and manipulating RGB colors | |
| 143 ### | |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
144 class GCMColor(): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
145 def __init__(self, src = None): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
146 if src == None: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
147 self.r = self.g = self.b = 0 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
148 elif isinstance(src, basestring): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
149 if len(src) == 6: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
150 self.r = int(src[0:2], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
151 self.g = int(src[2:4], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
152 self.b = int(src[4:6], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
153 elif len(src) == 7 and src[0] == "#": |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
154 self.r = int(src[1:3], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
155 self.g = int(src[3:5], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
156 self.b = int(src[6:7], 16) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
157 else: |
| 65 | 158 gcm_fatal(u"Expected hex-triplet string for GCMColor() initializer: {0}".format(src)) |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
159 elif isinstance(src, GCMColor): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
160 self.r = src.r |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
161 self.g = src.g |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
162 self.b = src.b |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
163 else: |
| 65 | 164 gcm_fatal(u"Invalid initializer for GCMColor() object.") |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
165 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
166 def delta(self, other): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
167 ctmp = GCMColor() |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
168 ctmp.r = other.r - self.r |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
169 ctmp.g = other.g - self.g |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
170 ctmp.b = other.b - self.b |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
171 return ctmp |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
172 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
173 def dist(self, other): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
174 ctmp = self.delta(other) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
175 return math.sqrt(ctmp.r * ctmp.r + ctmp.g * ctmp.g + ctmp.b * ctmp.b) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
176 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
177 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
178 def gcm_find_nearest_color(colors, cfind, maxdist): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
179 c_fg = GCMColor(cfind["foreground"]) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
180 c_bg = GCMColor(cfind["background"]) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
181 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
182 bdist_fg = 99999999999 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
183 bdist_bg = 99999999999 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
184 best_fit = None |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
185 for id, col in colors.iteritems(): |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
186 dist_fg = GCMColor(col["foreground"]).dist(c_fg) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
187 dist_bg = GCMColor(col["background"]).dist(c_bg) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
188 if dist_fg <= bdist_fg and dist_bg <= bdist_bg: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
189 best_fit = id |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
190 bdist_fg = dist_fg |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
191 bdist_bg = dist_bg |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
192 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
193 if bdist_fg <= maxdist and bdist_bg <= maxdist: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
194 return best_fit |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
195 else: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
196 return None |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
197 |
|
53
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
198 |
| 24 | 199 ## |
| 200 ## Class for handling configuration / settings | |
| 201 ## | |
| 1 | 202 class GCMSettings(dict): |
| 203 def __init__(self): | |
| 204 self.m_data = {} | |
| 205 self.m_saveable = {} | |
|
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
206 self.m_validate = {} |
| 1 | 207 self.m_translate = {} |
| 208 | |
| 209 def __getattr__(self, name): | |
| 210 if name in self.m_data: | |
| 211 return self.m_data[name] | |
| 212 else: | |
| 65 | 213 gcm_fatal(u"GCMSettings.__getattr__(): No such attribute '"+ name +"'.") |
| 1 | 214 |
|
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
215 def mvalidate(self, name, value): |
|
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
216 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
|
217 if not self.m_validate[name](value): |
| 65 | 218 gcm_fatal(u"GCMSettings.mvalidate(): Invalid value for attribute '{0}': {1}".format(name, value)) |
|
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
219 |
| 1 | 220 def mtranslate(self, name, value): |
| 221 if name in self.m_translate and self.m_translate[name]: | |
| 222 return self.m_translate[name](value) | |
| 223 else: | |
| 224 return value | |
| 225 | |
| 226 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
|
227 self.mvalidate(name, value) |
| 1 | 228 self.m_saveable[name] = saveable |
|
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
229 self.m_validate[name] = validate |
|
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
230 self.m_translate[name] = translate |
| 1 | 231 self.m_data[name] = self.mtranslate(name, value) |
| 232 | |
| 233 def mset(self, name, value): | |
|
5
9d4152f32223
Add some code for settings validation.
Matti Hamalainen <ccr@tnsp.org>
parents:
4
diff
changeset
|
234 self.mvalidate(name, value) |
| 1 | 235 if name in self.m_data: |
| 236 self.m_data[name] = self.mtranslate(name, value) | |
| 237 else: | |
| 43 | 238 gcm_fatal(u"GCMSettings.mset(): No such attribute '"+ name +"'.") |
| 1 | 239 |
| 240 def mget(self, name): | |
| 241 if name in self.m_data: | |
| 242 return self.m_data[name] | |
| 243 else: | |
| 244 return None | |
| 245 | |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
246 def mread(self, cfgparser, sect): |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
247 for name in self.m_saveable: |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
248 if cfgparser.has_option(sect, name): |
|
46
51c87abe5a1e
This Unicode conversion here is probably redundant.
Matti Hamalainen <ccr@tnsp.org>
parents:
45
diff
changeset
|
249 value = cfgparser.get(sect, name) |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
250 self.mset(name, value) |
| 43 | 251 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
|
252 |
| 23 | 253 def is_str(self, mvalue): |
| 254 return isinstance(mvalue, basestring) | |
| 255 | |
| 256 def is_string(self, mvalue): | |
| 257 return mvalue == None or self.is_str(mvalue) | |
| 258 | |
| 259 def is_log_level(self, mvalue): | |
| 260 if not self.is_str(mvalue): | |
| 261 return False | |
| 262 else: | |
| 263 return mvalue.upper() in ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"] | |
| 264 | |
| 265 def trans_log_level(self, mvalue): | |
| 266 return mvalue.upper() | |
| 267 | |
| 268 def is_filename(self, mvalue): | |
| 269 if not self.is_str(mvalue): | |
| 270 return False | |
| 271 else: | |
| 272 return re.match("^[a-z0-9][a-z0-9\.\_\-]+$", mvalue, flags=re.IGNORECASE) | |
| 273 | |
| 274 def trans_bool(self, mvalue): | |
| 275 if self.is_str(mvalue): | |
| 276 if re.match("^\s*(true|1|on|yes)\s*$", mvalue, re.IGNORECASE): | |
| 277 mvalue = True | |
| 278 elif re.match("^\s*(false|0|off|no)\s*$", mvalue, re.IGNORECASE): | |
| 279 mvalue = False | |
| 280 else: | |
| 281 return None | |
| 282 return mvalue | |
| 283 | |
| 284 def is_bool(self, mvalue): | |
| 285 mval = self.trans_bool(mvalue) | |
| 286 if not isinstance(mval, bool): | |
| 65 | 287 gcm_fatal(u"GCMSettings.is_bool(): Invalid boolean value '{0}', should be true|false|1|0|on|off|yes|no.".format(mvalue)) |
| 23 | 288 else: |
| 289 return True | |
| 290 | |
| 291 def trans_list(self, mvalue): | |
| 292 morig = mvalue | |
| 293 if self.is_str(mvalue): | |
| 294 mvalue = re.split("\s*,\s*", mvalue, flags=re.IGNORECASE) | |
| 295 if not isinstance(mvalue, list): | |
| 65 | 296 gcm_fatal(u"GCMSettings.trans_list(): Could not parse list '{0}'.".format(mvalue)) |
| 23 | 297 elif not isinstance(mvalue, list): |
| 65 | 298 gcm_fatal(u"GCMSettings.trans_list(): Invalid value '{0}'.".format(mvalue)) |
| 23 | 299 return mvalue |
| 300 | |
| 301 def is_list(self, mvalue): | |
| 302 return self.trans_list(mvalue) | |
| 303 | |
| 304 def is_email(self, mvalue): | |
| 305 if not self.is_string(mvalue): | |
| 306 return False | |
| 307 else: | |
| 308 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) | |
| 309 | |
| 310 def trans_email_list(self, mvalue): | |
| 311 if mvalue == None: | |
| 312 return mvalue | |
| 313 else: | |
| 314 return self.trans_list(mvalue.strip()) | |
| 315 | |
| 316 def is_email_list(self, mvalue): | |
| 317 mvalue = self.trans_email_list(mvalue) | |
| 318 if mvalue != None: | |
| 319 for email in mvalue: | |
| 320 if not self.is_email(email): | |
| 65 | 321 gcm_fatal(u"Invalid e-mail address '{0}' in list {1}.".format(email, ", ".join(mvalue))) |
| 23 | 322 return True |
| 323 | |
| 1 | 324 |
| 325 ### | |
| 326 ### Main program starts | |
| 327 ### | |
| 328 gcm_msgbuf = [] | |
| 329 signal.signal(signal.SIGINT, gcm_signal_handler) | |
| 330 | |
| 331 | |
| 332 ## Settings | |
| 333 cfg = GCMSettings() | |
| 334 | |
| 23 | 335 cfg.mdef("debug", True, cfg.is_bool, cfg.trans_bool, False) |
| 1 | 336 |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
337 cfg.mdef("email_ok", False, None, None, False) |
| 23 | 338 cfg.mdef("email", True, cfg.is_bool, cfg.trans_bool, False) |
| 339 cfg.mdef("email_to", True, cfg.is_email_list, cfg.trans_email_list, None) | |
| 340 cfg.mdef("email_sender", True, cfg.is_email, None, None) | |
| 49 | 341 cfg.mdef("email_subject", True, cfg.is_string, None, u"Google Calendar MultiMerge status") |
|
71
41881ba09ba8
Allow setting SMTP server .. needs more work tho.
Matti Hamalainen <ccr@tnsp.org>
parents:
70
diff
changeset
|
342 cfg.mdef("email_server", True, cfg.is_string, None, None) |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
343 |
| 49 | 344 cfg.mdef("source_regex", True, cfg.is_string, None, u"^R:\s*(.*?)\s*\(\s*(.+?)\s*\)\s*$") |
| 23 | 345 cfg.mdef("source_regmap", False, cfg.is_list, cfg.trans_list, [1, 2]) |
| 1 | 346 cfg.mdef("source_regmap_len", False, None, None, len(cfg.source_regmap)) |
| 347 | |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
348 cfg.mdef("dest_name", True, cfg.is_string, None, u"Raahen kansainvälisyystoiminta [{0}]") |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
349 cfg.mdef("dest_regex", True, cfg.is_string, None, u"^Raahen kansainvälisyystoiminta") |
| 23 | 350 cfg.mdef("dest_id", True, cfg.is_string, None, None) |
| 1 | 351 |
| 352 cfg.mdef("noauth_local_webserver", False, None, None, True) | |
| 353 #cfg.mdef("auth_host_name", False, None, None, "localhost") | |
| 354 #cfg.mdef("auth_host_port", False, None, None, [8080, 8090]) | |
| 23 | 355 cfg.mdef("logging_level", True, cfg.is_log_level, cfg.trans_log_level, "ERROR") |
| 1 | 356 |
| 357 # No need to touch these | |
| 358 cfg.mdef("app_name", False, None, None, "Google Calendar MultiMerge") | |
| 359 cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar") | |
| 360 #cfg.mdef("scope", False, None, None, "https://www.googleapis.com/auth/calendar.readonly") | |
| 23 | 361 cfg.mdef("secret_file", True, cfg.is_filename, None, "client_secret.json") |
| 362 cfg.mdef("credential_file", True, cfg.is_filename, None, "client_credentials.json") | |
| 1 | 363 |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
364 |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
365 ## Read, parse and validate configuration file |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
366 if len(sys.argv) > 1: |
| 54 | 367 gcm_debug(u"Reading configuration from '{0}'.".format(sys.argv[1])) |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
368 try: |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
369 cfgparser = ConfigParser.RawConfigParser() |
|
45
035be8a9e982
Force reading of configuration in Unicode UTF-8.
Matti Hamalainen <ccr@tnsp.org>
parents:
44
diff
changeset
|
370 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
|
371 except Exception as e: |
| 65 | 372 gcm_fatal(u"Failed to read configuration file '{0}': {1}".format(sys.argv[1], str(e))) |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
373 |
|
7
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
374 # Check that the required section exists |
|
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
375 section = "gcm" |
|
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
376 if not cfgparser.has_section(section): |
| 65 | 377 gcm_fatal(u"Invalid configuration, missing '{0}' section.".format(section)) |
|
7
f2ecfb3e04ee
Check that the required section exists in configuration.
Matti Hamalainen <ccr@tnsp.org>
parents:
6
diff
changeset
|
378 |
|
10
b237b96602ad
We need to handle "debug" setting before other settings, so we need a
Matti Hamalainen <ccr@tnsp.org>
parents:
9
diff
changeset
|
379 # 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
|
380 # 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
|
381 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
|
382 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
|
383 |
|
6
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
384 # Parse the settings and validate |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
385 cfg.mread(cfgparser, section) |
|
ee6bf617f839
Implement configuration file reading.
Matti Hamalainen <ccr@tnsp.org>
parents:
5
diff
changeset
|
386 |
| 8 | 387 |
| 388 ## Validate settings | |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
389 if cfg.email: |
|
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
390 if cfg.email_subject == None or len(cfg.email_subject) == 0: |
| 65 | 391 gcm_fatal(u"E-mail enabled but email_subject not set.") |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
392 elif cfg.email_sender == None: |
| 65 | 393 gcm_fatal(u"E-mail enabled but email_sender not set.") |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
394 elif cfg.email_to == None: |
| 65 | 395 gcm_fatal(u"E-mail enabled but email_to not set.") |
|
14
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
396 else: |
|
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
397 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
|
398 |
|
8262efacf3fb
Initial implementation of sending e-mail in fatal error cases.
Matti Hamalainen <ccr@tnsp.org>
parents:
13
diff
changeset
|
399 |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
400 if len(cfg.source_regmap) != cfg.source_regmap_len: |
| 65 | 401 gcm_fatal(u"Setting source_regmap list must be {0} items.".format(cfg.source_regmap_len)) |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
402 else: |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
403 # Force to integers |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
404 try: |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
405 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
|
406 except Exception as e: |
| 65 | 407 gcm_fatal(u"Invalid source_regmap: {0}".format(str(e))) |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
408 |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
409 |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
410 if not cfg.dest_regex and not cfg.dest_id: |
| 65 | 411 gcm_fatal(u"Target calendar ID or name required, but not set.") |
| 8 | 412 |
| 413 | |
| 414 | |
| 1 | 415 ## Initialize and authorize API connection |
| 416 credentials = gcm_get_credentials(cfg) | |
| 417 http = credentials.authorize(httplib2.Http()) | |
| 418 service = discovery.build("calendar", "v3", http=http) | |
| 419 | |
| 420 | |
| 421 ## Fetch complete calendar list | |
| 54 | 422 gcm_debug(u"Fetching available calendars ..") |
| 1 | 423 calendars = [] |
| 19 | 424 cal_token = None |
| 1 | 425 while True: |
| 426 # We want everything except deleted and hidden calendars | |
| 17 | 427 result = service.calendarList().list( |
| 1 | 428 showHidden=False, |
| 429 showDeleted=False, | |
| 19 | 430 pageToken=cal_token |
| 1 | 431 ).execute() |
| 432 | |
| 17 | 433 calendars.extend(result.get("items", [])) |
| 19 | 434 cal_token = result.get("nextPageToken") |
| 435 if not cal_token: | |
| 1 | 436 break |
| 437 | |
| 438 if len(calendars) == 0: | |
| 65 | 439 gcm_fatal(u"No calendars found?") |
| 1 | 440 |
|
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
441 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
|
442 |
| 1 | 443 |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
444 ## Filter desired SOURCE calendars based on specified regexp |
| 49 | 445 src_re = re.compile(cfg.source_regex, re.UNICODE) |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
446 dst_re = re.compile(cfg.dest_regex, re.UNICODE) |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
447 src_calendars = [] |
|
60
37705c4a35e1
Show target calendar information in debug mode.
Matti Hamalainen <ccr@tnsp.org>
parents:
59
diff
changeset
|
448 dst_calendar = None |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
449 for calendar in calendars: |
|
51
54644b29a9a3
Match also summaryOverride attribute against the source calendar regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
50
diff
changeset
|
450 if u"summary" in calendar: |
|
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
451 # Find destination calendar ID if not set |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
452 if not cfg.dest_id and dst_re.match(calendar["summary"]): |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
453 cfg.mset("dest_id", calendar["id"]) |
|
60
37705c4a35e1
Show target calendar information in debug mode.
Matti Hamalainen <ccr@tnsp.org>
parents:
59
diff
changeset
|
454 dst_calendar = calendar |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
455 |
|
51
54644b29a9a3
Match also summaryOverride attribute against the source calendar regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
50
diff
changeset
|
456 # 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
|
457 mre = src_re.match(calendar["summary"]) |
| 52 | 458 if not mre and u"summaryOverride" in calendar: |
| 459 mre = src_re.match(calendar[u"summaryOverride"]) | |
| 460 | |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
461 if mre: |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
462 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
|
463 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
|
464 src_calendars.append(calendar) |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
465 |
|
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
466 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
|
467 |
|
50
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
468 |
|
f8618bae162a
Improve debug messages and comments.
Matti Hamalainen <ccr@tnsp.org>
parents:
49
diff
changeset
|
469 ## Check if we have destination calendar ID |
|
60
37705c4a35e1
Show target calendar information in debug mode.
Matti Hamalainen <ccr@tnsp.org>
parents:
59
diff
changeset
|
470 if not dst_calendar: |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
471 gcm_fatal(u"Could not find target/destination calendar ID for '"+ cfg.dest_name +"'.") |
|
60
37705c4a35e1
Show target calendar information in debug mode.
Matti Hamalainen <ccr@tnsp.org>
parents:
59
diff
changeset
|
472 else: |
|
37705c4a35e1
Show target calendar information in debug mode.
Matti Hamalainen <ccr@tnsp.org>
parents:
59
diff
changeset
|
473 gcm_debug(u"Target calendar '{0}' id {1}.".format(dst_calendar["summary"], dst_calendar["id"])) |
|
9
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
474 |
|
01c933dba120
Filter source calendars based on regexp.
Matti Hamalainen <ccr@tnsp.org>
parents:
8
diff
changeset
|
475 |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
476 ## Fetch colors |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
477 try: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
478 colors = service.colors().get().execute() |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
479 except Exception as e: |
| 65 | 480 gcm_fatal(u"Failed to fetch calendar color settings:\n\n{0}".format(str(e))) |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
481 |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
482 |
|
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
483 ## Now, we fetch and collect events |
|
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
484 gcm_debug(u"Fetching calendar events .. ") |
| 19 | 485 src_events = [] |
|
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
486 for calendar in src_calendars: |
| 54 | 487 gcm_debug(u"- "+calendar["id"]) |
| 68 | 488 try: |
| 489 result = service.events().list( | |
| 490 timeZone="EEST", | |
| 491 calendarId=calendar["id"], | |
| 492 singleEvents=True, | |
| 493 showDeleted=False, | |
| 494 # orderBy="startTime", | |
| 495 ).execute() | |
| 496 except Exception as e: | |
| 497 gcm_fatal(u"Failed to fetch calendar events for {0}:\n\n{1}\n\nERROR: {2}\n".format(calendar["id"], calendar, str(e))) | |
|
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
498 |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
499 c_found = None |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
500 if "colorId" in calendar and calendar["colorId"] in colors["calendar"]: |
| 65 | 501 gcm_debug(u"Calendar color: {0}".format(colors["calendar"][calendar["colorId"]])) |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
502 c_found = gcm_find_nearest_color(colors["event"], colors["calendar"][calendar["colorId"]], 100) |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
503 if c_found: |
| 65 | 504 gcm_debug(u"Found nearest event color ID: {0}, {1}".format(c_found, colors["event"][c_found])) |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
505 else: |
| 65 | 506 gcm_debug(u"No matching event color found!") |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
507 |
|
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
508 # Add events, if any, to main list |
| 63 | 509 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
|
510 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
|
511 for event in events: |
|
62
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
512 if c_found != None: |
|
4891ed8d77d5
Implement nearest matching color search via cubic distance, and use it for
Matti Hamalainen <ccr@tnsp.org>
parents:
61
diff
changeset
|
513 event["colorId"] = c_found |
|
59
a8941896c21c
Swap order of source name and event title in target events.
Matti Hamalainen <ccr@tnsp.org>
parents:
58
diff
changeset
|
514 event["summary"] = u"[{1}] {0}".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
|
515 src_events.extend(events) |
|
11
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
516 if cfg.debug: |
|
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
517 gcm_dump_events(events) |
|
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
518 |
|
fcdee7c04ed8
Implement fetching of source events.
Matti Hamalainen <ccr@tnsp.org>
parents:
10
diff
changeset
|
519 |
|
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
520 ## Get current events |
|
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
521 gcm_debug(u"Fetching current target calendar events {0}".format(cfg.dest_id)) |
| 17 | 522 result = service.events().list( |
|
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
523 calendarId=cfg.dest_id, |
|
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
524 singleEvents=True, |
| 32 | 525 showDeleted=True, |
| 526 ).execute() | |
|
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
527 |
| 63 | 528 dst_events = gcm_generate_ids(result.get("items", []), "", "") |
| 58 | 529 gcm_debug(u"Found {0} event(s).".format(len(dst_events))) |
| 530 | |
|
13
dd240a7ad913
Fetch current events in destination calendar.
Matti Hamalainen <ccr@tnsp.org>
parents:
11
diff
changeset
|
531 |
| 27 | 532 ## Start merging events .. |
| 54 | 533 gcm_debug(u"Re-merging events to target calendar ..") |
| 55 | 534 dst_ids = frozenset(map(lambda x: x["gcm_id"], dst_events)) |
| 535 src_ids = frozenset(map(lambda x: x["gcm_id"], src_events)) | |
| 27 | 536 |
| 537 for event in src_events: | |
| 538 # Does the event exist already in the target? | |
| 55 | 539 if event["gcm_id"] in dst_ids: |
|
39
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
540 # Check if event NEEDS updating .. aka compare data |
| 63 | 541 gcm_debug(u"Event {0} : {1} exists, checking ..".format(event["id"], event["gcm_id"])) |
|
53
ea62e0ed05ae
Initial implementation of event comparision amd generally more clever about
Matti Hamalainen <ccr@tnsp.org>
parents:
52
diff
changeset
|
542 d_event = gcm_get_event_by_gcm_id(dst_events, event["gcm_id"]) |
| 55 | 543 if not gcm_compare_events(event, d_event): |
| 57 | 544 # Seems we need to update |
| 63 | 545 gcm_debug(u"Updating event {0} : {1}..".format(event["id"], event["gcm_id"])) |
|
39
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
546 try: |
| 55 | 547 event.pop("sequence", None) |
| 63 | 548 event.pop("id", None) |
| 549 event["iCalUID"] = event["gcm_id"] | |
| 550 new_event = service.events().update(calendarId=cfg.dest_id, eventId=d_event["id"], body=event).execute() | |
|
39
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
551 except Exception as e: |
| 65 | 552 gcm_fatal(u"Failed to update event {0}:\n\n{1}\n\nERROR: {2}\n".format(event["gcm_id"], event, str(e))) |
|
39
693db3f8cbe5
Begin importing event comparision stuff.
Matti Hamalainen <ccr@tnsp.org>
parents:
38
diff
changeset
|
553 else: |
| 63 | 554 gcm_debug(u"No need to update event {0} : {1}.".format(event["id"], event["gcm_id"])) |
| 27 | 555 else: |
| 34 | 556 ## Event does not seem to exist. Insert new event. |
| 54 | 557 gcm_debug(u"Inserting new event {0}".format(event["gcm_id"])) |
| 63 | 558 event.pop("id", None) |
| 559 event["iCalUID"] = event["gcm_id"] # Replace Google generated ID with our own | |
| 34 | 560 try: |
| 561 new_event = service.events().insert(calendarId=cfg.dest_id, body=event).execute() | |
| 562 except Exception as e: | |
| 65 | 563 gcm_fatal(u"Failed to insert new event:\n\n{0}\n\nERROR: {1}\n".format(event, str(e))) |
| 63 | 564 |
| 34 | 565 |
| 566 ## Remove "stale" events | |
| 54 | 567 gcm_debug(u"Purging 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
|
568 for event in dst_events: |
| 56 | 569 gcm_debug(u"Checking event {0}".format(event["gcm_id"])) |
| 55 | 570 if not event["gcm_id"] in src_ids and event["status"] != u"cancelled": |
| 571 gcm_debug(u"Deleting event {0}".format(event["gcm_id"])) | |
| 63 | 572 try: |
| 573 service.events().delete(calendarId=cfg.dest_id, eventId=event["id"]).execute() | |
| 574 except Exception as e: | |
| 65 | 575 gcm_fatal(u"Failed to delete stale event:\n{0}\n\nERROR: {1}\n".format(event, str(e))) |
|
38
54405de302d0
Attempt to delete stale events. Still needs a check for already deleted
Matti Hamalainen <ccr@tnsp.org>
parents:
37
diff
changeset
|
576 |
| 75 | 577 ## |
| 578 ## Finally, update the calendar name with timestamp | |
| 579 ## | |
|
73
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
580 t_time = time.localtime() |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
581 t_str = time.strftime("%d.%m.%Y %H:%M", t_time) |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
582 gcm_debug(u"Updating target calendar name timestamp {0}".format(t_str)) |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
583 |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
584 try: |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
585 dst_calendar["summary"] = cfg.dest_name.format(t_str) |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
586 new_calendar = service.calendars().update(calendarId=cfg.dest_id, body=dst_calendar).execute() |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
587 except Exception as e: |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
588 gcm_fatal(u"Failed to update target calendar:\n{0}\n\nERROR: {1}\n".format(dst_calendar, str(e))) |
|
b3f8621f1a25
Change how the target calendar settings work a bit. Also implement display
Matti Hamalainen <ccr@tnsp.org>
parents:
72
diff
changeset
|
589 |
| 27 | 590 |
| 54 | 591 gcm_debug(u"Finished.") |