Browse Source

Simplify some bits of logic

Daniel Martí 9 years ago
parent
commit
78aaf07a8a
2 changed files with 4 additions and 14 deletions
  1. 2 8
      lib/chutney/TorNet.py
  2. 2 6
      lib/chutney/Traffic.py

+ 2 - 8
lib/chutney/TorNet.py

@@ -715,10 +715,7 @@ class TorEnviron(chutney.Templating.Environ):
         return "test%03d%s" % (my['nodenum'], my['tag'])
 
     def _get_tor_gencert(self, my):
-        if my['tor-gencert']:
-            return my['tor-gencert']
-        else:
-            return '{0}-gencert'.format(my['tor'])
+        return my['tor-gencert'] or '{0}-gencert'.format(my['tor'])
 
     def _get_auth_passphrase(self, my):
         return self['nick']  # OMG TEH SECURE!
@@ -832,10 +829,7 @@ class Network(object):
         sys.stdout.write("Verifying data transmission: ")
         sys.stdout.flush()
         status = self._verify_traffic()
-        if status:
-            print("Success")
-        else:
-            print("Failure")
+        print "Success" if status else "Failure"
         return status
 
     def _verify_traffic(self):

+ 2 - 6
lib/chutney/Traffic.py

@@ -152,8 +152,7 @@ class Sink(Peer):
         if len(self.inbuf) == len(data):
             if self.inbuf != data:
                 return -1       # Failed verification.
-            else:
-                debug("successful verification")
+            debug("successful verification")
         return len(data) - len(self.inbuf)
 
 
@@ -178,10 +177,7 @@ class Source(Peer):
     def connect(self, endpoint):
         self.dest = endpoint
         self.state = self.CONNECTING
-        if self.proxy is None:
-            dest = self.dest
-        else:
-            dest = self.proxy
+        dest = self.proxy or self.dest
         try:
             self.s.connect(dest)
         except socket.error, e: