]>
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
=""
28 #: Fichier contenant les anniversaires
29 self
.birthdays_file
= "birthdays.txt"
30 #: Fichier contenant les évènements à venir
31 self
.timers_file
= "timers.txt"
32 #: Fichier contenant les fêtes à souhaiter
33 self
.saints_file
= "saints.txt"
34 #: Fichier contenant les ids des derniers trucs vus/lus
35 self
.last_seen_file
= "lasts"
36 #: Fichier contenant le timestamp de dernière exécution
37 self
.lasttime_file
= ".lasttime"
38 #: 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é
39 self
.something_file
= ".something"
41 if "--color" in sys
.argv
:
42 sys
.argv
.remove("--color")
44 elif sys
.stdout
.isatty():
49 config
= Config(color
=color
)
51 def print_date(timestamp
=None,color
=True):
52 """Afficher la date"""
54 timestamp
= time
.time()
55 return time
.strftime(" "*30 + config
.blue
+"%d/%m/%Y %H:%M:%S"+config
.endcolor
+ "\n",
56 time
.localtime(timestamp
))
58 def add_title(titre
, texte
):
59 """Ajoute un titre à la partie si il y a quelque chose dedans"""
61 texte
= " %s%s :%s\n" % (config
.green
, titre
, config
.endcolor
) + texte
+ "\n"
65 """Obtenir la date actuelle sous le bon format"""
66 timestamp
= time
.time()
67 now
= datetime
.datetime(*time
.localtime(timestamp
)[:7])
71 """Récupère la liste des derniers trucs vus/lus"""
72 with
open(config
.last_seen_file
) as f
:
73 return json
.loads(f
.read())
75 def update_last_seen(what
, value
):
76 """Met à jour un des derniers trucs vus/lus"""
77 lasts
= get_last_seen()
78 if not what
in lasts
.keys():
79 print """%r n'est pas un "truc vu/lu" valide""" % (what
,)
82 with
open(config
.last_seen_file
, "w") as f
:
83 f
.write(json
.dumps(lasts
))
85 def parse_datefile(namefile
):
86 """Ouvre et parse un fichier avec des lignes de la forme
89 with
open(namefile
) as f
:
90 rawdata
= [l
.strip().decode("utf8") for l
in f
.readlines()[1:] if not l
.strip().startswith("#")]
93 date
, truc
= l
.split("\t",1)
94 date
= datetime
.datetime(*time
.strptime(date
,"%d/%m/%Y")[:7])
95 data
.append([date
, truc
])
100 """Obtenir la liste des évènements à venir (J-n)"""
102 data
= parse_datefile(config
.timers_file
)
104 for [date
, event
] in data
:
105 delta
= date
- now
+ datetime
.timedelta(1)
106 if delta
> datetime
.timedelta(0):
107 timers
.append([event
,delta
.days
])
108 eventsize
= max([0]+[len(l
[0]) for l
in timers
])
109 timers
= "\n".join([("%%-%ss J-%%s" % eventsize
) % (event
, timer
) for event
,timer
in timers
])
110 timers
= add_title("Timers", timers
)
114 """Renvoie True si au moins un élément de l est vrai"""
120 def get_birthdays(*search
):
121 """Obtenir la liste des anniversaires à venir,
122 ou la liste des anniversaires correspondants à une recherche"""
124 liste
= parse_datefile(config
.birthdays_file
)
127 # Simple demande d'anniversaires courants
128 for date
, nom
in liste
:
129 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
130 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
131 if delta
> datetime
.timedelta(0):
132 age
= now
.year
- date
.year
133 if abs(delta
) < datetime
.timedelta(1):
134 birthdays
.append([config
.red
,"%s a %s ans AUJOURD'HUI !"%(nom
,age
),0,config
.endcolor
])
135 elif delta
< datetime
.timedelta(7):
136 birthdays
.append([config
.red
,"%s va avoir %s ans"%(nom
,age
),-delta
.days
,config
.endcolor
])
137 elif delta
< datetime
.timedelta(14):
138 birthdays
.append(["","%s va avoir %s ans"%(nom
,age
),-delta
.days
,""])
139 elif datetime
.timedelta(30-4) < delta
< datetime
.timedelta(30+4):
140 birthdays
.append(["","%s va avoir %s ans"%(nom
,age
),-delta
.days
,""])
142 # Recherche spécifique
143 search
= [i
.lower() for i
in search
]
144 tous
= ("--all" in search
)
145 for date
, nom
in liste
:
146 if tous
or exists([term
.lower() in nom
.lower() for term
in search
]):
147 thisyeardate
= datetime
.datetime(now
.year
, date
.month
, date
.day
)
148 delta
= thisyeardate
- now
+ datetime
.timedelta(1)
149 age
= now
.year
- date
.year
151 birthdays
.append(["", "%s a eu %s ans"%(nom
,age
), -delta
.days
, ""])
153 birthdays
.append(["", "%s va avoir %s ans"%(nom
,age
), -delta
.days
, ""])
154 birthdays
.sort(lambda x
,y
: -cmp(x
[2], y
[2]))
155 eventsize
= max([0]+[len(l
[1]) for l
in birthdays
])
156 template
= ("%%s%%-%ss J%%+d%%s" % eventsize
)
157 birthdays
= "\n".join([template
% tuple(tup
) for tup
in birthdays
])
158 birthdays
= add_title("Anniversaires", birthdays
)
161 def check_birthdays():
162 """Compte combien il y a d'anniversaires à afficher"""
163 birthdays
= get_birthdays()
165 n
= birthdays
.count("\n") - 1
170 def format_quotes(liste
):
171 """Formate les quotes de dicos à texte"""
172 t
= ("\n" + "_"*80 + "\n").join(["%(id)s (%(date)s)\n%(quote)s" % q
for q
in liste
])
176 """Récupère les quotes DTC non lues"""
178 last_dtc
= get_last_seen().get("dtc", 0)
179 cmd
= "~/bin/dtc %s + --json" % (last_dtc
+1,)
181 cmd
= "~/bin/dtc %s --json" % args
[0]
183 cmd
= "~/bin/dtc %s %s --json" % (args
[0], args
[1])
186 proc
= subprocess
.Popen(["ssh", "pimeys", cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
187 out
, err
= proc
.communicate()
189 quotes
= json
.loads(out
)
191 last_id
= max([q
["id"] for q
in quotes
])
192 update_lasts("dtc", last_id
)
193 textquotes
= format_quotes(quotes
)
196 def update_xkcd(newid
):
197 update_last_seen("xkcd", int(newid
))
201 """Vérifie si il y a des derniers trucs non lus/vus."""
202 cmd
= "/home/vincent/scripts/today/today-server.py whatsup"
203 proc
= subprocess
.Popen(["ssh", "pimeys", cmd
], stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
204 out
, err
= proc
.communicate()
205 news
= json
.loads(out
)
206 seen
= get_last_seen()
207 n_birth
= check_birthdays()
208 n_dtc
= news
["dtc"] - seen
.get("dtc", 0)
209 n_xkcd
= news
["xkcd"] - seen
.get("xkcd", 0)
210 l
= [["Anniversaires", n_birth
],
211 ["Quotes DTC", n_dtc
],
212 ["XKCD non lus", n_xkcd
]]
218 checks
.append("%s : %s" % (text
, n
))
219 checks
= "\n".join(checks
)
220 checks
= add_title("Checks", checks
)
223 def get_everything():
224 """Récupère toutes les infos"""
225 work
= [action() for action
in AUTOMATED_ACTIONS
.values()]
226 chain
= "\n\n".join([result
for result
in work
if result
])
229 def _is_there_something_in_today():
230 """Teste si il y a des choses non lue dans le today"""
231 return open(config
.something_file
, "r").read() == "True"
233 def _there_is_something_in_today(something
):
234 """Met à jour le something"""
235 f
= open(config
.something_file
, "w")
236 f
.write(str(bool(something
)))
240 """Récupère la dernière fois que today a été exécuté"""
241 if os
.path
.isfile(config
.lasttime_file
):
242 lasttime
= open(config
.lasttime_file
, "r").read()
245 lasttime
= datetime
.datetime(*time
.localtime(float(lasttime
))[:7])
248 def _update_lasttime(when
):
249 """Met à jour le timestamp de dernière exécution de today"""
250 f
= open(config
.lasttime_file
, "w")
251 f
.write(str(time
.mktime(when
.timetuple())))
255 """Dit juste si il y a quelque chose à voir.
256 La première exécution de la journée peut être lente parce qu'elle va bosser avnt de répondre."""
258 lasttime
= _get_lasttime()
259 if (lasttime
.date() < now
.date()):
260 # On a changé de jour
261 _update_lasttime(now
)
262 something
= get_everything()
263 _there_is_something_in_today(something
)
265 return "You have something in %stoday%s" % (config
.red
, config
.endcolor
)
267 return "Nothing in today"
269 if _is_there_something_in_today():
270 return "You have something in %stoday%s" % (config
.red
, config
.endcolor
)
273 """Action par défaut, affiche toutes les infos"""
275 out
+= get_everything()
276 _there_is_something_in_today(False)
280 """Crée les fichiers (vides) nécessaires au fonctionnement du script"""
281 files
= [config
.birthdays_file
, config
.timers_file
, config
.saints_file
,
282 config
.last_seen_file
]
283 contents
= ["# Anniversaires\n#jj/mm/aaaa Prénom Nom\n",
284 "# Évènements à venir\n#jj/mm/aaaa Évènement\n",
285 "# Fêtes\n#jj/mm/aaaa Saint\n",
287 for ifile
in range(len(files
)):
288 namefile
= files
[ifile
]
289 if os
.path
.isfile(namefile
):
290 print "%s exists, skipping." % (namefile
,)
292 f
= open(namefile
, "w")
293 f
.write(contents
[ifile
])
297 #: Les actions effectuées lors d'un appel sans paramètres
298 AUTOMATED_ACTIONS
= {
299 "timers" : get_timers
,
300 "birth" : get_birthdays
,
304 #: Les actions qu'on peut effectuer en rajoutant des paramètres
306 "xkcd" : update_xkcd
,
314 #: Toutes les actions
315 ACTIONS
= dict(AUTOMATED_ACTIONS
)
316 ACTIONS
.update(OTHER_ACTIONS
)
318 ACTIONS
[None] = affiche
# action par défaut
320 if __name__
== "__main__":
322 if "--no-color" in sys
.argv
:
323 config
.endcolor
, config
.red
, config
.blue
= "","",""
324 if len(sys
.argv
) == 1:
326 output
= ACTIONS
[None]()
328 commande
= sys
.argv
[1]
330 output
= ACTIONS
[commande
](*args
)
332 print output
.encode("utf-8")