]> git.rkrishnan.org Git - tahoe-lafs/tahoe-lafs.git/blob - src/allmydata/web/status.py
mutable WIP: re-enable publish/retrieve status
[tahoe-lafs/tahoe-lafs.git] / src / allmydata / web / status.py
1
2 import time, pprint
3 import simplejson
4 from twisted.internet import defer
5 from nevow import rend, inevow, tags as T
6 from allmydata.util import base32, idlib
7 from allmydata.web.common import IClient, getxmlfile, abbreviate_time, \
8      abbreviate_rate, abbreviate_size, get_arg
9 from allmydata.interfaces import IUploadStatus, IDownloadStatus, \
10      IPublishStatus, IRetrieveStatus
11
12 def plural(sequence_or_length):
13     if isinstance(sequence_or_length, int):
14         length = sequence_or_length
15     else:
16         length = len(sequence_or_length)
17     if length == 1:
18         return ""
19     return "s"
20
21 class RateAndTimeMixin:
22
23     def render_time(self, ctx, data):
24         return abbreviate_time(data)
25
26     def render_rate(self, ctx, data):
27         return abbreviate_rate(data)
28
29 class UploadResultsRendererMixin(RateAndTimeMixin):
30     # this requires a method named 'upload_results'
31
32     def render_pushed_shares(self, ctx, data):
33         d = self.upload_results()
34         d.addCallback(lambda res: res.pushed_shares)
35         return d
36
37     def render_preexisting_shares(self, ctx, data):
38         d = self.upload_results()
39         d.addCallback(lambda res: res.preexisting_shares)
40         return d
41
42     def render_sharemap(self, ctx, data):
43         d = self.upload_results()
44         d.addCallback(lambda res: res.sharemap)
45         def _render(sharemap):
46             if sharemap is None:
47                 return "None"
48             l = T.ul()
49             for shnum in sorted(sharemap.keys()):
50                 l[T.li["%d -> %s" % (shnum, sharemap[shnum])]]
51             return l
52         d.addCallback(_render)
53         return d
54
55     def render_servermap(self, ctx, data):
56         d = self.upload_results()
57         d.addCallback(lambda res: res.servermap)
58         def _render(servermap):
59             if servermap is None:
60                 return "None"
61             l = T.ul()
62             for peerid in sorted(servermap.keys()):
63                 peerid_s = idlib.shortnodeid_b2a(peerid)
64                 shares_s = ",".join(["#%d" % shnum
65                                      for shnum in servermap[peerid]])
66                 l[T.li["[%s] got share%s: %s" % (peerid_s,
67                                                  plural(servermap[peerid]),
68                                                  shares_s)]]
69             return l
70         d.addCallback(_render)
71         return d
72
73     def data_file_size(self, ctx, data):
74         d = self.upload_results()
75         d.addCallback(lambda res: res.file_size)
76         return d
77
78     def _get_time(self, name):
79         d = self.upload_results()
80         d.addCallback(lambda res: res.timings.get(name))
81         return d
82
83     def data_time_total(self, ctx, data):
84         return self._get_time("total")
85
86     def data_time_storage_index(self, ctx, data):
87         return self._get_time("storage_index")
88
89     def data_time_contacting_helper(self, ctx, data):
90         return self._get_time("contacting_helper")
91
92     def data_time_existence_check(self, ctx, data):
93         return self._get_time("existence_check")
94
95     def data_time_cumulative_fetch(self, ctx, data):
96         return self._get_time("cumulative_fetch")
97
98     def data_time_helper_total(self, ctx, data):
99         return self._get_time("helper_total")
100
101     def data_time_peer_selection(self, ctx, data):
102         return self._get_time("peer_selection")
103
104     def data_time_total_encode_and_push(self, ctx, data):
105         return self._get_time("total_encode_and_push")
106
107     def data_time_cumulative_encoding(self, ctx, data):
108         return self._get_time("cumulative_encoding")
109
110     def data_time_cumulative_sending(self, ctx, data):
111         return self._get_time("cumulative_sending")
112
113     def data_time_hashes_and_close(self, ctx, data):
114         return self._get_time("hashes_and_close")
115
116     def _get_rate(self, name):
117         d = self.upload_results()
118         def _convert(r):
119             file_size = r.file_size
120             time = r.timings.get(name)
121             if time is None:
122                 return None
123             try:
124                 return 1.0 * file_size / time
125             except ZeroDivisionError:
126                 return None
127         d.addCallback(_convert)
128         return d
129
130     def data_rate_total(self, ctx, data):
131         return self._get_rate("total")
132
133     def data_rate_storage_index(self, ctx, data):
134         return self._get_rate("storage_index")
135
136     def data_rate_encode(self, ctx, data):
137         return self._get_rate("cumulative_encoding")
138
139     def data_rate_push(self, ctx, data):
140         return self._get_rate("cumulative_sending")
141
142     def data_rate_encode_and_push(self, ctx, data):
143         d = self.upload_results()
144         def _convert(r):
145             file_size = r.file_size
146             if file_size is None:
147                 return None
148             time1 = r.timings.get("cumulative_encoding")
149             if time1 is None:
150                 return None
151             time2 = r.timings.get("cumulative_sending")
152             if time2 is None:
153                 return None
154             try:
155                 return 1.0 * file_size / (time1+time2)
156             except ZeroDivisionError:
157                 return None
158         d.addCallback(_convert)
159         return d
160
161     def data_rate_ciphertext_fetch(self, ctx, data):
162         d = self.upload_results()
163         def _convert(r):
164             fetch_size = r.ciphertext_fetched
165             if fetch_size is None:
166                 return None
167             time = r.timings.get("cumulative_fetch")
168             if time is None:
169                 return None
170             try:
171                 return 1.0 * fetch_size / time
172             except ZeroDivisionError:
173                 return None
174         d.addCallback(_convert)
175         return d
176
177 class UploadStatusPage(UploadResultsRendererMixin, rend.Page):
178     docFactory = getxmlfile("upload-status.xhtml")
179
180     def __init__(self, data):
181         rend.Page.__init__(self, data)
182         self.upload_status = data
183
184     def upload_results(self):
185         return defer.maybeDeferred(self.upload_status.get_results)
186
187     def render_results(self, ctx, data):
188         d = self.upload_results()
189         def _got_results(results):
190             if results:
191                 return ctx.tag
192             return ""
193         d.addCallback(_got_results)
194         return d
195
196     def render_started(self, ctx, data):
197         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
198         started_s = time.strftime(TIME_FORMAT,
199                                   time.localtime(data.get_started()))
200         return started_s
201
202     def render_si(self, ctx, data):
203         si_s = base32.b2a_or_none(data.get_storage_index())
204         if si_s is None:
205             si_s = "(None)"
206         return si_s
207
208     def render_helper(self, ctx, data):
209         return {True: "Yes",
210                 False: "No"}[data.using_helper()]
211
212     def render_total_size(self, ctx, data):
213         size = data.get_size()
214         if size is None:
215             size = "(unknown)"
216         return size
217
218     def render_progress_hash(self, ctx, data):
219         progress = data.get_progress()[0]
220         # TODO: make an ascii-art bar
221         return "%.1f%%" % (100.0 * progress)
222
223     def render_progress_ciphertext(self, ctx, data):
224         progress = data.get_progress()[1]
225         # TODO: make an ascii-art bar
226         return "%.1f%%" % (100.0 * progress)
227
228     def render_progress_encode_push(self, ctx, data):
229         progress = data.get_progress()[2]
230         # TODO: make an ascii-art bar
231         return "%.1f%%" % (100.0 * progress)
232
233     def render_status(self, ctx, data):
234         return data.get_status()
235
236 class DownloadResultsRendererMixin(RateAndTimeMixin):
237     # this requires a method named 'download_results'
238
239     def render_servermap(self, ctx, data):
240         d = self.download_results()
241         d.addCallback(lambda res: res.servermap)
242         def _render(servermap):
243             if servermap is None:
244                 return "None"
245             l = T.ul()
246             for peerid in sorted(servermap.keys()):
247                 peerid_s = idlib.shortnodeid_b2a(peerid)
248                 shares_s = ",".join(["#%d" % shnum
249                                      for shnum in servermap[peerid]])
250                 l[T.li["[%s] has share%s: %s" % (peerid_s,
251                                                  plural(servermap[peerid]),
252                                                  shares_s)]]
253             return l
254         d.addCallback(_render)
255         return d
256
257     def render_servers_used(self, ctx, data):
258         d = self.download_results()
259         d.addCallback(lambda res: res.servers_used)
260         def _got(servers_used):
261             if not servers_used:
262                 return ""
263             peerids_s = ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
264                                    for peerid in servers_used])
265             return T.li["Servers Used: ", peerids_s]
266         d.addCallback(_got)
267         return d
268
269     def render_problems(self, ctx, data):
270         d = self.download_results()
271         d.addCallback(lambda res: res.server_problems)
272         def _got(server_problems):
273             if not server_problems:
274                 return ""
275             l = T.ul()
276             for peerid in sorted(server_problems.keys()):
277                 peerid_s = idlib.shortnodeid_b2a(peerid)
278                 l[T.li["[%s]: %s" % (peerid_s, server_problems[peerid])]]
279             return T.li["Server Problems:", l]
280         d.addCallback(_got)
281         return d
282
283     def data_file_size(self, ctx, data):
284         d = self.download_results()
285         d.addCallback(lambda res: res.file_size)
286         return d
287
288     def _get_time(self, name):
289         d = self.download_results()
290         d.addCallback(lambda res: res.timings.get(name))
291         return d
292
293     def data_time_total(self, ctx, data):
294         return self._get_time("total")
295
296     def data_time_peer_selection(self, ctx, data):
297         return self._get_time("peer_selection")
298
299     def data_time_uri_extension(self, ctx, data):
300         return self._get_time("uri_extension")
301
302     def data_time_hashtrees(self, ctx, data):
303         return self._get_time("hashtrees")
304
305     def data_time_segments(self, ctx, data):
306         return self._get_time("segments")
307
308     def data_time_cumulative_fetch(self, ctx, data):
309         return self._get_time("cumulative_fetch")
310
311     def data_time_cumulative_decode(self, ctx, data):
312         return self._get_time("cumulative_decode")
313
314     def data_time_cumulative_decrypt(self, ctx, data):
315         return self._get_time("cumulative_decrypt")
316
317     def _get_rate(self, name):
318         d = self.download_results()
319         def _convert(r):
320             file_size = r.file_size
321             time = r.timings.get(name)
322             if time is None:
323                 return None
324             try:
325                 return 1.0 * file_size / time
326             except ZeroDivisionError:
327                 return None
328         d.addCallback(_convert)
329         return d
330
331     def data_rate_total(self, ctx, data):
332         return self._get_rate("total")
333
334     def data_rate_segments(self, ctx, data):
335         return self._get_rate("segments")
336
337     def data_rate_fetch(self, ctx, data):
338         return self._get_rate("cumulative_fetch")
339
340     def data_rate_decode(self, ctx, data):
341         return self._get_rate("cumulative_decode")
342
343     def data_rate_decrypt(self, ctx, data):
344         return self._get_rate("cumulative_decrypt")
345
346     def render_server_timings(self, ctx, data):
347         d = self.download_results()
348         d.addCallback(lambda res: res.timings.get("fetch_per_server"))
349         def _render(per_server):
350             if per_server is None:
351                 return ""
352             l = T.ul()
353             for peerid in sorted(per_server.keys()):
354                 peerid_s = idlib.shortnodeid_b2a(peerid)
355                 times_s = ", ".join([self.render_time(None, t)
356                                      for t in per_server[peerid]])
357                 l[T.li["[%s]: %s" % (peerid_s, times_s)]]
358             return T.li["Per-Server Segment Fetch Response Times: ", l]
359         d.addCallback(_render)
360         return d
361
362 class DownloadStatusPage(DownloadResultsRendererMixin, rend.Page):
363     docFactory = getxmlfile("download-status.xhtml")
364
365     def __init__(self, data):
366         rend.Page.__init__(self, data)
367         self.download_status = data
368
369     def download_results(self):
370         return defer.maybeDeferred(self.download_status.get_results)
371
372     def render_results(self, ctx, data):
373         d = self.download_results()
374         def _got_results(results):
375             if results:
376                 return ctx.tag
377             return ""
378         d.addCallback(_got_results)
379         return d
380
381     def render_started(self, ctx, data):
382         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
383         started_s = time.strftime(TIME_FORMAT,
384                                   time.localtime(data.get_started()))
385         return started_s
386
387     def render_si(self, ctx, data):
388         si_s = base32.b2a_or_none(data.get_storage_index())
389         if si_s is None:
390             si_s = "(None)"
391         return si_s
392
393     def render_helper(self, ctx, data):
394         return {True: "Yes",
395                 False: "No"}[data.using_helper()]
396
397     def render_total_size(self, ctx, data):
398         size = data.get_size()
399         if size is None:
400             size = "(unknown)"
401         return size
402
403     def render_progress(self, ctx, data):
404         progress = data.get_progress()
405         # TODO: make an ascii-art bar
406         return "%.1f%%" % (100.0 * progress)
407
408     def render_status(self, ctx, data):
409         return data.get_status()
410
411 class RetrieveStatusPage(rend.Page, RateAndTimeMixin):
412     docFactory = getxmlfile("retrieve-status.xhtml")
413
414     def __init__(self, data):
415         rend.Page.__init__(self, data)
416         self.retrieve_status = data
417
418     def render_started(self, ctx, data):
419         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
420         started_s = time.strftime(TIME_FORMAT,
421                                   time.localtime(data.get_started()))
422         return started_s
423
424     def render_si(self, ctx, data):
425         si_s = base32.b2a_or_none(data.get_storage_index())
426         if si_s is None:
427             si_s = "(None)"
428         return si_s
429
430     def render_helper(self, ctx, data):
431         return {True: "Yes",
432                 False: "No"}[data.using_helper()]
433
434     def render_current_size(self, ctx, data):
435         size = data.get_size()
436         if size is None:
437             size = "(unknown)"
438         return size
439
440     def render_progress(self, ctx, data):
441         progress = data.get_progress()
442         # TODO: make an ascii-art bar
443         return "%.1f%%" % (100.0 * progress)
444
445     def render_status(self, ctx, data):
446         return data.get_status()
447
448     def render_encoding(self, ctx, data):
449         k, n = data.get_encoding()
450         return ctx.tag["Encoding: %s of %s" % (k, n)]
451
452     def render_problems(self, ctx, data):
453         problems = data.problems
454         if not problems:
455             return ""
456         l = T.ul()
457         for peerid in sorted(problems.keys()):
458             peerid_s = idlib.shortnodeid_b2a(peerid)
459             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
460         return ctx.tag["Server Problems:", l]
461
462     def _get_rate(self, data, name):
463         file_size = self.retrieve_status.get_size()
464         time = self.retrieve_status.timings.get(name)
465         if time is None or file_size is None:
466             return None
467         try:
468             return 1.0 * file_size / time
469         except ZeroDivisionError:
470             return None
471
472     def data_time_total(self, ctx, data):
473         return self.retrieve_status.timings.get("total")
474     def data_rate_total(self, ctx, data):
475         return self._get_rate(data, "total")
476
477     def data_time_peer_selection(self, ctx, data):
478         return self.retrieve_status.timings.get("peer_selection")
479
480     def data_time_fetch(self, ctx, data):
481         return self.retrieve_status.timings.get("fetch")
482     def data_rate_fetch(self, ctx, data):
483         return self._get_rate(data, "fetch")
484
485     def data_time_cumulative_verify(self, ctx, data):
486         return self.retrieve_status.timings.get("cumulative_verify")
487
488     def data_time_decode(self, ctx, data):
489         return self.retrieve_status.timings.get("decode")
490     def data_rate_decode(self, ctx, data):
491         return self._get_rate(data, "decode")
492
493     def data_time_decrypt(self, ctx, data):
494         return self.retrieve_status.timings.get("decrypt")
495     def data_rate_decrypt(self, ctx, data):
496         return self._get_rate(data, "decrypt")
497
498     def render_server_timings(self, ctx, data):
499         per_server = self.retrieve_status.timings.get("fetch_per_server")
500         if not per_server:
501             return ""
502         l = T.ul()
503         for peerid in sorted(per_server.keys()):
504             peerid_s = idlib.shortnodeid_b2a(peerid)
505             times_s = ", ".join([self.render_time(None, t)
506                                  for t in per_server[peerid]])
507             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
508         return T.li["Per-Server Fetch Response Times: ", l]
509
510
511 class PublishStatusPage(rend.Page, RateAndTimeMixin):
512     docFactory = getxmlfile("publish-status.xhtml")
513
514     def __init__(self, data):
515         rend.Page.__init__(self, data)
516         self.publish_status = data
517
518     def render_started(self, ctx, data):
519         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
520         started_s = time.strftime(TIME_FORMAT,
521                                   time.localtime(data.get_started()))
522         return started_s
523
524     def render_si(self, ctx, data):
525         si_s = base32.b2a_or_none(data.get_storage_index())
526         if si_s is None:
527             si_s = "(None)"
528         return si_s
529
530     def render_helper(self, ctx, data):
531         return {True: "Yes",
532                 False: "No"}[data.using_helper()]
533
534     def render_current_size(self, ctx, data):
535         size = data.get_size()
536         if size is None:
537             size = "(unknown)"
538         return size
539
540     def render_progress(self, ctx, data):
541         progress = data.get_progress()
542         # TODO: make an ascii-art bar
543         return "%.1f%%" % (100.0 * progress)
544
545     def render_status(self, ctx, data):
546         return data.get_status()
547
548     def render_encoding(self, ctx, data):
549         k, n = data.get_encoding()
550         return ctx.tag["Encoding: %s of %s" % (k, n)]
551
552     def render_sharemap(self, ctx, data):
553         servermap = data.get_servermap()
554         if servermap is None:
555             return ctx.tag["None"]
556         l = T.ul()
557         sharemap = servermap.make_sharemap()
558         for shnum in sorted(sharemap.keys()):
559             l[T.li["%d -> Placed on " % shnum,
560                    ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
561                               for peerid in sharemap[shnum]])]]
562         return ctx.tag["Sharemap:", l]
563
564     def render_problems(self, ctx, data):
565         problems = data.problems
566         if not problems:
567             return ""
568         l = T.ul()
569         for peerid in sorted(problems.keys()):
570             peerid_s = idlib.shortnodeid_b2a(peerid)
571             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
572         return ctx.tag["Server Problems:", l]
573
574     def _get_rate(self, data, name):
575         file_size = self.publish_status.get_size()
576         time = self.publish_status.timings.get(name)
577         if time is None:
578             return None
579         try:
580             return 1.0 * file_size / time
581         except ZeroDivisionError:
582             return None
583
584     def data_time_total(self, ctx, data):
585         return self.publish_status.timings.get("total")
586     def data_rate_total(self, ctx, data):
587         return self._get_rate(data, "total")
588
589     def data_time_setup(self, ctx, data):
590         return self.publish_status.timings.get("setup")
591
592     def data_time_encrypt(self, ctx, data):
593         return self.publish_status.timings.get("encrypt")
594     def data_rate_encrypt(self, ctx, data):
595         return self._get_rate(data, "encrypt")
596
597     def data_time_encode(self, ctx, data):
598         return self.publish_status.timings.get("encode")
599     def data_rate_encode(self, ctx, data):
600         return self._get_rate(data, "encode")
601
602     def data_time_pack(self, ctx, data):
603         return self.publish_status.timings.get("pack")
604     def data_rate_pack(self, ctx, data):
605         return self._get_rate(data, "pack")
606     def data_time_sign(self, ctx, data):
607         return self.publish_status.timings.get("sign")
608
609     def data_time_push(self, ctx, data):
610         return self.publish_status.timings.get("push")
611     def data_rate_push(self, ctx, data):
612         return self._get_rate(data, "push")
613
614     def render_server_timings(self, ctx, data):
615         per_server = self.publish_status.timings.get("send_per_server")
616         if not per_server:
617             return ""
618         l = T.ul()
619         for peerid in sorted(per_server.keys()):
620             peerid_s = idlib.shortnodeid_b2a(peerid)
621             times_s = ", ".join([self.render_time(None, t)
622                                  for t in per_server[peerid]])
623             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
624         return T.li["Per-Server Response Times: ", l]
625
626
627 class Status(rend.Page):
628     docFactory = getxmlfile("status.xhtml")
629     addSlash = True
630
631     def data_active_operations(self, ctx, data):
632         client = IClient(ctx)
633         active =  (client.list_active_uploads() +
634                    client.list_active_downloads() +
635                    client.list_active_publish() +
636                    client.list_active_retrieve() +
637                    client.list_active_helper_statuses())
638         return active
639
640     def data_recent_operations(self, ctx, data):
641         client = IClient(ctx)
642         recent = [o for o in (client.list_recent_uploads() +
643                               client.list_recent_downloads() +
644                               client.list_recent_publish() +
645                               client.list_recent_retrieve() +
646                               client.list_recent_helper_statuses())
647                   if not o.get_active()]
648         recent.sort(lambda a,b: cmp(a.get_started(), b.get_started()))
649         recent.reverse()
650         return recent
651
652     def render_row(self, ctx, data):
653         s = data
654
655         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
656         started_s = time.strftime(TIME_FORMAT,
657                                   time.localtime(s.get_started()))
658         ctx.fillSlots("started", started_s)
659
660         si_s = base32.b2a_or_none(s.get_storage_index())
661         if si_s is None:
662             si_s = "(None)"
663         ctx.fillSlots("si", si_s)
664         ctx.fillSlots("helper", {True: "Yes",
665                                  False: "No"}[s.using_helper()])
666
667         size = s.get_size()
668         if size is None:
669             size = "(unknown)"
670         ctx.fillSlots("total_size", size)
671
672         progress = data.get_progress()
673         if IUploadStatus.providedBy(data):
674             link = "up-%d" % data.get_counter()
675             ctx.fillSlots("type", "upload")
676             # TODO: make an ascii-art bar
677             (chk, ciphertext, encandpush) = progress
678             progress_s = ("hash: %.1f%%, ciphertext: %.1f%%, encode: %.1f%%" %
679                           ( (100.0 * chk),
680                             (100.0 * ciphertext),
681                             (100.0 * encandpush) ))
682             ctx.fillSlots("progress", progress_s)
683         elif IDownloadStatus.providedBy(data):
684             link = "down-%d" % data.get_counter()
685             ctx.fillSlots("type", "download")
686             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
687         elif IPublishStatus.providedBy(data):
688             link = "publish-%d" % data.get_counter()
689             ctx.fillSlots("type", "publish")
690             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
691         else:
692             assert IRetrieveStatus.providedBy(data)
693             ctx.fillSlots("type", "retrieve")
694             link = "retrieve-%d" % data.get_counter()
695             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
696         ctx.fillSlots("status", T.a(href=link)[s.get_status()])
697         return ctx.tag
698
699     def childFactory(self, ctx, name):
700         client = IClient(ctx)
701         stype,count_s = name.split("-")
702         count = int(count_s)
703         if stype == "up":
704             for s in client.list_recent_uploads():
705                 if s.get_counter() == count:
706                     return UploadStatusPage(s)
707             for u in client.list_all_uploads():
708                 # u is an uploader object
709                 s = u.get_upload_status()
710                 if s.get_counter() == count:
711                     return UploadStatusPage(s)
712             for s in (client.list_active_helper_statuses() +
713                       client.list_recent_helper_statuses()):
714                 if s.get_counter() == count:
715                     # immutable-upload helpers use the same status object as
716                     # a regular immutable-upload
717                     return UploadStatusPage(s)
718         if stype == "down":
719             for s in client.list_recent_downloads():
720                 if s.get_counter() == count:
721                     return DownloadStatusPage(s)
722             for d in client.list_all_downloads():
723                 s = d.get_download_status()
724                 if s.get_counter() == count:
725                     return DownloadStatusPage(s)
726         if stype == "publish":
727             for s in client.list_recent_publish():
728                 if s.get_counter() == count:
729                     return PublishStatusPage(s)
730             for p in client.list_all_publish():
731                 s = p.get_status()
732                 if s.get_counter() == count:
733                     return PublishStatusPage(s)
734         if stype == "retrieve":
735             for s in client.list_recent_retrieve():
736                 if s.get_counter() == count:
737                     return RetrieveStatusPage(s)
738             for r in client.list_all_retrieve():
739                 s = r.get_status()
740                 if s.get_counter() == count:
741                     return RetrieveStatusPage(s)
742
743
744 class HelperStatus(rend.Page):
745     docFactory = getxmlfile("helper.xhtml")
746
747     def renderHTTP(self, ctx):
748         t = get_arg(inevow.IRequest(ctx), "t")
749         if t == "json":
750             return self.render_JSON(ctx)
751         # is there a better way to provide 'data' to all rendering methods?
752         helper = IClient(ctx).getServiceNamed("helper")
753         self.original = helper.get_stats()
754         return rend.Page.renderHTTP(self, ctx)
755
756     def render_JSON(self, ctx):
757         try:
758             h = IClient(ctx).getServiceNamed("helper")
759         except KeyError:
760             return simplejson.dumps({})
761
762         stats = h.get_stats()
763         return simplejson.dumps(stats, indent=1)
764
765     def render_active_uploads(self, ctx, data):
766         return data["chk_upload_helper.active_uploads"]
767
768     def render_incoming(self, ctx, data):
769         return "%d bytes in %d files" % (data["chk_upload_helper.incoming_size"],
770                                          data["chk_upload_helper.incoming_count"])
771
772     def render_encoding(self, ctx, data):
773         return "%d bytes in %d files" % (data["chk_upload_helper.encoding_size"],
774                                          data["chk_upload_helper.encoding_count"])
775
776     def render_upload_requests(self, ctx, data):
777         return str(data["chk_upload_helper.upload_requests"])
778
779     def render_upload_already_present(self, ctx, data):
780         return str(data["chk_upload_helper.upload_already_present"])
781
782     def render_upload_need_upload(self, ctx, data):
783         return str(data["chk_upload_helper.upload_need_upload"])
784
785     def render_upload_bytes_fetched(self, ctx, data):
786         return str(data["chk_upload_helper.fetched_bytes"])
787
788     def render_upload_bytes_encoded(self, ctx, data):
789         return str(data["chk_upload_helper.encoded_bytes"])
790
791
792 class Statistics(rend.Page):
793     docFactory = getxmlfile("statistics.xhtml")
794
795     def renderHTTP(self, ctx):
796         provider = IClient(ctx).stats_provider
797         stats = {'stats': {}, 'counters': {}}
798         if provider:
799             stats = provider.get_stats()
800         t = get_arg(inevow.IRequest(ctx), "t")
801         if t == "json":
802             return simplejson.dumps(stats, indent=1)
803         # is there a better way to provide 'data' to all rendering methods?
804         self.original = stats
805         return rend.Page.renderHTTP(self, ctx)
806
807     def render_load_average(self, ctx, data):
808         return str(data["stats"].get("load_monitor.avg_load"))
809
810     def render_peak_load(self, ctx, data):
811         return str(data["stats"].get("load_monitor.max_load"))
812
813     def render_uploads(self, ctx, data):
814         files = data["counters"].get("uploader.files_uploaded")
815         bytes = data["counters"].get("uploader.bytes_uploaded")
816         return ("%s files / %s bytes (%s)" %
817                 (files, bytes, abbreviate_size(bytes)))
818
819     def render_downloads(self, ctx, data):
820         files = data["counters"].get("downloader.files_downloaded")
821         bytes = data["counters"].get("downloader.bytes_downloaded")
822         return ("%s files / %s bytes (%s)" %
823                 (files, bytes, abbreviate_size(bytes)))
824
825     def render_publishes(self, ctx, data):
826         files = data["counters"].get("mutable.files_published")
827         return "%s files" % (files,)
828
829     def render_retrieves(self, ctx, data):
830         files = data["counters"].get("mutable.files_retrieved")
831         return "%s files" % (files,)
832
833     def render_raw(self, ctx, data):
834         raw = pprint.pformat(data)
835         return ctx.tag[raw]