summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorThomas Klausner <tk@giga.or.at>2016-01-02 20:23:54 +0100
committerThomas Klausner <tk@giga.or.at>2016-01-02 20:23:54 +0100
commitd1daa16fc91d6a9f9847e5822e086f1053a34537 (patch)
treee7d0b83826300376d5596e8c661c211ca4a7cf66
parent2a1d7f889e41ec2cf4f76db01b3871eef95268e5 (diff)
downloadpsutil-d1daa16fc91d6a9f9847e5822e086f1053a34537.tar.gz
Check arguments first before using it.
-rw-r--r--psutil/_psbsd.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/psutil/_psbsd.py b/psutil/_psbsd.py
index 89f29c76..3449473e 100644
--- a/psutil/_psbsd.py
+++ b/psutil/_psbsd.py
@@ -436,6 +436,10 @@ class Process(object):
@wrap_exceptions
def connections(self, kind='inet'):
+ if kind not in conn_tmap:
+ raise ValueError("invalid %r kind argument; choose between %s"
+ % (kind, ', '.join([repr(x) for x in conn_tmap])))
+
if NETBSD:
families, types = conn_tmap[kind]
ret = set()
@@ -453,9 +457,6 @@ class Process(object):
ret.add(nt)
return list(ret)
- if kind not in conn_tmap:
- raise ValueError("invalid %r kind argument; choose between %s"
- % (kind, ', '.join([repr(x) for x in conn_tmap])))
families, types = conn_tmap[kind]
rawlist = cext.proc_connections(self.pid, families, types)
ret = []