main_netrate.py 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2021 The MMGen Project <mmgen@tuta.io>
  5. #
  6. # This program is free software: you can redistribute it and/or modify it under
  7. # the terms of the GNU General Public License as published by the Free Software
  8. # Foundation, either version 3 of the License, or (at your option) any later
  9. # version.
  10. #
  11. # This program is distributed in the hope that it will be useful, but WITHOUT
  12. # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13. # FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  14. # details.
  15. #
  16. # You should have received a copy of the GNU General Public License along with
  17. # this program. If not, see <http://www.gnu.org/licenses/>.
  18. """
  19. mmnode-netrate: Bitcoin daemon network rate monitor
  20. """
  21. import time
  22. from mmgen.common import *
  23. opts_data = {
  24. 'text': {
  25. 'desc': 'Bitcoin daemon network rate monitor',
  26. 'usage': '[opts]',
  27. 'options': """
  28. -h, --help Print this help message
  29. --, --longhelp Print help message for long options (common options)
  30. """
  31. }
  32. }
  33. cmd_args = opts.init(opts_data)
  34. ERASE_LINE,CUR_UP = '\033[K','\033[1A'
  35. async def main():
  36. from mmgen.protocol import init_proto_from_opts
  37. proto = init_proto_from_opts()
  38. from mmgen.rpc import rpc_init
  39. c = await rpc_init(proto)
  40. async def get_data():
  41. d = await c.call('getnettotals')
  42. return [float(e) for e in (d['totalbytesrecv'],d['totalbytessent'],d['timemillis'])]
  43. rs = None
  44. while True:
  45. r,s,t = await get_data()
  46. if rs is not None:
  47. sys.stderr.write(
  48. '\rrcvd: {:9.2f} kB/s\nsent: {:9.2f} kB/s '.format(
  49. (r-rs)/(t-ts),
  50. (s-ss)/(t-ts) ))
  51. time.sleep(2)
  52. if rs is not None:
  53. sys.stderr.write('{}{}{}'.format(ERASE_LINE,CUR_UP,ERASE_LINE))
  54. rs,ss,ts = r,s,t
  55. try:
  56. async_run(main())
  57. except KeyboardInterrupt:
  58. sys.stderr.write('\n')