src/allmydata/web/status.py

   1
   2 import time, pprint, itertools
   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 getxmlfile, get_arg, \
   8      abbreviate_time, abbreviate_rate, abbreviate_size, plural, compute_rate
   9 from allmydata.interfaces import IUploadStatus, IDownloadStatus, \
  10      IPublishStatus, IRetrieveStatus, IServermapUpdaterStatus
  11
  12 class RateAndTimeMixin:
  13
  14     def render_time(self, ctx, data):
  15         return abbreviate_time(data)
  16
  17     def render_rate(self, ctx, data):
  18         return abbreviate_rate(data)
  19
  20 class UploadResultsRendererMixin(RateAndTimeMixin):
  21     # this requires a method named 'upload_results'
  22
  23     def render_pushed_shares(self, ctx, data):
  24         d = self.upload_results()
  25         d.addCallback(lambda res: res.pushed_shares)
  26         return d
  27
  28     def render_preexisting_shares(self, ctx, data):
  29         d = self.upload_results()
  30         d.addCallback(lambda res: res.preexisting_shares)
  31         return d
  32
  33     def render_sharemap(self, ctx, data):
  34         d = self.upload_results()
  35         d.addCallback(lambda res: res.sharemap)
  36         def _render(sharemap):
  37             if sharemap is None:
  38                 return "None"
  39             l = T.ul()
  40             for shnum, peerids in sorted(sharemap.items()):
  41                 peerids = ', '.join([idlib.shortnodeid_b2a(i) for i in peerids])
  42                 l[T.li["%d -> placed on [%s]" % (shnum, peerids)]]
  43             return l
  44         d.addCallback(_render)
  45         return d
  46
  47     def render_servermap(self, ctx, data):
  48         d = self.upload_results()
  49         d.addCallback(lambda res: res.servermap)
  50         def _render(servermap):
  51             if servermap is None:
  52                 return "None"
  53             l = T.ul()
  54             for peerid in sorted(servermap.keys()):
  55                 peerid_s = idlib.shortnodeid_b2a(peerid)
  56                 shares_s = ",".join(["#%d" % shnum
  57                                      for shnum in servermap[peerid]])
  58                 l[T.li["[%s] got share%s: %s" % (peerid_s,
  59                                                  plural(servermap[peerid]),
  60                                                  shares_s)]]
  61             return l
  62         d.addCallback(_render)
  63         return d
  64
  65     def data_file_size(self, ctx, data):
  66         d = self.upload_results()
  67         d.addCallback(lambda res: res.file_size)
  68         return d
  69
  70     def _get_time(self, name):
  71         d = self.upload_results()
  72         d.addCallback(lambda res: res.timings.get(name))
  73         return d
  74
  75     def data_time_total(self, ctx, data):
  76         return self._get_time("total")
  77
  78     def data_time_storage_index(self, ctx, data):
  79         return self._get_time("storage_index")
  80
  81     def data_time_contacting_helper(self, ctx, data):
  82         return self._get_time("contacting_helper")
  83
  84     def data_time_existence_check(self, ctx, data):
  85         return self._get_time("existence_check")
  86
  87     def data_time_cumulative_fetch(self, ctx, data):
  88         return self._get_time("cumulative_fetch")
  89
  90     def data_time_helper_total(self, ctx, data):
  91         return self._get_time("helper_total")
  92
  93     def data_time_peer_selection(self, ctx, data):
  94         return self._get_time("peer_selection")
  95
  96     def data_time_total_encode_and_push(self, ctx, data):
  97         return self._get_time("total_encode_and_push")
  98
  99     def data_time_cumulative_encoding(self, ctx, data):
 100         return self._get_time("cumulative_encoding")
 101
 102     def data_time_cumulative_sending(self, ctx, data):
 103         return self._get_time("cumulative_sending")
 104
 105     def data_time_hashes_and_close(self, ctx, data):
 106         return self._get_time("hashes_and_close")
 107
 108     def _get_rate(self, name):
 109         d = self.upload_results()
 110         def _convert(r):
 111             file_size = r.file_size
 112             time = r.timings.get(name)
 113             return compute_rate(file_size, time)
 114         d.addCallback(_convert)
 115         return d
 116
 117     def data_rate_total(self, ctx, data):
 118         return self._get_rate("total")
 119
 120     def data_rate_storage_index(self, ctx, data):
 121         return self._get_rate("storage_index")
 122
 123     def data_rate_encode(self, ctx, data):
 124         return self._get_rate("cumulative_encoding")
 125
 126     def data_rate_push(self, ctx, data):
 127         return self._get_rate("cumulative_sending")
 128
 129     def data_rate_encode_and_push(self, ctx, data):
 130         d = self.upload_results()
 131         def _convert(r):
 132             file_size = r.file_size
 133             time1 = r.timings.get("cumulative_encoding")
 134             time2 = r.timings.get("cumulative_sending")
 135             if (time1 is None or time2 is None):
 136                 return None
 137             else:
 138                 return compute_rate(file_size, time1+time2)
 139         d.addCallback(_convert)
 140         return d
 141
 142     def data_rate_ciphertext_fetch(self, ctx, data):
 143         d = self.upload_results()
 144         def _convert(r):
 145             fetch_size = r.ciphertext_fetched
 146             time = r.timings.get("cumulative_fetch")
 147             return compute_rate(fetch_size, time)
 148         d.addCallback(_convert)
 149         return d
 150
 151 class UploadStatusPage(UploadResultsRendererMixin, rend.Page):
 152     docFactory = getxmlfile("upload-status.xhtml")
 153
 154     def __init__(self, data):
 155         rend.Page.__init__(self, data)
 156         self.upload_status = data
 157
 158     def upload_results(self):
 159         return defer.maybeDeferred(self.upload_status.get_results)
 160
 161     def render_results(self, ctx, data):
 162         d = self.upload_results()
 163         def _got_results(results):
 164             if results:
 165                 return ctx.tag
 166             return ""
 167         d.addCallback(_got_results)
 168         return d
 169
 170     def render_started(self, ctx, data):
 171         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 172         started_s = time.strftime(TIME_FORMAT,
 173                                   time.localtime(data.get_started()))
 174         return started_s
 175
 176     def render_si(self, ctx, data):
 177         si_s = base32.b2a_or_none(data.get_storage_index())
 178         if si_s is None:
 179             si_s = "(None)"
 180         return si_s
 181
 182     def render_helper(self, ctx, data):
 183         return {True: "Yes",
 184                 False: "No"}[data.using_helper()]
 185
 186     def render_total_size(self, ctx, data):
 187         size = data.get_size()
 188         if size is None:
 189             return "(unknown)"
 190         return size
 191
 192     def render_progress_hash(self, ctx, data):
 193         progress = data.get_progress()[0]
 194         # TODO: make an ascii-art bar
 195         return "%.1f%%" % (100.0 * progress)
 196
 197     def render_progress_ciphertext(self, ctx, data):
 198         progress = data.get_progress()[1]
 199         # TODO: make an ascii-art bar
 200         return "%.1f%%" % (100.0 * progress)
 201
 202     def render_progress_encode_push(self, ctx, data):
 203         progress = data.get_progress()[2]
 204         # TODO: make an ascii-art bar
 205         return "%.1f%%" % (100.0 * progress)
 206
 207     def render_status(self, ctx, data):
 208         return data.get_status()
 209
 210 class DownloadResultsRendererMixin(RateAndTimeMixin):
 211     # this requires a method named 'download_results'
 212
 213     def render_servermap(self, ctx, data):
 214         d = self.download_results()
 215         d.addCallback(lambda res: res.servermap)
 216         def _render(servermap):
 217             if servermap is None:
 218                 return "None"
 219             l = T.ul()
 220             for peerid in sorted(servermap.keys()):
 221                 peerid_s = idlib.shortnodeid_b2a(peerid)
 222                 shares_s = ",".join(["#%d" % shnum
 223                                      for shnum in servermap[peerid]])
 224                 l[T.li["[%s] has share%s: %s" % (peerid_s,
 225                                                  plural(servermap[peerid]),
 226                                                  shares_s)]]
 227             return l
 228         d.addCallback(_render)
 229         return d
 230
 231     def render_servers_used(self, ctx, data):
 232         d = self.download_results()
 233         d.addCallback(lambda res: res.servers_used)
 234         def _got(servers_used):
 235             if not servers_used:
 236                 return ""
 237             peerids_s = ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 238                                    for peerid in servers_used])
 239             return T.li["Servers Used: ", peerids_s]
 240         d.addCallback(_got)
 241         return d
 242
 243     def render_problems(self, ctx, data):
 244         d = self.download_results()
 245         d.addCallback(lambda res: res.server_problems)
 246         def _got(server_problems):
 247             if not server_problems:
 248                 return ""
 249             l = T.ul()
 250             for peerid in sorted(server_problems.keys()):
 251                 peerid_s = idlib.shortnodeid_b2a(peerid)
 252                 l[T.li["[%s]: %s" % (peerid_s, server_problems[peerid])]]
 253             return T.li["Server Problems:", l]
 254         d.addCallback(_got)
 255         return d
 256
 257     def data_file_size(self, ctx, data):
 258         d = self.download_results()
 259         d.addCallback(lambda res: res.file_size)
 260         return d
 261
 262     def _get_time(self, name):
 263         d = self.download_results()
 264         d.addCallback(lambda res: res.timings.get(name))
 265         return d
 266
 267     def data_time_total(self, ctx, data):
 268         return self._get_time("total")
 269
 270     def data_time_peer_selection(self, ctx, data):
 271         return self._get_time("peer_selection")
 272
 273     def data_time_uri_extension(self, ctx, data):
 274         return self._get_time("uri_extension")
 275
 276     def data_time_hashtrees(self, ctx, data):
 277         return self._get_time("hashtrees")
 278
 279     def data_time_segments(self, ctx, data):
 280         return self._get_time("segments")
 281
 282     def data_time_cumulative_fetch(self, ctx, data):
 283         return self._get_time("cumulative_fetch")
 284
 285     def data_time_cumulative_decode(self, ctx, data):
 286         return self._get_time("cumulative_decode")
 287
 288     def data_time_cumulative_decrypt(self, ctx, data):
 289         return self._get_time("cumulative_decrypt")
 290
 291     def data_time_paused(self, ctx, data):
 292         return self._get_time("paused")
 293
 294     def _get_rate(self, name):
 295         d = self.download_results()
 296         def _convert(r):
 297             file_size = r.file_size
 298             time = r.timings.get(name)
 299             return compute_rate(file_size, time)
 300         d.addCallback(_convert)
 301         return d
 302
 303     def data_rate_total(self, ctx, data):
 304         return self._get_rate("total")
 305
 306     def data_rate_segments(self, ctx, data):
 307         return self._get_rate("segments")
 308
 309     def data_rate_fetch(self, ctx, data):
 310         return self._get_rate("cumulative_fetch")
 311
 312     def data_rate_decode(self, ctx, data):
 313         return self._get_rate("cumulative_decode")
 314
 315     def data_rate_decrypt(self, ctx, data):
 316         return self._get_rate("cumulative_decrypt")
 317
 318     def render_server_timings(self, ctx, data):
 319         d = self.download_results()
 320         d.addCallback(lambda res: res.timings.get("fetch_per_server"))
 321         def _render(per_server):
 322             if per_server is None:
 323                 return ""
 324             l = T.ul()
 325             for peerid in sorted(per_server.keys()):
 326                 peerid_s = idlib.shortnodeid_b2a(peerid)
 327                 times_s = ", ".join([self.render_time(None, t)
 328                                      for t in per_server[peerid]])
 329                 l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 330             return T.li["Per-Server Segment Fetch Response Times: ", l]
 331         d.addCallback(_render)
 332         return d
 333
 334 class DownloadStatusPage(DownloadResultsRendererMixin, rend.Page):
 335     docFactory = getxmlfile("download-status.xhtml")
 336
 337     def __init__(self, data):
 338         rend.Page.__init__(self, data)
 339         self.download_status = data
 340
 341     def download_results(self):
 342         return defer.maybeDeferred(self.download_status.get_results)
 343
 344     def relative_time(self, t):
 345         if t is None:
 346             return t
 347         if self.download_status.started is not None:
 348             return t - self.download_status.started
 349         return t
 350     def short_relative_time(self, t):
 351         t = self.relative_time(t)
 352         if t is None:
 353             return ""
 354         return "+%.6fs" % t
 355
 356     def renderHTTP(self, ctx):
 357         req = inevow.IRequest(ctx)
 358         t = get_arg(req, "t")
 359         if t == "json":
 360             return self.json(req)
 361         return rend.Page.renderHTTP(self, ctx)
 362
 363     def json(self, req):
 364         req.setHeader("content-type", "text/plain")
 365         data = {}
 366         dyhb_events = []
 367         for serverid,requests in self.download_status.dyhb_requests.iteritems():
 368             for req in requests:
 369                 dyhb_events.append( (base32.b2a(serverid),) + req )
 370         dyhb_events.sort(key=lambda req: req[1])
 371         data["dyhb"] = dyhb_events
 372         request_events = []
 373         for serverid,requests in self.download_status.requests.iteritems():
 374             for req in requests:
 375                 request_events.append( (base32.b2a(serverid),) + req )
 376         request_events.sort(key=lambda req: (req[4],req[1]))
 377         data["requests"] = request_events
 378         data["segment"] = self.download_status.segment_events
 379         data["read"] = self.download_status.read_events
 380         return simplejson.dumps(data, indent=1) + "\n"
 381
 382     def render_events(self, ctx, data):
 383         if not self.download_status.storage_index:
 384             return
 385         srt = self.short_relative_time
 386         l = T.div()
 387
 388         t = T.table(align="left", class_="status-download-events")
 389         t[T.tr[T.th["serverid"], T.th["sent"], T.th["received"],
 390                T.th["shnums"], T.th["RTT"]]]
 391         dyhb_events = []
 392         for serverid,requests in self.download_status.dyhb_requests.iteritems():
 393             for req in requests:
 394                 dyhb_events.append( (serverid,) + req )
 395         dyhb_events.sort(key=lambda req: req[1])
 396         for d_ev in dyhb_events:
 397             (serverid, sent, shnums, received) = d_ev
 398             serverid_s = idlib.shortnodeid_b2a(serverid)
 399             rtt = None
 400             if received is not None:
 401                 rtt = received - sent
 402             if not shnums:
 403                 shnums = ["-"]
 404             t[T.tr(style="background: %s" % self.color(serverid))[
 405                 [T.td[serverid_s], T.td[srt(sent)], T.td[srt(received)],
 406                  T.td[",".join([str(shnum) for shnum in shnums])],
 407                  T.td[self.render_time(None, rtt)],
 408                  ]]]
 409
 410         l[T.h2["DYHB Requests:"], t]
 411         l[T.br(clear="all")]
 412
 413         t = T.table(align="left",class_="status-download-events")
 414         t[T.tr[T.th["range"], T.th["start"], T.th["finish"], T.th["got"],
 415                T.th["time"], T.th["decrypttime"], T.th["pausedtime"],
 416                T.th["speed"]]]
 417         for r_ev in self.download_status.read_events:
 418             (start, length, requesttime, finishtime, bytes, decrypt, paused) = r_ev
 419             if finishtime is not None:
 420                 rtt = finishtime - requesttime - paused
 421                 speed = self.render_rate(None, compute_rate(bytes, rtt))
 422                 rtt = self.render_time(None, rtt)
 423                 decrypt = self.render_time(None, decrypt)
 424                 paused = self.render_time(None, paused)
 425             else:
 426                 speed, rtt, decrypt, paused = "","","",""
 427             t[T.tr[T.td["[%d:+%d]" % (start, length)],
 428                    T.td[srt(requesttime)], T.td[srt(finishtime)],
 429                    T.td[bytes], T.td[rtt], T.td[decrypt], T.td[paused],
 430                    T.td[speed],
 431                    ]]
 432
 433         l[T.h2["Read Events:"], t]
 434         l[T.br(clear="all")]
 435
 436         t = T.table(align="left",class_="status-download-events")
 437         t[T.tr[T.th["type"], T.th["segnum"], T.th["when"], T.th["range"],
 438                T.th["decodetime"], T.th["segtime"], T.th["speed"]]]
 439         reqtime = (None, None)
 440         for s_ev in self.download_status.segment_events:
 441             (etype, segnum, when, segstart, seglen, decodetime) = s_ev
 442             if etype == "request":
 443                 t[T.tr[T.td["request"],
 444                     T.td["seg%d" % segnum],
 445                     T.td[srt(when)],
 446                     T.td["-"],
 447                     T.td["-"],
 448                     T.td["-"],
 449                     T.td["-"]]]
 450
 451                 reqtime = (segnum, when)
 452             elif etype == "delivery":
 453                 if reqtime[0] == segnum:
 454                     segtime = when - reqtime[1]
 455                     speed = self.render_rate(None, compute_rate(seglen, segtime))
 456                     segtime = self.render_time(None, segtime)
 457                 else:
 458                     segtime, speed = "", ""
 459                 t[T.tr[T.td["delivery"], T.td["seg%d" % segnum],
 460                        T.td[srt(when)],
 461                        T.td["[%d:+%d]" % (segstart, seglen)],
 462                        T.td[self.render_time(None,decodetime)],
 463                        T.td[segtime], T.td[speed]]]
 464             elif etype == "error":
 465                 t[T.tr[T.td["error"], T.td["seg%d" % segnum]]]
 466
 467         l[T.h2["Segment Events:"], t]
 468         l[T.br(clear="all")]
 469
 470         t = T.table(align="left",class_="status-download-events")
 471         t[T.tr[T.th["serverid"], T.th["shnum"], T.th["range"],
 472                T.th["txtime"], T.th["rxtime"], T.th["received"], T.th["RTT"]]]
 473         reqtime = (None, None)
 474         request_events = []
 475         for serverid,requests in self.download_status.requests.iteritems():
 476             for req in requests:
 477                 request_events.append( (serverid,) + req )
 478         request_events.sort(key=lambda req: (req[4],req[1]))
 479         for r_ev in request_events:
 480             (peerid, shnum, start, length, sent, receivedlen, received) = r_ev
 481             rtt = None
 482             if received is not None:
 483                 rtt = received - sent
 484             peerid_s = idlib.shortnodeid_b2a(peerid)
 485             t[T.tr(style="background: %s" % self.color(peerid))[
 486                 T.td[peerid_s], T.td[shnum],
 487                 T.td["[%d:+%d]" % (start, length)],
 488                 T.td[srt(sent)], T.td[srt(received)], T.td[receivedlen],
 489                 T.td[self.render_time(None, rtt)],
 490                 ]]
 491
 492         l[T.h2["Requests:"], t]
 493         l[T.br(clear="all")]
 494
 495         return l
 496
 497     def color(self, peerid):
 498         def m(c):
 499             return min(ord(c) / 2 + 0x80, 0xff)
 500         return "#%02x%02x%02x" % (m(peerid[0]), m(peerid[1]), m(peerid[2]))
 501
 502     def render_results(self, ctx, data):
 503         d = self.download_results()
 504         def _got_results(results):
 505             if results:
 506                 return ctx.tag
 507             return ""
 508         d.addCallback(_got_results)
 509         return d
 510
 511     def render_started(self, ctx, data):
 512         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 513         started_s = time.strftime(TIME_FORMAT,
 514                                   time.localtime(data.get_started()))
 515         return started_s + " (%s)" % data.get_started()
 516
 517     def render_si(self, ctx, data):
 518         si_s = base32.b2a_or_none(data.get_storage_index())
 519         if si_s is None:
 520             si_s = "(None)"
 521         return si_s
 522
 523     def render_helper(self, ctx, data):
 524         return {True: "Yes",
 525                 False: "No"}[data.using_helper()]
 526
 527     def render_total_size(self, ctx, data):
 528         size = data.get_size()
 529         if size is None:
 530             return "(unknown)"
 531         return size
 532
 533     def render_progress(self, ctx, data):
 534         progress = data.get_progress()
 535         # TODO: make an ascii-art bar
 536         return "%.1f%%" % (100.0 * progress)
 537
 538     def render_status(self, ctx, data):
 539         return data.get_status()
 540
 541 class RetrieveStatusPage(rend.Page, RateAndTimeMixin):
 542     docFactory = getxmlfile("retrieve-status.xhtml")
 543
 544     def __init__(self, data):
 545         rend.Page.__init__(self, data)
 546         self.retrieve_status = data
 547
 548     def render_started(self, ctx, data):
 549         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 550         started_s = time.strftime(TIME_FORMAT,
 551                                   time.localtime(data.get_started()))
 552         return started_s
 553
 554     def render_si(self, ctx, data):
 555         si_s = base32.b2a_or_none(data.get_storage_index())
 556         if si_s is None:
 557             si_s = "(None)"
 558         return si_s
 559
 560     def render_helper(self, ctx, data):
 561         return {True: "Yes",
 562                 False: "No"}[data.using_helper()]
 563
 564     def render_current_size(self, ctx, data):
 565         size = data.get_size()
 566         if size is None:
 567             size = "(unknown)"
 568         return size
 569
 570     def render_progress(self, ctx, data):
 571         progress = data.get_progress()
 572         # TODO: make an ascii-art bar
 573         return "%.1f%%" % (100.0 * progress)
 574
 575     def render_status(self, ctx, data):
 576         return data.get_status()
 577
 578     def render_encoding(self, ctx, data):
 579         k, n = data.get_encoding()
 580         return ctx.tag["Encoding: %s of %s" % (k, n)]
 581
 582     def render_problems(self, ctx, data):
 583         problems = data.problems
 584         if not problems:
 585             return ""
 586         l = T.ul()
 587         for peerid in sorted(problems.keys()):
 588             peerid_s = idlib.shortnodeid_b2a(peerid)
 589             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 590         return ctx.tag["Server Problems:", l]
 591
 592     def _get_rate(self, data, name):
 593         file_size = self.retrieve_status.get_size()
 594         time = self.retrieve_status.timings.get(name)
 595         return compute_rate(file_size, time)
 596
 597     def data_time_total(self, ctx, data):
 598         return self.retrieve_status.timings.get("total")
 599     def data_rate_total(self, ctx, data):
 600         return self._get_rate(data, "total")
 601
 602     def data_time_fetch(self, ctx, data):
 603         return self.retrieve_status.timings.get("fetch")
 604     def data_rate_fetch(self, ctx, data):
 605         return self._get_rate(data, "fetch")
 606
 607     def data_time_decode(self, ctx, data):
 608         return self.retrieve_status.timings.get("decode")
 609     def data_rate_decode(self, ctx, data):
 610         return self._get_rate(data, "decode")
 611
 612     def data_time_decrypt(self, ctx, data):
 613         return self.retrieve_status.timings.get("decrypt")
 614     def data_rate_decrypt(self, ctx, data):
 615         return self._get_rate(data, "decrypt")
 616
 617     def render_server_timings(self, ctx, data):
 618         per_server = self.retrieve_status.timings.get("fetch_per_server")
 619         if not per_server:
 620             return ""
 621         l = T.ul()
 622         for peerid in sorted(per_server.keys()):
 623             peerid_s = idlib.shortnodeid_b2a(peerid)
 624             times_s = ", ".join([self.render_time(None, t)
 625                                  for t in per_server[peerid]])
 626             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 627         return T.li["Per-Server Fetch Response Times: ", l]
 628
 629
 630 class PublishStatusPage(rend.Page, RateAndTimeMixin):
 631     docFactory = getxmlfile("publish-status.xhtml")
 632
 633     def __init__(self, data):
 634         rend.Page.__init__(self, data)
 635         self.publish_status = data
 636
 637     def render_started(self, ctx, data):
 638         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 639         started_s = time.strftime(TIME_FORMAT,
 640                                   time.localtime(data.get_started()))
 641         return started_s
 642
 643     def render_si(self, ctx, data):
 644         si_s = base32.b2a_or_none(data.get_storage_index())
 645         if si_s is None:
 646             si_s = "(None)"
 647         return si_s
 648
 649     def render_helper(self, ctx, data):
 650         return {True: "Yes",
 651                 False: "No"}[data.using_helper()]
 652
 653     def render_current_size(self, ctx, data):
 654         size = data.get_size()
 655         if size is None:
 656             size = "(unknown)"
 657         return size
 658
 659     def render_progress(self, ctx, data):
 660         progress = data.get_progress()
 661         # TODO: make an ascii-art bar
 662         return "%.1f%%" % (100.0 * progress)
 663
 664     def render_status(self, ctx, data):
 665         return data.get_status()
 666
 667     def render_encoding(self, ctx, data):
 668         k, n = data.get_encoding()
 669         return ctx.tag["Encoding: %s of %s" % (k, n)]
 670
 671     def render_sharemap(self, ctx, data):
 672         servermap = data.get_servermap()
 673         if servermap is None:
 674             return ctx.tag["None"]
 675         l = T.ul()
 676         sharemap = servermap.make_sharemap()
 677         for shnum in sorted(sharemap.keys()):
 678             l[T.li["%d -> Placed on " % shnum,
 679                    ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 680                               for peerid in sharemap[shnum]])]]
 681         return ctx.tag["Sharemap:", l]
 682
 683     def render_problems(self, ctx, data):
 684         problems = data.problems
 685         if not problems:
 686             return ""
 687         l = T.ul()
 688         for peerid in sorted(problems.keys()):
 689             peerid_s = idlib.shortnodeid_b2a(peerid)
 690             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 691         return ctx.tag["Server Problems:", l]
 692
 693     def _get_rate(self, data, name):
 694         file_size = self.publish_status.get_size()
 695         time = self.publish_status.timings.get(name)
 696         return compute_rate(file_size, time)
 697
 698     def data_time_total(self, ctx, data):
 699         return self.publish_status.timings.get("total")
 700     def data_rate_total(self, ctx, data):
 701         return self._get_rate(data, "total")
 702
 703     def data_time_setup(self, ctx, data):
 704         return self.publish_status.timings.get("setup")
 705
 706     def data_time_encrypt(self, ctx, data):
 707         return self.publish_status.timings.get("encrypt")
 708     def data_rate_encrypt(self, ctx, data):
 709         return self._get_rate(data, "encrypt")
 710
 711     def data_time_encode(self, ctx, data):
 712         return self.publish_status.timings.get("encode")
 713     def data_rate_encode(self, ctx, data):
 714         return self._get_rate(data, "encode")
 715
 716     def data_time_pack(self, ctx, data):
 717         return self.publish_status.timings.get("pack")
 718     def data_rate_pack(self, ctx, data):
 719         return self._get_rate(data, "pack")
 720     def data_time_sign(self, ctx, data):
 721         return self.publish_status.timings.get("sign")
 722
 723     def data_time_push(self, ctx, data):
 724         return self.publish_status.timings.get("push")
 725     def data_rate_push(self, ctx, data):
 726         return self._get_rate(data, "push")
 727
 728     def render_server_timings(self, ctx, data):
 729         per_server = self.publish_status.timings.get("send_per_server")
 730         if not per_server:
 731             return ""
 732         l = T.ul()
 733         for peerid in sorted(per_server.keys()):
 734             peerid_s = idlib.shortnodeid_b2a(peerid)
 735             times_s = ", ".join([self.render_time(None, t)
 736                                  for t in per_server[peerid]])
 737             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 738         return T.li["Per-Server Response Times: ", l]
 739
 740 class MapupdateStatusPage(rend.Page, RateAndTimeMixin):
 741     docFactory = getxmlfile("map-update-status.xhtml")
 742
 743     def __init__(self, data):
 744         rend.Page.__init__(self, data)
 745         self.update_status = data
 746
 747     def render_started(self, ctx, data):
 748         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 749         started_s = time.strftime(TIME_FORMAT,
 750                                   time.localtime(data.get_started()))
 751         return started_s
 752
 753     def render_finished(self, ctx, data):
 754         when = data.get_finished()
 755         if not when:
 756             return "not yet"
 757         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 758         started_s = time.strftime(TIME_FORMAT,
 759                                   time.localtime(data.get_finished()))
 760         return started_s
 761
 762     def render_si(self, ctx, data):
 763         si_s = base32.b2a_or_none(data.get_storage_index())
 764         if si_s is None:
 765             si_s = "(None)"
 766         return si_s
 767
 768     def render_helper(self, ctx, data):
 769         return {True: "Yes",
 770                 False: "No"}[data.using_helper()]
 771
 772     def render_progress(self, ctx, data):
 773         progress = data.get_progress()
 774         # TODO: make an ascii-art bar
 775         return "%.1f%%" % (100.0 * progress)
 776
 777     def render_status(self, ctx, data):
 778         return data.get_status()
 779
 780     def render_problems(self, ctx, data):
 781         problems = data.problems
 782         if not problems:
 783             return ""
 784         l = T.ul()
 785         for peerid in sorted(problems.keys()):
 786             peerid_s = idlib.shortnodeid_b2a(peerid)
 787             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 788         return ctx.tag["Server Problems:", l]
 789
 790     def render_privkey_from(self, ctx, data):
 791         peerid = data.get_privkey_from()
 792         if peerid:
 793             return ctx.tag["Got privkey from: [%s]"
 794                            % idlib.shortnodeid_b2a(peerid)]
 795         else:
 796             return ""
 797
 798     def data_time_total(self, ctx, data):
 799         return self.update_status.timings.get("total")
 800
 801     def data_time_initial_queries(self, ctx, data):
 802         return self.update_status.timings.get("initial_queries")
 803
 804     def data_time_cumulative_verify(self, ctx, data):
 805         return self.update_status.timings.get("cumulative_verify")
 806
 807     def render_server_timings(self, ctx, data):
 808         per_server = self.update_status.timings.get("per_server")
 809         if not per_server:
 810             return ""
 811         l = T.ul()
 812         for peerid in sorted(per_server.keys()):
 813             peerid_s = idlib.shortnodeid_b2a(peerid)
 814             times = []
 815             for op,started,t in per_server[peerid]:
 816                 #times.append("%s/%.4fs/%s/%s" % (op,
 817                 #                              started,
 818                 #                              self.render_time(None, started - self.update_status.get_started()),
 819                 #                              self.render_time(None,t)))
 820                 if op == "query":
 821                     times.append( self.render_time(None, t) )
 822                 elif op == "late":
 823                     times.append( "late(" + self.render_time(None, t) + ")" )
 824                 else:
 825                     times.append( "privkey(" + self.render_time(None, t) + ")" )
 826             times_s = ", ".join(times)
 827             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 828         return T.li["Per-Server Response Times: ", l]
 829
 830     def render_timing_chart(self, ctx, data):
 831         imageurl = self._timing_chart()
 832         return ctx.tag[imageurl]
 833
 834     def _timing_chart(self):
 835         started = self.update_status.get_started()
 836         total = self.update_status.timings.get("total")
 837         per_server = self.update_status.timings.get("per_server")
 838         base = "http://chart.apis.google.com/chart?"
 839         pieces = ["cht=bhs"]
 840         pieces.append("chco=ffffff,4d89f9,c6d9fd") # colors
 841         data0 = []
 842         data1 = []
 843         data2 = []
 844         nb_nodes = 0
 845         graph_botom_margin= 21
 846         graph_top_margin = 5
 847         peerids_s = []
 848         top_abs = started
 849         # we sort the queries by the time at which we sent the first request
 850         sorttable = [ (times[0][1], peerid)
 851                       for peerid, times in per_server.items() ]
 852         sorttable.sort()
 853         peerids = [t[1] for t in sorttable]
 854
 855         for peerid in peerids:
 856             nb_nodes += 1
 857             times = per_server[peerid]
 858             peerid_s = idlib.shortnodeid_b2a(peerid)
 859             peerids_s.append(peerid_s)
 860             # for servermap updates, there are either one or two queries per
 861             # peer. The second (if present) is to get the privkey.
 862             op,q_started,q_elapsed = times[0]
 863             data0.append("%.3f" % (q_started-started))
 864             data1.append("%.3f" % q_elapsed)
 865             top_abs = max(top_abs, q_started+q_elapsed)
 866             if len(times) > 1:
 867                 op,p_started,p_elapsed = times[0]
 868                 data2.append("%.3f" % p_elapsed)
 869                 top_abs = max(top_abs, p_started+p_elapsed)
 870             else:
 871                 data2.append("0.0")
 872         finished = self.update_status.get_finished()
 873         if finished:
 874             top_abs = max(top_abs, finished)
 875         top_rel = top_abs - started
 876         chs ="chs=400x%d" % ( (nb_nodes*28) + graph_top_margin + graph_botom_margin )
 877         chd = "chd=t:" + "|".join([",".join(data0),
 878                                    ",".join(data1),
 879                                    ",".join(data2)])
 880         pieces.append(chd)
 881         pieces.append(chs)
 882         chds = "chds=0,%0.3f" % top_rel
 883         pieces.append(chds)
 884         pieces.append("chxt=x,y")
 885         pieces.append("chxr=0,0.0,%0.3f" % top_rel)
 886         pieces.append("chxl=1:|" + "|".join(reversed(peerids_s)))
 887         # use up to 10 grid lines, at decimal multiples.
 888         # mathutil.next_power_of_k doesn't handle numbers smaller than one,
 889         # unfortunately.
 890         #pieces.append("chg="
 891
 892         if total is not None:
 893             finished_f = 1.0 * total / top_rel
 894             pieces.append("chm=r,FF0000,0,%0.3f,%0.3f" % (finished_f,
 895                                                           finished_f+0.01))
 896         url = base + "&".join(pieces)
 897         return T.img(src=url,border="1",align="right", float="right")
 898
 899
 900 class Status(rend.Page):
 901     docFactory = getxmlfile("status.xhtml")
 902     addSlash = True
 903
 904     def __init__(self, history):
 905         rend.Page.__init__(self, history)
 906         self.history = history
 907
 908     def renderHTTP(self, ctx):
 909         req = inevow.IRequest(ctx)
 910         t = get_arg(req, "t")
 911         if t == "json":
 912             return self.json(req)
 913         return rend.Page.renderHTTP(self, ctx)
 914
 915     def json(self, req):
 916         req.setHeader("content-type", "text/plain")
 917         data = {}
 918         data["active"] = active = []
 919         for s in self._get_active_operations():
 920             si_s = base32.b2a_or_none(s.get_storage_index())
 921             size = s.get_size()
 922             status = s.get_status()
 923             if IUploadStatus.providedBy(s):
 924                 h,c,e = s.get_progress()
 925                 active.append({"type": "upload",
 926                                "storage-index-string": si_s,
 927                                "total-size": size,
 928                                "status": status,
 929                                "progress-hash": h,
 930                                "progress-ciphertext": c,
 931                                "progress-encode-push": e,
 932                                })
 933             elif IDownloadStatus.providedBy(s):
 934                 active.append({"type": "download",
 935                                "storage-index-string": si_s,
 936                                "total-size": size,
 937                                "status": status,
 938                                "progress": s.get_progress(),
 939                                })
 940
 941         return simplejson.dumps(data, indent=1) + "\n"
 942
 943     def _get_all_statuses(self):
 944         h = self.history
 945         return itertools.chain(h.list_all_upload_statuses(),
 946                                h.list_all_download_statuses(),
 947                                h.list_all_mapupdate_statuses(),
 948                                h.list_all_publish_statuses(),
 949                                h.list_all_retrieve_statuses(),
 950                                h.list_all_helper_statuses(),
 951                                )
 952
 953     def data_active_operations(self, ctx, data):
 954         return self._get_active_operations()
 955
 956     def _get_active_operations(self):
 957         active = [s
 958                   for s in self._get_all_statuses()
 959                   if s.get_active()]
 960         return active
 961
 962     def data_recent_operations(self, ctx, data):
 963         return self._get_recent_operations()
 964
 965     def _get_recent_operations(self):
 966         recent = [s
 967                   for s in self._get_all_statuses()
 968                   if not s.get_active()]
 969         recent.sort(lambda a,b: cmp(a.get_started(), b.get_started()))
 970         recent.reverse()
 971         return recent
 972
 973     def render_row(self, ctx, data):
 974         s = data
 975
 976         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 977         started_s = time.strftime(TIME_FORMAT,
 978                                   time.localtime(s.get_started()))
 979         ctx.fillSlots("started", started_s)
 980
 981         si_s = base32.b2a_or_none(s.get_storage_index())
 982         if si_s is None:
 983             si_s = "(None)"
 984         ctx.fillSlots("si", si_s)
 985         ctx.fillSlots("helper", {True: "Yes",
 986                                  False: "No"}[s.using_helper()])
 987
 988         size = s.get_size()
 989         if size is None:
 990             size = "(unknown)"
 991         elif isinstance(size, (int, long, float)):
 992             size = abbreviate_size(size)
 993         ctx.fillSlots("total_size", size)
 994
 995         progress = data.get_progress()
 996         if IUploadStatus.providedBy(data):
 997             link = "up-%d" % data.get_counter()
 998             ctx.fillSlots("type", "upload")
 999             # TODO: make an ascii-art bar
1000             (chk, ciphertext, encandpush) = progress
1001             progress_s = ("hash: %.1f%%, ciphertext: %.1f%%, encode: %.1f%%" %
1002                           ( (100.0 * chk),
1003                             (100.0 * ciphertext),
1004                             (100.0 * encandpush) ))
1005             ctx.fillSlots("progress", progress_s)
1006         elif IDownloadStatus.providedBy(data):
1007             link = "down-%d" % data.get_counter()
1008             ctx.fillSlots("type", "download")
1009             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1010         elif IPublishStatus.providedBy(data):
1011             link = "publish-%d" % data.get_counter()
1012             ctx.fillSlots("type", "publish")
1013             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1014         elif IRetrieveStatus.providedBy(data):
1015             ctx.fillSlots("type", "retrieve")
1016             link = "retrieve-%d" % data.get_counter()
1017             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1018         else:
1019             assert IServermapUpdaterStatus.providedBy(data)
1020             ctx.fillSlots("type", "mapupdate %s" % data.get_mode())
1021             link = "mapupdate-%d" % data.get_counter()
1022             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1023         ctx.fillSlots("status", T.a(href=link)[s.get_status()])
1024         return ctx.tag
1025
1026     def childFactory(self, ctx, name):
1027         h = self.history
1028         stype,count_s = name.split("-")
1029         count = int(count_s)
1030         if stype == "up":
1031             for s in itertools.chain(h.list_all_upload_statuses(),
1032                                      h.list_all_helper_statuses()):
1033                 # immutable-upload helpers use the same status object as a
1034                 # regular immutable-upload
1035                 if s.get_counter() == count:
1036                     return UploadStatusPage(s)
1037         if stype == "down":
1038             for s in h.list_all_download_statuses():
1039                 if s.get_counter() == count:
1040                     return DownloadStatusPage(s)
1041         if stype == "mapupdate":
1042             for s in h.list_all_mapupdate_statuses():
1043                 if s.get_counter() == count:
1044                     return MapupdateStatusPage(s)
1045         if stype == "publish":
1046             for s in h.list_all_publish_statuses():
1047                 if s.get_counter() == count:
1048                     return PublishStatusPage(s)
1049         if stype == "retrieve":
1050             for s in h.list_all_retrieve_statuses():
1051                 if s.get_counter() == count:
1052                     return RetrieveStatusPage(s)
1053
1054
1055 class HelperStatus(rend.Page):
1056     docFactory = getxmlfile("helper.xhtml")
1057
1058     def __init__(self, helper):
1059         rend.Page.__init__(self, helper)
1060         self.helper = helper
1061
1062     def renderHTTP(self, ctx):
1063         req = inevow.IRequest(ctx)
1064         t = get_arg(req, "t")
1065         if t == "json":
1066             return self.render_JSON(req)
1067         return rend.Page.renderHTTP(self, ctx)
1068
1069     def data_helper_stats(self, ctx, data):
1070         return self.helper.get_stats()
1071
1072     def render_JSON(self, req):
1073         req.setHeader("content-type", "text/plain")
1074         if self.helper:
1075             stats = self.helper.get_stats()
1076             return simplejson.dumps(stats, indent=1) + "\n"
1077         return simplejson.dumps({}) + "\n"
1078
1079     def render_active_uploads(self, ctx, data):
1080         return data["chk_upload_helper.active_uploads"]
1081
1082     def render_incoming(self, ctx, data):
1083         return "%d bytes in %d files" % (data["chk_upload_helper.incoming_size"],
1084                                          data["chk_upload_helper.incoming_count"])
1085
1086     def render_encoding(self, ctx, data):
1087         return "%d bytes in %d files" % (data["chk_upload_helper.encoding_size"],
1088                                          data["chk_upload_helper.encoding_count"])
1089
1090     def render_upload_requests(self, ctx, data):
1091         return str(data["chk_upload_helper.upload_requests"])
1092
1093     def render_upload_already_present(self, ctx, data):
1094         return str(data["chk_upload_helper.upload_already_present"])
1095
1096     def render_upload_need_upload(self, ctx, data):
1097         return str(data["chk_upload_helper.upload_need_upload"])
1098
1099     def render_upload_bytes_fetched(self, ctx, data):
1100         return str(data["chk_upload_helper.fetched_bytes"])
1101
1102     def render_upload_bytes_encoded(self, ctx, data):
1103         return str(data["chk_upload_helper.encoded_bytes"])
1104
1105
1106 class Statistics(rend.Page):
1107     docFactory = getxmlfile("statistics.xhtml")
1108
1109     def __init__(self, provider):
1110         rend.Page.__init__(self, provider)
1111         self.provider = provider
1112
1113     def renderHTTP(self, ctx):
1114         req = inevow.IRequest(ctx)
1115         t = get_arg(req, "t")
1116         if t == "json":
1117             stats = self.provider.get_stats()
1118             req.setHeader("content-type", "text/plain")
1119             return simplejson.dumps(stats, indent=1) + "\n"
1120         return rend.Page.renderHTTP(self, ctx)
1121
1122     def data_get_stats(self, ctx, data):
1123         return self.provider.get_stats()
1124
1125     def render_load_average(self, ctx, data):
1126         return str(data["stats"].get("load_monitor.avg_load"))
1127
1128     def render_peak_load(self, ctx, data):
1129         return str(data["stats"].get("load_monitor.max_load"))
1130
1131     def render_uploads(self, ctx, data):
1132         files = data["counters"].get("uploader.files_uploaded", 0)
1133         bytes = data["counters"].get("uploader.bytes_uploaded", 0)
1134         return ("%s files / %s bytes (%s)" %
1135                 (files, bytes, abbreviate_size(bytes)))
1136
1137     def render_downloads(self, ctx, data):
1138         files = data["counters"].get("downloader.files_downloaded", 0)
1139         bytes = data["counters"].get("downloader.bytes_downloaded", 0)
1140         return ("%s files / %s bytes (%s)" %
1141                 (files, bytes, abbreviate_size(bytes)))
1142
1143     def render_publishes(self, ctx, data):
1144         files = data["counters"].get("mutable.files_published", 0)
1145         bytes = data["counters"].get("mutable.bytes_published", 0)
1146         return "%s files / %s bytes (%s)" % (files, bytes,
1147                                              abbreviate_size(bytes))
1148
1149     def render_retrieves(self, ctx, data):
1150         files = data["counters"].get("mutable.files_retrieved", 0)
1151         bytes = data["counters"].get("mutable.bytes_retrieved", 0)
1152         return "%s files / %s bytes (%s)" % (files, bytes,
1153                                              abbreviate_size(bytes))
1154
1155     def render_raw(self, ctx, data):
1156         raw = pprint.pformat(data)
1157         return ctx.tag[raw]