]>
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"""
15 os
.chdir('/home/vincent/scripts/today/')
19 """Configuration (pas de couleurs si on n'output pas dans un terminal"""
20 def __init__(self
, color
=True):
22 self
.endcolor
= "\e[0m"
23 self
.blue
= "\e[1;36m"
25 self
.green
= "\e[1;32m"
27 self
.endcolor
=self
.blue
=self
.red
=self
.green
=""
29 if "--color" in sys
.argv
:
30 sys
.argv
.remove("--color")
32 elif sys
.stdout
.isatty():
37 config
= Config(color
=color
)
39 def print_date(timestamp
=None,color
=True):
40 """Afficher la date"""
42 timestamp
= time
.time()
43 return time
.strftime(" "*30 + config
.blue
+"%d/%m/%Y %H:%M:%S"+config
.endcolor
+ "\n",
44 time
.localtime(timestamp
))
46 def add_title(titre
, texte
):
47 """Ajoute un titre à la partie si il y a quelque chose dedans"""
49 texte
= " %s%s :%s\n" % (config
.green
, titre
, config
.endcolor
) + texte
+ "\n"
53 """Obtenir la date actuelle sous le bon format"""
54 timestamp
= time
.time()
55 now
= datetime
.datetime(*time
.localtime(timestamp
)[:7])
59 """Récupère la liste des derniers trucs vus/lus"""
60 with
open("lasts") as f
:
61 return json
.loads(f
.read())
63 def update_lasts(what
, value
):
64 """Met à jour un des derniers trucs vus/lus"""
66 if not what
in lasts
.keys():
67 print """%r n'est pas un "truc vu/lu" valide""" % (what
,)
70 with
open("lasts", "w") as f
:
71 f
.write(json
.dumps(lasts
))
74 """Obtenir la liste des évènements à venir (J-n)"""
76 f
=open("timers.txt", "r")
77 data
= [l
.strip("\n").decode("utf8") for l
in f
.readlines()[1:]]
81 date
, event
= l
.split("\t",1)
82 date
= datetime
.datetime(*time
.strptime(date
,"%d/%m/%Y")[:7])
83 delta
= date
- now
+ datetime
.timedelta(1)
84 if delta
> datetime
.timedelta(0):
85 timers
.append([event
,delta
.days
])
86 eventsize
= max([0]+[len(l
[0]) for l
in timers
])
87 timers
= "\n".join([("%%-%ss J-%%s" % eventsize
) % (event
, timer
) for event
,timer
in timers
])
88 timers
= add_title("Timers", timers
)
92 """Renvoie True si au moins un élément de l est vrai"""
98 def get_birthdays(*search
):
99 """Obtenir la liste des anniversaires à venir,
100 ou la liste des anniversaires correspondants à une recherche"""
102 text
= open("birthdays.txt","r").readlines()
103 liste
= [l
.decode("utf8").split() for l
in text
if not l
.startswith("#")]
104 liste
= [[l
[0], " ".join(l
[1:])] for l
in liste
if l
!=[]]
105 liste
= [ [datetime
.datetime(*time
.strptime(l
[0],"%d/%m/%Y")[:7]),
110 # Simple demande d'anniversaires courants
111 for date
, nom
in liste
:
112 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
113 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
114 if delta
> datetime
.timedelta(0):
115 age
= now
.year
- date
.year
116 if abs(delta
) < datetime
.timedelta(1):
117 birthdays
.append([config
.red
,"%s a %s ans AUJOURD'HUI !"%(nom
,age
),0,config
.endcolor
])
118 elif delta
< datetime
.timedelta(7):
119 birthdays
.append([config
.red
,"%s va avoir %s ans"%(nom
,age
),-delta
.days
,config
.endcolor
])
120 elif delta
< datetime
.timedelta(14):
121 birthdays
.append(["","%s va avoir %s ans"%(nom
,age
),-delta
.days
,""])
122 elif datetime
.timedelta(30-4) < delta
< datetime
.timedelta(30+4):
123 birthdays
.append(["","%s va avoir %s ans"%(nom
,age
),-delta
.days
,""])
125 # Recherche spécifique
126 search
= [i
.lower() for i
in search
]
127 tous
= ("--all" in search
)
128 for date
, nom
in liste
:
129 if tous
or exists([term
.lower() in nom
.lower() for term
in search
]):
130 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
131 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
132 age
= now
.year
- date
.year
134 birthdays
.append(["", "%s a eu %s ans"%(nom
,age
), -delta
.days
, ""])
136 birthdays
.append(["", "%s va avoir %s ans"%(nom
,age
), -delta
.days
, ""])
137 birthdays
.sort(lambda x
,y
: -cmp(x
[2], y
[2]))
138 eventsize
= max([0]+[len(l
[1]) for l
in birthdays
])
139 template
= ("%%s%%-%ss J%%+d%%s" % eventsize
)
140 birthdays
= "\n".join([template
% tuple(tup
) for tup
in birthdays
])
141 birthdays
= add_title("Anniversaires", birthdays
)
144 def check_birthdays():
145 """Compte combien il y a d'anniversaires à afficher"""
146 birthdays
= get_birthdays()
148 n
= birthdays
.count("\n") - 1
154 """Récupère les quotes DTC non lues"""
156 cmd
= "~/bin/dtc last"
158 cmd
= "~/bin/dtc %s" % args
[0]
160 cmd
= "~/bin/dtc %s %s" % (args
[0], args
[1])
163 proc
= subprocess
.Popen(["ssh", "pimeys", cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
164 out
, err
= proc
.communicate()
166 return out
.decode("UTF-8")
169 """Vérifie si il y a des quotes DTC non lues"""
170 cmd
= "~/bin/dtc check"
171 proc
= subprocess
.Popen(["ssh", "pimeys", cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
172 out
, err
= proc
.communicate()
180 last_read
= get_lasts()["xkcd"]
182 p
= urllib
.urlopen("http://xkcd.com")
184 return "Impossible de se connecter à xkcd"
186 current_id
= int(re
.findall("Permanent link to this comic: http://xkcd.com/(.*?)/", t
)[0])
187 return current_id
- last_read
189 def update_xkcd(newid
):
190 update_lasts("xkcd", int(newid
))
194 n_birth
= check_birthdays()
196 n_xkcd
= check_xkcd()
197 l
= [["Anniversaires", n_birth
],
198 ["Quotes DTC", n_dtc
],
199 ["XKCD non lus", n_xkcd
]]
205 checks
.append("%s : %s" % (text
, n
))
206 checks
= "\n".join(checks
)
207 checks
= add_title("Checks", checks
)
210 def get_everything():
211 """Récupère toutes les infos"""
212 work
= [action() for action
in AUTOMATED_ACTIONS
.values()]
213 chain
= "\n\n".join([result
for result
in work
if result
])
216 def _is_there_something_in_today():
217 """Teste si il y a des choses non lue dans le today"""
218 return open(".something", "r").read() == "True"
220 def _there_is_something_in_today(something
):
221 """Met à jour le something"""
222 f
= open(".something", "w")
223 f
.write(str(bool(something
)))
227 """Récupère la dernière fois que today a été exécuté"""
228 lasttime
= open(".lasttime", "r").read()
229 lasttime
= datetime
.datetime(*time
.localtime(float(lasttime
))[:7])
232 def _update_lasttime(when
):
233 """Met à jour le timestamp de dernière exécution de today"""
234 f
= open(".lasttime", "w")
235 f
.write(str(time
.mktime(when
.timetuple())))
239 """Dit juste si il y a quelque chose à voir.
240 La première exécution de la journée peut être lente parce qu'elle va bosser avnt de répondre."""
242 lasttime
= _get_lasttime()
243 if (lasttime
.date() < now
.date()):
244 # On a changé de jour
245 _update_lasttime(now
)
246 something
= get_everything()
247 _there_is_something_in_today(something
)
249 return "You have something in %stoday%s" % (config
.red
, config
.endcolor
)
251 return "Nothing in today"
253 if _is_there_something_in_today():
254 return "You have something in %stoday%s" % (config
.red
, config
.endcolor
)
257 """Action par défaut, affiche toutes les infos"""
259 out
+= get_everything()
260 _there_is_something_in_today(False)
263 #: Les actions effectuées lors d'un appel sans paramètres
264 AUTOMATED_ACTIONS
= {
265 "timers" : get_timers
,
266 "birth" : get_birthdays
,
270 #: Les actions qu'on peut effectuer en rajoutant des paramètres
272 "xkcd" : update_xkcd
,
278 #: Toutes les actions
279 ACTIONS
= dict(AUTOMATED_ACTIONS
)
280 ACTIONS
.update(OTHER_ACTIONS
)
282 ACTIONS
[None] = affiche
# action par défaut
284 if __name__
== "__main__":
286 if "--no-color" in sys
.argv
:
287 config
.endcolor
, config
.red
, config
.blue
= "","",""
288 if len(sys
.argv
) == 1:
290 output
= ACTIONS
[None]()
292 commande
= sys
.argv
[1]
294 output
= ACTIONS
[commande
](*args
)
296 print output
.encode("utf-8")