]>
gitweb.pimeys.fr Git - today.git/blob - today.py
2 # -*- encoding: utf-8 -*-
5 script qui affiche des trucs à penser, des J-n des conneries
6 or that kind of stuff"""
14 os
.chdir('/home/vincent/scripts/today/')
17 # Import de la configuration
20 if "--color" in sys
.argv
:
21 sys
.argv
.remove("--color")
23 elif sys
.stdout
.isatty():
28 config
= config
.Config(color
=color
)
30 def print_date(timestamp
=None,color
=True):
31 """Afficher la date"""
33 timestamp
= time
.time()
34 return time
.strftime(" "*30 + config
.blue
+"%d/%m/%Y %H:%M:%S"+config
.endcolor
+ "\n",
35 time
.localtime(timestamp
))
37 def add_title(titre
, texte
):
38 """Ajoute un titre à la partie si il y a quelque chose dedans"""
40 texte
= u
" %s%s :%s\n" % (config
.green
, titre
, config
.endcolor
) + texte
+ "\n"
44 """Obtenir la date actuelle sous le bon format"""
45 timestamp
= time
.time()
46 now
= datetime
.datetime(*time
.localtime(timestamp
)[:7])
50 """Récupère la liste des derniers trucs vus/lus"""
51 with
open(config
.last_seen_file
) as f
:
52 return json
.loads(f
.read())
61 def update_last_seen(newdict
, lasts_seen
=None):
62 """Met à jour un des derniers trucs vus/lus"""
63 if not all([is_int(v
) or (v
== "last") for v
in newdict
.values()]):
64 raise ValueError("Donner un entier ou last")
65 # Il est possible qu'on ait mis "last" quelque part
66 if "last" in newdict
.values():
67 # il faut alors récupérer les derniers pour convertir "last" en <last_id>
68 lasts_published
= get_whatsup()
69 for key
in [k
for (k
, v
) in newdict
.items() if v
== "last"]:
70 newdict
[key
] = lasts_published
[key
]
71 # On convertit les chaînes en entier (pour les <last_id> il leur arrive rien)
72 newdict
= {k
: int(v
) for (k
,v
) in newdict
.items()}
73 if lasts_seen
is None:
74 lasts_seen
= get_last_seen()
75 lasts_seen
.update(newdict
)
76 with
open(config
.last_seen_file
, "w") as f
:
77 f
.write(json
.dumps(lasts_seen
))
78 return ", ".join(["last %s to %s" % (id, n
) for (id, n
) in newdict
.items()])
80 def parse_datefile(namefile
):
81 """Ouvre et parse un fichier avec des lignes de la forme
84 with
open(namefile
) as f
:
85 rawdata
= [l
.strip().decode("utf8") for l
in f
.readlines()[1:] if not l
.strip().startswith("#") and not l
.strip() == '']
88 date
, truc
= l
.split("\t",1)
89 date
= datetime
.datetime(*time
.strptime(date
, "%d/%m/%Y")[:7])
90 data
.append([date
, truc
])
95 """Obtenir la liste des évènements à venir (J-n)"""
97 data
= parse_datefile(config
.timers_file
)
99 for [date
, event
] in data
:
100 delta
= date
- now
+ datetime
.timedelta(1)
101 if delta
> datetime
.timedelta(0):
102 timers
.append([event
,delta
.days
])
103 eventsize
= max([0]+[len(l
[0]) for l
in timers
])
104 timers
= u
"\n".join([(u
"%%-%ss J-%%s" % eventsize
) % (event
, timer
) for event
,timer
in timers
])
105 timers
= add_title(u
"Timers", timers
)
108 def get_birthdays(*search
):
109 """Obtenir la liste des anniversaires à venir,
110 ou la liste des anniversaires correspondants à une recherche"""
112 liste
= parse_datefile(config
.birthdays_file
)
115 # Simple demande d'anniversaires courants
116 for date
, nom
in liste
:
117 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
118 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
119 if delta
> datetime
.timedelta(0):
120 age
= now
.year
- date
.year
121 if abs(delta
) < datetime
.timedelta(1):
122 birthdays
.append([config
.red
, u
"%s a %s ans AUJOURD'HUI !" % (nom
, age
), 0, config
.endcolor
])
123 elif delta
< datetime
.timedelta(7):
124 birthdays
.append([config
.red
, u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, config
.endcolor
])
125 elif delta
< datetime
.timedelta(30):
126 birthdays
.append([u
"", u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, u
""])
128 # Recherche spécifique
129 search
= [i
.lower() for i
in search
]
130 tous
= ("--all" in search
)
131 for date
, nom
in liste
:
132 if tous
or any([term
.lower() in nom
.lower() for term
in search
]):
133 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
134 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
135 age
= now
.year
- date
.year
137 birthdays
.append([u
"", u
"%s a eu %s ans" % (nom
, age
), -delta
.days
, ""])
139 birthdays
.append([u
"", u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, ""])
140 birthdays
.sort(lambda x
,y
: -cmp(x
[2], y
[2]))
141 eventsize
= max([0]+[len(l
[1]) for l
in birthdays
])
142 template
= (u
"%%s%%-%ss J%%+d%%s" % eventsize
)
143 birthdays
= u
"\n".join([template
% tuple(tup
) for tup
in birthdays
])
144 birthdays
= add_title(u
"Anniversaires", birthdays
)
148 """Renvoie la liste des fêtes contenue dans un jour"""
155 ll
.append(["St ", st
[1:]])
157 ll
.append(["Ste ", st
[1:]])
160 def _get_firstnames():
161 """Récupère la liste des noms des gens connus"""
162 birthdays
= parse_datefile(config
.birthdays_file
)
163 firstnames
= [b
[1].split()[0] for b
in birthdays
]
167 """Renvoie la liste des fêtes à souhaiter aujourd'hui et demain"""
168 sourcesaints
= json
.load(open(config
.saints_file
))
171 saints
["today"] = _parse_saint(sourcesaints
[now
.month
- 1][now
.day
- 1])
172 nbdays
= len(sourcesaints
[now
.month
- 1])
173 if now
.day
== nbdays
: # il faut regarder le mois suivant
174 if now
.month
== 12: # il faut regarder l'année suivante
175 saints
["tomorrow"] = _parse_saint(sourcesaints
[0][0])
177 saints
["tomorrow"] = _parse_saint(sourcesaints
[now
.month
][0])
179 saints
["tomorrow"] = _parse_saint(sourcesaints
[now
.month
- 1][now
.day
])
180 firstnames
= _get_firstnames()
181 towish
= {"today" : [], "tomorrow" : []}
182 for day
in ["today", "tomorrow"]:
183 ssaints
= saints
[day
]
184 for (sexe
, saint
) in ssaints
:
185 if any([firstname
.lower() in saint
.lower().split() for firstname
in firstnames
]):
186 towish
[day
].append(sexe
+ saint
)
189 ttowish
.append(u
"Aujourd'hui :\n" + "\n".join(towish
["today"]))
190 if towish
["tomorrow"]:
191 ttowish
.append(u
"Demain :\n" + "\n".join(towish
["tomorrow"]))
192 saints
= "\n".join(ttowish
)
193 saints
= add_title(u
"Fêtes à souhaiter", saints
)
196 def check_birthdays():
197 """Compte combien il y a d'anniversaires à afficher"""
198 birthdays
= get_birthdays()
200 n
= birthdays
.count(u
"\n") - 1
206 """Compte combien il y a de fêtes à afficher"""
207 saints
= get_saints()
208 return len(re
.findall("\nSt", saints
))
210 def format_quotes(liste
):
211 """Formate les quotes de dicos à texte"""
212 t
= (u
"\n" + u
"_"*80 + u
"\n").join([u
"%(id)s (%(date)s)\n%(quote)s" % q
for q
in liste
])
216 """Récupère les quotes DTC non lues"""
218 last_dtc
= get_last_seen().get("dtc", 0)
219 cmd
= "~/bin/dtc %s + --json" % (last_dtc
+1,)
221 cmd
= "~/bin/dtc %s --json" % args
[0]
223 cmd
= "~/bin/dtc %s %s --json" % (args
[0], args
[1])
226 proc
= subprocess
.Popen(["ssh", "-4", config
.distant_server
, cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
227 out
, err
= proc
.communicate()
229 quotes
= json
.loads(out
)
231 last_id
= max([q
["id"] for q
in quotes
])
232 update_last_seen({"dtc" : last_id
})
233 textquotes
= format_quotes(quotes
)
238 """Récupère sur le serveur les derniers trucs."""
239 cmd
= "%s whatsup" % (config
.path_today_server
,)
240 proc
= subprocess
.Popen(["ssh", "-4", config
.distant_server
, cmd
], stdout
=subprocess
.PIPE
)
241 out
, err
= proc
.communicate()
242 news
= json
.loads(out
)
246 """Vérifie si il y a des derniers trucs non lus/vus."""
248 seen
= get_last_seen()
250 length
= max([len(thing
) for thing
in news
.keys()])
251 for (thing
, lastid
) in news
.items():
252 n
= lastid
- seen
.get(thing
, 0)
256 checks
.append("%s :%s%s (last : %s)" % (thing
, " " *(length
+ 1 - len(thing
)), n
, news
[thing
]))
257 checks
= u
"\n".join(checks
)
258 checks
= add_title(u
"Checks", checks
)
261 def get_everything():
262 """Récupère toutes les infos."""
263 work
= [action() for (keyword
, action
) in AUTOMATED_ACTIONS
.iteritems()
264 if (config
.do_remote
or not keyword
in REMOTE_ACTIONS
)]
265 chain
= u
"\n\n".join([result
for result
in work
if result
])
268 def _is_there_something_in_today():
269 """Teste si il y a des choses non lue dans le today"""
270 return open(config
.something_file
, "r").read() == "True"
272 def _there_is_something_in_today(something
):
273 """Met à jour le something"""
274 f
= open(config
.something_file
, "w")
275 f
.write(str(bool(something
)))
279 """Récupère la dernière fois que today a été exécuté"""
280 if os
.path
.isfile(config
.lasttime_file
):
281 lasttime
= open(config
.lasttime_file
, "r").read()
284 lasttime
= datetime
.datetime(*time
.localtime(float(lasttime
))[:7])
287 def _update_lasttime(when
):
288 """Met à jour le timestamp de dernière exécution de today"""
289 f
= open(config
.lasttime_file
, "w")
290 f
.write(str(time
.mktime(when
.timetuple())))
294 """Dit juste si il y a quelque chose à voir.
295 La première exécution de la journée peut être lente parce qu'elle va bosser avant de répondre.
298 lasttime
= _get_lasttime()
299 if (lasttime
.date() < now
.date()):
300 # On a changé de jour
301 _update_lasttime(now
)
302 something
= get_everything()
303 _there_is_something_in_today(something
)
305 return u
"You have something in %stoday%s" % (config
.red
, config
.endcolor
)
307 return u
"Nothing in today"
309 if _is_there_something_in_today():
310 return u
"You have something in %stoday%s" % (config
.red
, config
.endcolor
)
313 """Action par défaut, affiche toutes les infos."""
315 out
+= get_everything()
316 _there_is_something_in_today(False)
320 """Crée les fichiers (vides) nécessaires au fonctionnement du script"""
321 files
= [config
.birthdays_file
, config
.timers_file
, config
.saints_file
,
322 config
.last_seen_file
]
323 contents
= [u
"# Anniversaires\n#jj/mm/aaaa Prénom Nom\n",
324 u
"# Évènements à venir\n#jj/mm/aaaa Évènement\n",
325 json
.dumps([[""]*31]*12).decode(),
327 for ifile
in range(len(files
)):
328 namefile
= files
[ifile
]
329 if os
.path
.isfile(namefile
):
330 print (u
"%s exists, skipping." % (namefile
,)).encode("utf-8")
332 f
= open(namefile
, "w")
333 f
.write(contents
[ifile
].encode("utf-8"))
337 """Synchronise les last_seen avec le serveur distant qui en garde une copie,
338 le maximum de chaque truc vu est gardé des deux côtés."""
339 lasts
= get_last_seen()
340 cmd
= "%s sync" % (config
.path_today_server
,)
341 proc
= subprocess
.Popen(["ssh", "-4", config
.distant_server
, cmd
],
342 stdin
= subprocess
.PIPE
, stdout
=subprocess
.PIPE
,
344 lasts_raw
= json
.dumps(lasts
)
345 proc
.stdin
.write(lasts_raw
)
347 out
= proc
.stdout
.read()
348 newdict
= json
.loads(out
)
349 update_last_seen(newdict
)
350 print (u
"Nouvel état : %r" % newdict
).encode("utf-8")
354 #: Les actions effectuées lors d'un appel sans paramètres
355 AUTOMATED_ACTIONS
= {
356 "timers" : get_timers
,
357 "birth" : get_birthdays
,
358 "saints" : get_saints
,
362 #: Les actions qu'on peut effectuer en rajoutant des paramètres
372 #: Les actions qui nécessitent un accès au serveur distant
373 REMOTE_ACTIONS
= ["check", "sync"]
375 #: Toutes les actions
376 ACTIONS
= dict(AUTOMATED_ACTIONS
)
377 ACTIONS
.update(OTHER_ACTIONS
)
379 ACTIONS
[None] = affiche
# action par défaut
381 if __name__
== "__main__":
383 if "--no-color" in sys
.argv
:
385 sys
.argv
.remove("--no-color")
386 if "--no-remote" in sys
.argv
:
387 config
.do_remote
= False
388 sys
.argv
.remove("--no-remote")
390 if len(sys
.argv
) == 1:
392 output
= ACTIONS
[None]()
394 commande
= sys
.argv
[1]
395 args
= [s
.decode("utf-8") for s
in sys
.argv
[2:]]
396 if commande
in ACTIONS
:
397 output
= ACTIONS
[commande
](*args
)
400 lasts_seen
= get_last_seen()
401 if not commande
in lasts_seen
:
402 print "Attention : n'existait pas avant."
403 output
= update_last_seen({commande
: args
[0]}, lasts_seen
)
405 print output
.encode("utf-8")