]>
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"""
13 os
.chdir('/home/vincent/scripts/today/')
17 """Configuration (pas de couleurs si on n'output pas dans un terminal"""
18 def __init__(self
, color
=True):
20 self
.endcolor
= u
"\e[0m"
21 self
.blue
= u
"\e[1;36m"
22 self
.red
= u
"\e[1;31m"
23 self
.green
= u
"\e[1;32m"
25 self
.endcolor
=self
.blue
=self
.red
=self
.green
=""
26 #: Serveur distant où aller récupérer les checks
27 self
.distant_server
= "pimeys"
28 #: path de today-server.py sur le serveur distant
29 self
.path_today_server
= "/home/vincent/scripts/today/today_server.py"
30 #: Fichier contenant les anniversaires
31 self
.birthdays_file
= "birthdays.txt"
32 #: Fichier contenant les évènements à venir
33 self
.timers_file
= "timers.txt"
34 #: Fichier contenant les fêtes à souhaiter
35 self
.saints_file
= "saints.txt"
36 #: Fichier contenant les ids des derniers trucs vus/lus
37 self
.last_seen_file
= "lasts"
38 #: Fichier contenant le timestamp de dernière exécution
39 self
.lasttime_file
= ".lasttime"
40 #: Fichier contenant un booléen mémorisant si il y a quelquechose dans le today du jour et qu'il n'a pas encore été regardé
41 self
.something_file
= ".something"
43 if "--color" in sys
.argv
:
44 sys
.argv
.remove("--color")
46 elif sys
.stdout
.isatty():
51 config
= Config(color
=color
)
53 def print_date(timestamp
=None,color
=True):
54 """Afficher la date"""
56 timestamp
= time
.time()
57 return time
.strftime(" "*30 + config
.blue
+"%d/%m/%Y %H:%M:%S"+config
.endcolor
+ "\n",
58 time
.localtime(timestamp
))
60 def add_title(titre
, texte
):
61 """Ajoute un titre à la partie si il y a quelque chose dedans"""
63 texte
= u
" %s%s :%s\n" % (config
.green
, titre
, config
.endcolor
) + texte
+ "\n"
67 """Obtenir la date actuelle sous le bon format"""
68 timestamp
= time
.time()
69 now
= datetime
.datetime(*time
.localtime(timestamp
)[:7])
73 """Récupère la liste des derniers trucs vus/lus"""
74 with
open(config
.last_seen_file
) as f
:
75 return json
.loads(f
.read())
77 def update_last_seen(newdict
):
78 """Met à jour un des derniers trucs vus/lus"""
79 lasts
= get_last_seen()
81 with
open(config
.last_seen_file
, "w") as f
:
82 f
.write(json
.dumps(lasts
))
84 def parse_datefile(namefile
):
85 """Ouvre et parse un fichier avec des lignes de la forme
88 with
open(namefile
) as f
:
89 rawdata
= [l
.strip().decode("utf8") for l
in f
.readlines()[1:] if not l
.strip().startswith("#") and not l
.strip() == '']
92 date
, truc
= l
.split("\t",1)
93 date
= datetime
.datetime(*time
.strptime(date
, "%d/%m/%Y")[:7])
94 data
.append([date
, truc
])
99 """Obtenir la liste des évènements à venir (J-n)"""
101 data
= parse_datefile(config
.timers_file
)
103 for [date
, event
] in data
:
104 delta
= date
- now
+ datetime
.timedelta(1)
105 if delta
> datetime
.timedelta(0):
106 timers
.append([event
,delta
.days
])
107 eventsize
= max([0]+[len(l
[0]) for l
in timers
])
108 timers
= u
"\n".join([(u
"%%-%ss J-%%s" % eventsize
) % (event
, timer
) for event
,timer
in timers
])
109 timers
= add_title(u
"Timers", timers
)
113 """Renvoie True si au moins un élément de l est vrai"""
119 def get_birthdays(*search
):
120 """Obtenir la liste des anniversaires à venir,
121 ou la liste des anniversaires correspondants à une recherche"""
123 liste
= parse_datefile(config
.birthdays_file
)
126 # Simple demande d'anniversaires courants
127 for date
, nom
in liste
:
128 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
129 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
130 if delta
> datetime
.timedelta(0):
131 age
= now
.year
- date
.year
132 if abs(delta
) < datetime
.timedelta(1):
133 birthdays
.append([config
.red
, u
"%s a %s ans AUJOURD'HUI !" % (nom
, age
), 0, config
.endcolor
])
134 elif delta
< datetime
.timedelta(7):
135 birthdays
.append([config
.red
, u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, config
.endcolor
])
136 elif delta
< datetime
.timedelta(14):
137 birthdays
.append([u
"", u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, u
""])
138 elif datetime
.timedelta(30-4) < delta
< datetime
.timedelta(30+4):
139 birthdays
.append([u
"", u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, u
""])
141 # Recherche spécifique
142 search
= [i
.lower() for i
in search
]
143 tous
= ("--all" in search
)
144 for date
, nom
in liste
:
145 if tous
or exists([term
.lower() in nom
.lower() for term
in search
]):
146 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
147 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
148 age
= now
.year
- date
.year
150 birthdays
.append([u
"", u
"%s a eu %s ans" % (nom
, age
), -delta
.days
, ""])
152 birthdays
.append([u
"", u
"%s va avoir %s ans" % (nom
, age
), -delta
.days
, ""])
153 birthdays
.sort(lambda x
,y
: -cmp(x
[2], y
[2]))
154 eventsize
= max([0]+[len(l
[1]) for l
in birthdays
])
155 template
= (u
"%%s%%-%ss J%%+d%%s" % eventsize
)
156 birthdays
= u
"\n".join([template
% tuple(tup
) for tup
in birthdays
])
157 birthdays
= add_title(u
"Anniversaires", birthdays
)
160 def check_birthdays():
161 """Compte combien il y a d'anniversaires à afficher"""
162 birthdays
= get_birthdays()
164 n
= birthdays
.count(u
"\n") - 1
169 def format_quotes(liste
):
170 """Formate les quotes de dicos à texte"""
171 t
= (u
"\n" + u
"_"*80 + u
"\n").join([u
"%(id)s (%(date)s)\n%(quote)s" % q
for q
in liste
])
175 """Récupère les quotes DTC non lues"""
177 last_dtc
= get_last_seen().get("dtc", 0)
178 cmd
= "~/bin/dtc %s + --json" % (last_dtc
+1,)
180 cmd
= "~/bin/dtc %s --json" % args
[0]
182 cmd
= "~/bin/dtc %s %s --json" % (args
[0], args
[1])
185 proc
= subprocess
.Popen(["ssh", config
.distant_server
, cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
186 out
, err
= proc
.communicate()
188 quotes
= json
.loads(out
)
190 last_id
= max([q
["id"] for q
in quotes
])
191 update_last_seen({"dtc" : last_id
})
192 textquotes
= format_quotes(quotes
)
195 def update_xkcd(newid
):
196 update_last_seen({"xkcd" : int(newid
)})
198 def update_xantah(newid
):
199 update_last_seen({"xantah" : int(newid
)})
202 "dtc" : u
"Quotes DTC",
203 "xkcd" : u
"Épisodes de XKCD",
204 "xantah" : u
"Épisodes de La Légende de Xantah",
206 "birthdays" : u
"Anniversaires à souhaiter",
210 """Vérifie si il y a des derniers trucs non lus/vus."""
211 cmd
= "%s whatsup" % (config
.path_today_server
,)
212 proc
= subprocess
.Popen(["ssh", config
.distant_server
, cmd
], stdout
=subprocess
.PIPE
)
213 out
, err
= proc
.communicate()
214 news
= json
.loads(out
)
215 seen
= get_last_seen()
216 news
["birthdays"] = check_birthdays()
218 for (thing
, comm
) in THINGS
.iteritems():
219 n
= news
[thing
] - seen
.get(thing
, 0)
223 checks
.append("%s : %s" % (comm
, n
))
224 checks
= u
"\n".join(checks
)
225 checks
= add_title(u
"Checks", checks
)
228 def get_everything():
229 """Récupère toutes les infos"""
230 work
= [action() for action
in AUTOMATED_ACTIONS
.values()]
231 chain
= u
"\n\n".join([result
for result
in work
if result
])
234 def _is_there_something_in_today():
235 """Teste si il y a des choses non lue dans le today"""
236 return open(config
.something_file
, "r").read() == "True"
238 def _there_is_something_in_today(something
):
239 """Met à jour le something"""
240 f
= open(config
.something_file
, "w")
241 f
.write(str(bool(something
)))
245 """Récupère la dernière fois que today a été exécuté"""
246 if os
.path
.isfile(config
.lasttime_file
):
247 lasttime
= open(config
.lasttime_file
, "r").read()
250 lasttime
= datetime
.datetime(*time
.localtime(float(lasttime
))[:7])
253 def _update_lasttime(when
):
254 """Met à jour le timestamp de dernière exécution de today"""
255 f
= open(config
.lasttime_file
, "w")
256 f
.write(str(time
.mktime(when
.timetuple())))
260 """Dit juste si il y a quelque chose à voir.
261 La première exécution de la journée peut être lente parce qu'elle va bosser avnt de répondre."""
263 lasttime
= _get_lasttime()
264 if (lasttime
.date() < now
.date()):
265 # On a changé de jour
266 _update_lasttime(now
)
267 something
= get_everything()
268 _there_is_something_in_today(something
)
270 return u
"You have something in %stoday%s" % (config
.red
, config
.endcolor
)
272 return u
"Nothing in today"
274 if _is_there_something_in_today():
275 return u
"You have something in %stoday%s" % (config
.red
, config
.endcolor
)
278 """Action par défaut, affiche toutes les infos"""
280 out
+= get_everything()
281 _there_is_something_in_today(False)
285 """Crée les fichiers (vides) nécessaires au fonctionnement du script"""
286 files
= [config
.birthdays_file
, config
.timers_file
, config
.saints_file
,
287 config
.last_seen_file
]
288 contents
= [u
"# Anniversaires\n#jj/mm/aaaa Prénom Nom\n",
289 u
"# Évènements à venir\n#jj/mm/aaaa Évènement\n",
290 u
"# Fêtes\n#jj/mm/aaaa Saint\n",
292 for ifile
in range(len(files
)):
293 namefile
= files
[ifile
]
294 if os
.path
.isfile(namefile
):
295 print "%s exists, skipping." % (namefile
,)
297 f
= open(namefile
, "w")
298 f
.write(contents
[ifile
].encode("utf-8"))
302 """Synchronise les last_seen avec le serveur distant qui en garde une copie,
303 le maximum de chaque truc vu est gardé des deux côtés."""
304 lasts
= get_last_seen()
305 cmd
= "%s sync" % (config
.path_today_server
,)
306 proc
= subprocess
.Popen(["ssh", config
.distant_server
, cmd
],
307 stdin
= subprocess
.PIPE
, stdout
=subprocess
.PIPE
,
309 lasts_raw
= json
.dumps(lasts
)
310 proc
.stdin
.write(lasts_raw
)
312 out
= proc
.stdout
.read()
313 newdict
= json
.loads(out
)
314 update_last_seen(newdict
)
315 print u
"Nouvel état : %r" % newdict
319 #: Les actions effectuées lors d'un appel sans paramètres
320 AUTOMATED_ACTIONS
= {
321 "timers" : get_timers
,
322 "birth" : get_birthdays
,
326 #: Les actions qu'on peut effectuer en rajoutant des paramètres
328 "xkcd" : update_xkcd
,
329 "xantah" : update_xantah
,
339 #: Toutes les actions
340 ACTIONS
= dict(AUTOMATED_ACTIONS
)
341 ACTIONS
.update(OTHER_ACTIONS
)
343 ACTIONS
[None] = affiche
# action par défaut
345 if __name__
== "__main__":
347 if "--no-color" in sys
.argv
:
348 config
.endcolor
, config
.red
, config
.blue
= u
"", u
"", u
""
349 if len(sys
.argv
) == 1:
351 output
= ACTIONS
[None]()
353 commande
= sys
.argv
[1]
355 output
= ACTIONS
[commande
](*args
)