source file: /home/buildslave/tahoe/edgy/build/src/allmydata/web/status.py
file stats: 768 lines, 715 executed: 93.1% covered
   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 IClient, getxmlfile, abbreviate_time, \
   8.      abbreviate_rate, abbreviate_size, get_arg
   9. from allmydata.interfaces import IUploadStatus, IDownloadStatus, \
  10.      IPublishStatus, IRetrieveStatus, IServermapUpdaterStatus
  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.             time1 = r.timings.get("cumulative_encoding")
 147.             time2 = r.timings.get("cumulative_sending")
 148.             if (file_size is None or time1 is None or time2 is None):
 149.                 return None
 150.             try:
 151.                 return 1.0 * file_size / (time1+time2)
 152.             except ZeroDivisionError:
 153.                 return None
 154.         d.addCallback(_convert)
 155.         return d
 156. 
 157.     def data_rate_ciphertext_fetch(self, ctx, data):
 158.         d = self.upload_results()
 159.         def _convert(r):
 160.             fetch_size = r.ciphertext_fetched
 161.             time = r.timings.get("cumulative_fetch")
 162.             if (fetch_size is None or time is None):
 163.                 return None
 164.             try:
 165.                 return 1.0 * fetch_size / time
 166.             except ZeroDivisionError:
 167.                 return None
 168.         d.addCallback(_convert)
 169.         return d
 170. 
 171. class UploadStatusPage(UploadResultsRendererMixin, rend.Page):
 172.     docFactory = getxmlfile("upload-status.xhtml")
 173. 
 174.     def __init__(self, data):
 175.         rend.Page.__init__(self, data)
 176.         self.upload_status = data
 177. 
 178.     def upload_results(self):
 179.         return defer.maybeDeferred(self.upload_status.get_results)
 180. 
 181.     def render_results(self, ctx, data):
 182.         d = self.upload_results()
 183.         def _got_results(results):
 184.             if results:
 185.                 return ctx.tag
 186.             return ""
 187.         d.addCallback(_got_results)
 188.         return d
 189. 
 190.     def render_started(self, ctx, data):
 191.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 192.         started_s = time.strftime(TIME_FORMAT,
 193.                                   time.localtime(data.get_started()))
 194.         return started_s
 195. 
 196.     def render_si(self, ctx, data):
 197.         si_s = base32.b2a_or_none(data.get_storage_index())
 198.         if si_s is None:
 199.             si_s = "(None)"
 200.         return si_s
 201. 
 202.     def render_helper(self, ctx, data):
 203.         return {True: "Yes",
 204.                 False: "No"}[data.using_helper()]
 205. 
 206.     def render_total_size(self, ctx, data):
 207.         size = data.get_size()
 208.         if size is None:
 209.             return "(unknown)"
 210.         return size
 211. 
 212.     def render_progress_hash(self, ctx, data):
 213.         progress = data.get_progress()[0]
 214.         # TODO: make an ascii-art bar
 215.         return "%.1f%%" % (100.0 * progress)
 216. 
 217.     def render_progress_ciphertext(self, ctx, data):
 218.         progress = data.get_progress()[1]
 219.         # TODO: make an ascii-art bar
 220.         return "%.1f%%" % (100.0 * progress)
 221. 
 222.     def render_progress_encode_push(self, ctx, data):
 223.         progress = data.get_progress()[2]
 224.         # TODO: make an ascii-art bar
 225.         return "%.1f%%" % (100.0 * progress)
 226. 
 227.     def render_status(self, ctx, data):
 228.         return data.get_status()
 229. 
 230. class DownloadResultsRendererMixin(RateAndTimeMixin):
 231.     # this requires a method named 'download_results'
 232. 
 233.     def render_servermap(self, ctx, data):
 234.         d = self.download_results()
 235.         d.addCallback(lambda res: res.servermap)
 236.         def _render(servermap):
 237.             if servermap is None:
 238.                 return "None"
 239.             l = T.ul()
 240.             for peerid in sorted(servermap.keys()):
 241.                 peerid_s = idlib.shortnodeid_b2a(peerid)
 242.                 shares_s = ",".join(["#%d" % shnum
 243.                                      for shnum in servermap[peerid]])
 244.                 l[T.li["[%s] has share%s: %s" % (peerid_s,
 245.                                                  plural(servermap[peerid]),
 246.                                                  shares_s)]]
 247.             return l
 248.         d.addCallback(_render)
 249.         return d
 250. 
 251.     def render_servers_used(self, ctx, data):
 252.         d = self.download_results()
 253.         d.addCallback(lambda res: res.servers_used)
 254.         def _got(servers_used):
 255.             if not servers_used:
 256.                 return ""
 257.             peerids_s = ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 258.                                    for peerid in servers_used])
 259.             return T.li["Servers Used: ", peerids_s]
 260.         d.addCallback(_got)
 261.         return d
 262. 
 263.     def render_problems(self, ctx, data):
 264.         d = self.download_results()
 265.         d.addCallback(lambda res: res.server_problems)
 266.         def _got(server_problems):
 267.             if not server_problems:
 268.                 return ""
 269.             l = T.ul()
 270.             for peerid in sorted(server_problems.keys()):
 271.                 peerid_s = idlib.shortnodeid_b2a(peerid)
 272.                 l[T.li["[%s]: %s" % (peerid_s, server_problems[peerid])]]
 273.             return T.li["Server Problems:", l]
 274.         d.addCallback(_got)
 275.         return d
 276. 
 277.     def data_file_size(self, ctx, data):
 278.         d = self.download_results()
 279.         d.addCallback(lambda res: res.file_size)
 280.         return d
 281. 
 282.     def _get_time(self, name):
 283.         d = self.download_results()
 284.         d.addCallback(lambda res: res.timings.get(name))
 285.         return d
 286. 
 287.     def data_time_total(self, ctx, data):
 288.         return self._get_time("total")
 289. 
 290.     def data_time_peer_selection(self, ctx, data):
 291.         return self._get_time("peer_selection")
 292. 
 293.     def data_time_uri_extension(self, ctx, data):
 294.         return self._get_time("uri_extension")
 295. 
 296.     def data_time_hashtrees(self, ctx, data):
 297.         return self._get_time("hashtrees")
 298. 
 299.     def data_time_segments(self, ctx, data):
 300.         return self._get_time("segments")
 301. 
 302.     def data_time_cumulative_fetch(self, ctx, data):
 303.         return self._get_time("cumulative_fetch")
 304. 
 305.     def data_time_cumulative_decode(self, ctx, data):
 306.         return self._get_time("cumulative_decode")
 307. 
 308.     def data_time_cumulative_decrypt(self, ctx, data):
 309.         return self._get_time("cumulative_decrypt")
 310. 
 311.     def data_time_paused(self, ctx, data):
 312.         return self._get_time("paused")
 313. 
 314.     def _get_rate(self, name):
 315.         d = self.download_results()
 316.         def _convert(r):
 317.             file_size = r.file_size
 318.             time = r.timings.get(name)
 319.             if time is None:
 320.                 return None
 321.             try:
 322.                 return 1.0 * file_size / time
 323.             except ZeroDivisionError:
 324.                 return None
 325.         d.addCallback(_convert)
 326.         return d
 327. 
 328.     def data_rate_total(self, ctx, data):
 329.         return self._get_rate("total")
 330. 
 331.     def data_rate_segments(self, ctx, data):
 332.         return self._get_rate("segments")
 333. 
 334.     def data_rate_fetch(self, ctx, data):
 335.         return self._get_rate("cumulative_fetch")
 336. 
 337.     def data_rate_decode(self, ctx, data):
 338.         return self._get_rate("cumulative_decode")
 339. 
 340.     def data_rate_decrypt(self, ctx, data):
 341.         return self._get_rate("cumulative_decrypt")
 342. 
 343.     def render_server_timings(self, ctx, data):
 344.         d = self.download_results()
 345.         d.addCallback(lambda res: res.timings.get("fetch_per_server"))
 346.         def _render(per_server):
 347.             if per_server is None:
 348.                 return ""
 349.             l = T.ul()
 350.             for peerid in sorted(per_server.keys()):
 351.                 peerid_s = idlib.shortnodeid_b2a(peerid)
 352.                 times_s = ", ".join([self.render_time(None, t)
 353.                                      for t in per_server[peerid]])
 354.                 l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 355.             return T.li["Per-Server Segment Fetch Response Times: ", l]
 356.         d.addCallback(_render)
 357.         return d
 358. 
 359. class DownloadStatusPage(DownloadResultsRendererMixin, rend.Page):
 360.     docFactory = getxmlfile("download-status.xhtml")
 361. 
 362.     def __init__(self, data):
 363.         rend.Page.__init__(self, data)
 364.         self.download_status = data
 365. 
 366.     def download_results(self):
 367.         return defer.maybeDeferred(self.download_status.get_results)
 368. 
 369.     def render_results(self, ctx, data):
 370.         d = self.download_results()
 371.         def _got_results(results):
 372.             if results:
 373.                 return ctx.tag
 374.             return ""
 375.         d.addCallback(_got_results)
 376.         return d
 377. 
 378.     def render_started(self, ctx, data):
 379.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 380.         started_s = time.strftime(TIME_FORMAT,
 381.                                   time.localtime(data.get_started()))
 382.         return started_s
 383. 
 384.     def render_si(self, ctx, data):
 385.         si_s = base32.b2a_or_none(data.get_storage_index())
 386.         if si_s is None:
 387.             si_s = "(None)"
 388.         return si_s
 389. 
 390.     def render_helper(self, ctx, data):
 391.         return {True: "Yes",
 392.                 False: "No"}[data.using_helper()]
 393. 
 394.     def render_total_size(self, ctx, data):
 395.         size = data.get_size()
 396.         if size is None:
 397.             return "(unknown)"
 398.         return size
 399. 
 400.     def render_progress(self, ctx, data):
 401.         progress = data.get_progress()
 402.         # TODO: make an ascii-art bar
 403.         return "%.1f%%" % (100.0 * progress)
 404. 
 405.     def render_status(self, ctx, data):
 406.         return data.get_status()
 407. 
 408. class RetrieveStatusPage(rend.Page, RateAndTimeMixin):
 409.     docFactory = getxmlfile("retrieve-status.xhtml")
 410. 
 411.     def __init__(self, data):
 412.         rend.Page.__init__(self, data)
 413.         self.retrieve_status = data
 414. 
 415.     def render_started(self, ctx, data):
 416.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 417.         started_s = time.strftime(TIME_FORMAT,
 418.                                   time.localtime(data.get_started()))
 419.         return started_s
 420. 
 421.     def render_si(self, ctx, data):
 422.         si_s = base32.b2a_or_none(data.get_storage_index())
 423.         if si_s is None:
 424.             si_s = "(None)"
 425.         return si_s
 426. 
 427.     def render_helper(self, ctx, data):
 428.         return {True: "Yes",
 429.                 False: "No"}[data.using_helper()]
 430. 
 431.     def render_current_size(self, ctx, data):
 432.         size = data.get_size()
 433.         if size is None:
 434.             size = "(unknown)"
 435.         return size
 436. 
 437.     def render_progress(self, ctx, data):
 438.         progress = data.get_progress()
 439.         # TODO: make an ascii-art bar
 440.         return "%.1f%%" % (100.0 * progress)
 441. 
 442.     def render_status(self, ctx, data):
 443.         return data.get_status()
 444. 
 445.     def render_encoding(self, ctx, data):
 446.         k, n = data.get_encoding()
 447.         return ctx.tag["Encoding: %s of %s" % (k, n)]
 448. 
 449.     def render_problems(self, ctx, data):
 450.         problems = data.problems
 451.         if not problems:
 452.             return ""
 453.         l = T.ul()
 454.         for peerid in sorted(problems.keys()):
 455.             peerid_s = idlib.shortnodeid_b2a(peerid)
 456.             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 457.         return ctx.tag["Server Problems:", l]
 458. 
 459.     def _get_rate(self, data, name):
 460.         file_size = self.retrieve_status.get_size()
 461.         time = self.retrieve_status.timings.get(name)
 462.         if time is None or file_size is None:
 463.             return None
 464.         try:
 465.             return 1.0 * file_size / time
 466.         except ZeroDivisionError:
 467.             return None
 468. 
 469.     def data_time_total(self, ctx, data):
 470.         return self.retrieve_status.timings.get("total")
 471.     def data_rate_total(self, ctx, data):
 472.         return self._get_rate(data, "total")
 473. 
 474.     def data_time_fetch(self, ctx, data):
 475.         return self.retrieve_status.timings.get("fetch")
 476.     def data_rate_fetch(self, ctx, data):
 477.         return self._get_rate(data, "fetch")
 478. 
 479.     def data_time_decode(self, ctx, data):
 480.         return self.retrieve_status.timings.get("decode")
 481.     def data_rate_decode(self, ctx, data):
 482.         return self._get_rate(data, "decode")
 483. 
 484.     def data_time_decrypt(self, ctx, data):
 485.         return self.retrieve_status.timings.get("decrypt")
 486.     def data_rate_decrypt(self, ctx, data):
 487.         return self._get_rate(data, "decrypt")
 488. 
 489.     def render_server_timings(self, ctx, data):
 490.         per_server = self.retrieve_status.timings.get("fetch_per_server")
 491.         if not per_server:
 492.             return ""
 493.         l = T.ul()
 494.         for peerid in sorted(per_server.keys()):
 495.             peerid_s = idlib.shortnodeid_b2a(peerid)
 496.             times_s = ", ".join([self.render_time(None, t)
 497.                                  for t in per_server[peerid]])
 498.             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 499.         return T.li["Per-Server Fetch Response Times: ", l]
 500. 
 501. 
 502. class PublishStatusPage(rend.Page, RateAndTimeMixin):
 503.     docFactory = getxmlfile("publish-status.xhtml")
 504. 
 505.     def __init__(self, data):
 506.         rend.Page.__init__(self, data)
 507.         self.publish_status = data
 508. 
 509.     def render_started(self, ctx, data):
 510.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 511.         started_s = time.strftime(TIME_FORMAT,
 512.                                   time.localtime(data.get_started()))
 513.         return started_s
 514. 
 515.     def render_si(self, ctx, data):
 516.         si_s = base32.b2a_or_none(data.get_storage_index())
 517.         if si_s is None:
 518.             si_s = "(None)"
 519.         return si_s
 520. 
 521.     def render_helper(self, ctx, data):
 522.         return {True: "Yes",
 523.                 False: "No"}[data.using_helper()]
 524. 
 525.     def render_current_size(self, ctx, data):
 526.         size = data.get_size()
 527.         if size is None:
 528.             size = "(unknown)"
 529.         return size
 530. 
 531.     def render_progress(self, ctx, data):
 532.         progress = data.get_progress()
 533.         # TODO: make an ascii-art bar
 534.         return "%.1f%%" % (100.0 * progress)
 535. 
 536.     def render_status(self, ctx, data):
 537.         return data.get_status()
 538. 
 539.     def render_encoding(self, ctx, data):
 540.         k, n = data.get_encoding()
 541.         return ctx.tag["Encoding: %s of %s" % (k, n)]
 542. 
 543.     def render_sharemap(self, ctx, data):
 544.         servermap = data.get_servermap()
 545.         if servermap is None:
 546.             return ctx.tag["None"]
 547.         l = T.ul()
 548.         sharemap = servermap.make_sharemap()
 549.         for shnum in sorted(sharemap.keys()):
 550.             l[T.li["%d -> Placed on " % shnum,
 551.                    ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
 552.                               for peerid in sharemap[shnum]])]]
 553.         return ctx.tag["Sharemap:", l]
 554. 
 555.     def render_problems(self, ctx, data):
 556.         problems = data.problems
 557.         if not problems:
 558.             return ""
 559.         l = T.ul()
 560.         for peerid in sorted(problems.keys()):
 561.             peerid_s = idlib.shortnodeid_b2a(peerid)
 562.             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 563.         return ctx.tag["Server Problems:", l]
 564. 
 565.     def _get_rate(self, data, name):
 566.         file_size = self.publish_status.get_size()
 567.         time = self.publish_status.timings.get(name)
 568.         if time is None:
 569.             return None
 570.         try:
 571.             return 1.0 * file_size / time
 572.         except ZeroDivisionError:
 573.             return None
 574. 
 575.     def data_time_total(self, ctx, data):
 576.         return self.publish_status.timings.get("total")
 577.     def data_rate_total(self, ctx, data):
 578.         return self._get_rate(data, "total")
 579. 
 580.     def data_time_setup(self, ctx, data):
 581.         return self.publish_status.timings.get("setup")
 582. 
 583.     def data_time_encrypt(self, ctx, data):
 584.         return self.publish_status.timings.get("encrypt")
 585.     def data_rate_encrypt(self, ctx, data):
 586.         return self._get_rate(data, "encrypt")
 587. 
 588.     def data_time_encode(self, ctx, data):
 589.         return self.publish_status.timings.get("encode")
 590.     def data_rate_encode(self, ctx, data):
 591.         return self._get_rate(data, "encode")
 592. 
 593.     def data_time_pack(self, ctx, data):
 594.         return self.publish_status.timings.get("pack")
 595.     def data_rate_pack(self, ctx, data):
 596.         return self._get_rate(data, "pack")
 597.     def data_time_sign(self, ctx, data):
 598.         return self.publish_status.timings.get("sign")
 599. 
 600.     def data_time_push(self, ctx, data):
 601.         return self.publish_status.timings.get("push")
 602.     def data_rate_push(self, ctx, data):
 603.         return self._get_rate(data, "push")
 604. 
 605.     def render_server_timings(self, ctx, data):
 606.         per_server = self.publish_status.timings.get("send_per_server")
 607.         if not per_server:
 608.             return ""
 609.         l = T.ul()
 610.         for peerid in sorted(per_server.keys()):
 611.             peerid_s = idlib.shortnodeid_b2a(peerid)
 612.             times_s = ", ".join([self.render_time(None, t)
 613.                                  for t in per_server[peerid]])
 614.             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 615.         return T.li["Per-Server Response Times: ", l]
 616. 
 617. class MapupdateStatusPage(rend.Page, RateAndTimeMixin):
 618.     docFactory = getxmlfile("map-update-status.xhtml")
 619. 
 620.     def __init__(self, data):
 621.         rend.Page.__init__(self, data)
 622.         self.update_status = data
 623. 
 624.     def render_started(self, ctx, data):
 625.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 626.         started_s = time.strftime(TIME_FORMAT,
 627.                                   time.localtime(data.get_started()))
 628.         return started_s
 629. 
 630.     def render_finished(self, ctx, data):
 631.         when = data.get_finished()
 632.         if not when:
 633.             return "not yet"
 634.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 635.         started_s = time.strftime(TIME_FORMAT,
 636.                                   time.localtime(data.get_finished()))
 637.         return started_s
 638. 
 639.     def render_si(self, ctx, data):
 640.         si_s = base32.b2a_or_none(data.get_storage_index())
 641.         if si_s is None:
 642.             si_s = "(None)"
 643.         return si_s
 644. 
 645.     def render_helper(self, ctx, data):
 646.         return {True: "Yes",
 647.                 False: "No"}[data.using_helper()]
 648. 
 649.     def render_progress(self, ctx, data):
 650.         progress = data.get_progress()
 651.         # TODO: make an ascii-art bar
 652.         return "%.1f%%" % (100.0 * progress)
 653. 
 654.     def render_status(self, ctx, data):
 655.         return data.get_status()
 656. 
 657.     def render_problems(self, ctx, data):
 658.         problems = data.problems
 659.         if not problems:
 660.             return ""
 661.         l = T.ul()
 662.         for peerid in sorted(problems.keys()):
 663.             peerid_s = idlib.shortnodeid_b2a(peerid)
 664.             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
 665.         return ctx.tag["Server Problems:", l]
 666. 
 667.     def render_privkey_from(self, ctx, data):
 668.         peerid = data.get_privkey_from()
 669.         if peerid:
 670.             return ctx.tag["Got privkey from: [%s]"
 671.                            % idlib.shortnodeid_b2a(peerid)]
 672.         else:
 673.             return ""
 674. 
 675.     def data_time_total(self, ctx, data):
 676.         return self.update_status.timings.get("total")
 677. 
 678.     def data_time_initial_queries(self, ctx, data):
 679.         return self.update_status.timings.get("initial_queries")
 680. 
 681.     def data_time_cumulative_verify(self, ctx, data):
 682.         return self.update_status.timings.get("cumulative_verify")
 683. 
 684.     def render_server_timings(self, ctx, data):
 685.         per_server = self.update_status.timings.get("per_server")
 686.         if not per_server:
 687.             return ""
 688.         l = T.ul()
 689.         for peerid in sorted(per_server.keys()):
 690.             peerid_s = idlib.shortnodeid_b2a(peerid)
 691.             times = []
 692.             for op,started,t in per_server[peerid]:
 693.                 #times.append("%s/%.4fs/%s/%s" % (op,
 694.                 #                              started,
 695.                 #                              self.render_time(None, started - self.update_status.get_started()),
 696.                 #                              self.render_time(None,t)))
 697.                 if op == "query":
 698.                     times.append( self.render_time(None, t) )
 699.                 elif op == "late":
 700.                     times.append( "late(" + self.render_time(None, t) + ")" )
 701.                 else:
 702.                     times.append( "privkey(" + self.render_time(None, t) + ")" )
 703.             times_s = ", ".join(times)
 704.             l[T.li["[%s]: %s" % (peerid_s, times_s)]]
 705.         return T.li["Per-Server Response Times: ", l]
 706. 
 707.     def render_timing_chart(self, ctx, data):
 708.         imageurl = self._timing_chart()
 709.         return ctx.tag[imageurl]
 710. 
 711.     def _timing_chart(self):
 712.         started = self.update_status.get_started()
 713.         total = self.update_status.timings.get("total")
 714.         per_server = self.update_status.timings.get("per_server")
 715.         base = "http://chart.apis.google.com/chart?"
 716.         pieces = ["cht=bhs", "chs=400x300"]
 717.         pieces.append("chco=ffffff,4d89f9,c6d9fd") # colors
 718.         data0 = []
 719.         data1 = []
 720.         data2 = []
 721.         peerids_s = []
 722.         top_abs = started
 723.         # we sort the queries by the time at which we sent the first request
 724.         sorttable = [ (times[0][1], peerid)
 725.                       for peerid, times in per_server.items() ]
 726.         sorttable.sort()
 727.         peerids = [t[1] for t in sorttable]
 728. 
 729.         for peerid in peerids:
 730.             times = per_server[peerid]
 731.             peerid_s = idlib.shortnodeid_b2a(peerid)
 732.             peerids_s.append(peerid_s)
 733.             # for servermap updates, there are either one or two queries per
 734.             # peer. The second (if present) is to get the privkey.
 735.             op,q_started,q_elapsed = times[0]
 736.             data0.append("%.3f" % (q_started-started))
 737.             data1.append("%.3f" % q_elapsed)
 738.             top_abs = max(top_abs, q_started+q_elapsed)
 739.             if len(times) > 1:
 740.                 op,p_started,p_elapsed = times[0]
 741.                 data2.append("%.3f" % p_elapsed)
 742.                 top_abs = max(top_abs, p_started+p_elapsed)
 743.             else:
 744.                 data2.append("0.0")
 745.         finished = self.update_status.get_finished()
 746.         if finished:
 747.             top_abs = max(top_abs, finished)
 748.         top_rel = top_abs - started
 749.         chd = "chd=t:" + "|".join([",".join(data0),
 750.                                    ",".join(data1),
 751.                                    ",".join(data2)])
 752.         pieces.append(chd)
 753.         chds = "chds=0,%0.3f" % top_rel
 754.         pieces.append(chds)
 755.         pieces.append("chxt=x,y")
 756.         pieces.append("chxr=0,0.0,%0.3f" % top_rel)
 757.         pieces.append("chxl=1:|" + "|".join(reversed(peerids_s)))
 758.         # use up to 10 grid lines, at decimal multiples.
 759.         # mathutil.next_power_of_k doesn't handle numbers smaller than one,
 760.         # unfortunately.
 761.         #pieces.append("chg="
 762. 
 763.         if total is not None:
 764.             finished_f = 1.0 * total / top_rel
 765.             pieces.append("chm=r,FF0000,0,%0.3f,%0.3f" % (finished_f,
 766.                                                           finished_f+0.01))
 767.         url = base + "&".join(pieces)
 768.         return T.img(src=url, align="right", float="right")
 769. 
 770. 
 771. class Status(rend.Page):
 772.     docFactory = getxmlfile("status.xhtml")
 773.     addSlash = True
 774. 
 775.     def renderHTTP(self, ctx):
 776.         t = get_arg(inevow.IRequest(ctx), "t")
 777.         if t == "json":
 778.             return self.json(ctx)
 779.         return rend.Page.renderHTTP(self, ctx)
 780. 
 781.     def json(self, ctx):
 782.         inevow.IRequest(ctx).setHeader("content-type", "text/plain")
 783.         client = IClient(ctx)
 784.         data = {}
 785.         data["active"] = active = []
 786.         for s in self.data_active_operations(ctx, None):
 787.             si_s = base32.b2a_or_none(s.get_storage_index())
 788.             size = s.get_size()
 789.             status = s.get_status()
 790.             if IUploadStatus.providedBy(s):
 791.                 h,c,e = s.get_progress()
 792.                 active.append({"type": "upload",
 793.                                "storage-index-string": si_s,
 794.                                "total-size": size,
 795.                                "status": status,
 796.                                "progress-hash": h,
 797.                                "progress-ciphertext": c,
 798.                                "progress-encode-push": e,
 799.                                })
 800.             elif IDownloadStatus.providedBy(s):
 801.                 active.append({"type": "download",
 802.                                "storage-index-string": si_s,
 803.                                "total-size": size,
 804.                                "status": status,
 805.                                "progress": s.get_progress(),
 806.                                })
 807. 
 808.         return simplejson.dumps(data, indent=1) + "\n"
 809. 
 810.     def _get_all_statuses(self, client):
 811.         return itertools.chain(client.list_all_upload_statuses(),
 812.                                client.list_all_download_statuses(),
 813.                                client.list_all_mapupdate_statuses(),
 814.                                client.list_all_publish_statuses(),
 815.                                client.list_all_retrieve_statuses(),
 816.                                client.list_all_helper_statuses(),
 817.                                )
 818. 
 819.     def data_active_operations(self, ctx, data):
 820.         client = IClient(ctx)
 821.         active = [s
 822.                   for s in self._get_all_statuses(client)
 823.                   if s.get_active()]
 824.         return active
 825. 
 826.     def data_recent_operations(self, ctx, data):
 827.         client = IClient(ctx)
 828.         recent = [s
 829.                   for s in self._get_all_statuses(client)
 830.                   if not s.get_active()]
 831.         recent.sort(lambda a,b: cmp(a.get_started(), b.get_started()))
 832.         recent.reverse()
 833.         return recent
 834. 
 835.     def render_row(self, ctx, data):
 836.         s = data
 837. 
 838.         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
 839.         started_s = time.strftime(TIME_FORMAT,
 840.                                   time.localtime(s.get_started()))
 841.         ctx.fillSlots("started", started_s)
 842. 
 843.         si_s = base32.b2a_or_none(s.get_storage_index())
 844.         if si_s is None:
 845.             si_s = "(None)"
 846.         ctx.fillSlots("si", si_s)
 847.         ctx.fillSlots("helper", {True: "Yes",
 848.                                  False: "No"}[s.using_helper()])
 849. 
 850.         size = s.get_size()
 851.         if size is None:
 852.             size = "(unknown)"
 853.         elif isinstance(size, (int, long, float)):
 854.             size = abbreviate_size(size)
 855.         ctx.fillSlots("total_size", size)
 856. 
 857.         progress = data.get_progress()
 858.         if IUploadStatus.providedBy(data):
 859.             link = "up-%d" % data.get_counter()
 860.             ctx.fillSlots("type", "upload")
 861.             # TODO: make an ascii-art bar
 862.             (chk, ciphertext, encandpush) = progress
 863.             progress_s = ("hash: %.1f%%, ciphertext: %.1f%%, encode: %.1f%%" %
 864.                           ( (100.0 * chk),
 865.                             (100.0 * ciphertext),
 866.                             (100.0 * encandpush) ))
 867.             ctx.fillSlots("progress", progress_s)
 868.         elif IDownloadStatus.providedBy(data):
 869.             link = "down-%d" % data.get_counter()
 870.             ctx.fillSlots("type", "download")
 871.             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
 872.         elif IPublishStatus.providedBy(data):
 873.             link = "publish-%d" % data.get_counter()
 874.             ctx.fillSlots("type", "publish")
 875.             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
 876.         elif IRetrieveStatus.providedBy(data):
 877.             ctx.fillSlots("type", "retrieve")
 878.             link = "retrieve-%d" % data.get_counter()
 879.             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
 880.         else:
 881.             assert IServermapUpdaterStatus.providedBy(data)
 882.             ctx.fillSlots("type", "mapupdate %s" % data.get_mode())
 883.             link = "mapupdate-%d" % data.get_counter()
 884.             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
 885.         ctx.fillSlots("status", T.a(href=link)[s.get_status()])
 886.         return ctx.tag
 887. 
 888.     def childFactory(self, ctx, name):
 889.         client = IClient(ctx)
 890.         stype,count_s = name.split("-")
 891.         count = int(count_s)
 892.         if stype == "up":
 893.             for s in itertools.chain(client.list_all_upload_statuses(),
 894.                                      client.list_all_helper_statuses()):
 895.                 # immutable-upload helpers use the same status object as a
 896.                 # regular immutable-upload
 897.                 if s.get_counter() == count:
 898.                     return UploadStatusPage(s)
 899.         if stype == "down":
 900.             for s in client.list_all_download_statuses():
 901.                 if s.get_counter() == count:
 902.                     return DownloadStatusPage(s)
 903.         if stype == "mapupdate":
 904.             for s in client.list_all_mapupdate_statuses():
 905.                 if s.get_counter() == count:
 906.                     return MapupdateStatusPage(s)
 907.         if stype == "publish":
 908.             for s in client.list_all_publish_statuses():
 909.                 if s.get_counter() == count:
 910.                     return PublishStatusPage(s)
 911.         if stype == "retrieve":
 912.             for s in client.list_all_retrieve_statuses():
 913.                 if s.get_counter() == count:
 914.                     return RetrieveStatusPage(s)
 915. 
 916. 
 917. class HelperStatus(rend.Page):
 918.     docFactory = getxmlfile("helper.xhtml")
 919. 
 920.     def renderHTTP(self, ctx):
 921.         t = get_arg(inevow.IRequest(ctx), "t")
 922.         if t == "json":
 923.             return self.render_JSON(ctx)
 924.         # is there a better way to provide 'data' to all rendering methods?
 925.         helper = IClient(ctx).getServiceNamed("helper")
 926.         self.original = helper.get_stats()
 927.         return rend.Page.renderHTTP(self, ctx)
 928. 
 929.     def render_JSON(self, ctx):
 930.         inevow.IRequest(ctx).setHeader("content-type", "text/plain")
 931.         try:
 932.             h = IClient(ctx).getServiceNamed("helper")
 933.         except KeyError:
 934.             return simplejson.dumps({}) + "\n"
 935. 
 936.         stats = h.get_stats()
 937.         return simplejson.dumps(stats, indent=1) + "\n"
 938. 
 939.     def render_active_uploads(self, ctx, data):
 940.         return data["chk_upload_helper.active_uploads"]
 941. 
 942.     def render_incoming(self, ctx, data):
 943.         return "%d bytes in %d files" % (data["chk_upload_helper.incoming_size"],
 944.                                          data["chk_upload_helper.incoming_count"])
 945. 
 946.     def render_encoding(self, ctx, data):
 947.         return "%d bytes in %d files" % (data["chk_upload_helper.encoding_size"],
 948.                                          data["chk_upload_helper.encoding_count"])
 949. 
 950.     def render_upload_requests(self, ctx, data):
 951.         return str(data["chk_upload_helper.upload_requests"])
 952. 
 953.     def render_upload_already_present(self, ctx, data):
 954.         return str(data["chk_upload_helper.upload_already_present"])
 955. 
 956.     def render_upload_need_upload(self, ctx, data):
 957.         return str(data["chk_upload_helper.upload_need_upload"])
 958. 
 959.     def render_upload_bytes_fetched(self, ctx, data):
 960.         return str(data["chk_upload_helper.fetched_bytes"])
 961. 
 962.     def render_upload_bytes_encoded(self, ctx, data):
 963.         return str(data["chk_upload_helper.encoded_bytes"])
 964. 
 965. 
 966. class Statistics(rend.Page):
 967.     docFactory = getxmlfile("statistics.xhtml")
 968. 
 969.     def renderHTTP(self, ctx):
 970.         provider = IClient(ctx).stats_provider
 971.         stats = {'stats': {}, 'counters': {}}
 972.         if provider:
 973.             stats = provider.get_stats()
 974.         t = get_arg(inevow.IRequest(ctx), "t")
 975.         if t == "json":
 976.             inevow.IRequest(ctx).setHeader("content-type", "text/plain")
 977.             return simplejson.dumps(stats, indent=1) + "\n"
 978.         # is there a better way to provide 'data' to all rendering methods?
 979.         self.original = stats
 980.         return rend.Page.renderHTTP(self, ctx)
 981. 
 982.     def render_load_average(self, ctx, data):
 983.         return str(data["stats"].get("load_monitor.avg_load"))
 984. 
 985.     def render_peak_load(self, ctx, data):
 986.         return str(data["stats"].get("load_monitor.max_load"))
 987. 
 988.     def render_uploads(self, ctx, data):
 989.         files = data["counters"].get("uploader.files_uploaded", 0)
 990.         bytes = data["counters"].get("uploader.bytes_uploaded", 0)
 991.         return ("%s files / %s bytes (%s)" %
 992.                 (files, bytes, abbreviate_size(bytes)))
 993. 
 994.     def render_downloads(self, ctx, data):
 995.         files = data["counters"].get("downloader.files_downloaded", 0)
 996.         bytes = data["counters"].get("downloader.bytes_downloaded", 0)
 997.         return ("%s files / %s bytes (%s)" %
 998.                 (files, bytes, abbreviate_size(bytes)))
 999. 
1000.     def render_publishes(self, ctx, data):
1001.         files = data["counters"].get("mutable.files_published", 0)
1002.         bytes = data["counters"].get("mutable.bytes_published", 0)
1003.         return "%s files / %s bytes (%s)" % (files, bytes,
1004.                                              abbreviate_size(bytes))
1005. 
1006.     def render_retrieves(self, ctx, data):
1007.         files = data["counters"].get("mutable.files_retrieved", 0)
1008.         bytes = data["counters"].get("mutable.bytes_retrieved", 0)
1009.         return "%s files / %s bytes (%s)" % (files, bytes,
1010.                                              abbreviate_size(bytes))
1011. 
1012.     def render_raw(self, ctx, data):
1013.         raw = pprint.pformat(data)
1014.         return ctx.tag[raw]