raise attribute error for non-rpc functions in Contact

2018-05-29 10:59:07 -04:00 · 2018-05-29 10:59:07 -04:00 · 0d23c68797
commit 0d23c68797
parent ec1b6b2387
1 changed files with 3 additions and 0 deletions
--- a/lbrynet/dht/contact.py
+++ b/lbrynet/dht/contact.py
@ -139,6 +139,9 @@ class _Contact(object):
        host Node's C{_protocol} object).
        """

+        if name not in ['ping', 'findValue', 'findNode', 'store']:
+            raise AttributeError("unknown command: %s" % name)
+
        def _sendRPC(*args, **kwargs):
            return self._networkProtocol.sendRPC(self, name, args, **kwargs)