Merge pull request #935 from mitmproxy/coding-style

Improve Coding Style
This commit is contained in:
Maximilian Hils 2016-02-14 15:10:09 +01:00
commit ea0ba6d143
13 changed files with 6 additions and 36 deletions

View File

@ -13,7 +13,6 @@ from .. import version
from .flow import Flow
class MessageMixin(object):
def get_decoded_content(self):

View File

@ -242,7 +242,6 @@ class HttpLayer(Layer):
self.send_response(response)
except (NetlibException, H2Error):
self.log(traceback.format_exc(), "debug")
pass
def change_upstream_proxy_server(self, address):
# Make set_upstream_proxy_server always available,

View File

@ -1,12 +1,9 @@
from __future__ import (absolute_import, print_function, division)
import six
from netlib import tcp
from netlib.http import http1
from .http import _HttpTransmissionLayer, HttpLayer
from .. import utils
from ..models import HTTPRequest, HTTPResponse

View File

@ -18,8 +18,6 @@ from .base import Layer
from .http import _HttpTransmissionLayer, HttpLayer
from .. import utils
from ..models import HTTPRequest, HTTPResponse
from ..exceptions import HttpProtocolException
from ..exceptions import ProtocolException
class SafeH2Connection(H2Connection):
@ -234,7 +232,6 @@ class Http2Layer(Layer):
stream.zombie = time.time()
return
frame, _ = hyperframe.frame.Frame.parse_frame_header(raw_frame[:9])
if is_server:

View File

@ -419,7 +419,7 @@ class TlsLayer(Layer):
try:
self.ctx.connect()
self._establish_tls_with_server()
except Exception as e:
except Exception:
try:
self._establish_tls_with_client()
except:

View File

@ -3,6 +3,7 @@ from netlib.utils import Serializable
class StateObject(Serializable):
"""
An object with serializable state.

View File

@ -42,6 +42,7 @@ class APIError(tornado.web.HTTPError):
class BasicAuth(object):
def set_auth_headers(self):
self.set_status(401)
self.set_header('WWW-Authenticate', 'Basic realm=MITMWeb')

View File

@ -6,21 +6,6 @@ if os.name == "nt":
import libmproxy.console.help as help
class DummyLoop:
def __init__(self):
self.widget = None
class DummyMaster:
def __init__(self):
self.loop = DummyLoop()
def make_view(self):
pass
class TestHelp:
def test_helptext(self):
@ -28,7 +13,6 @@ class TestHelp:
assert h.helptext()
def test_keypress(self):
master = DummyMaster()
h = help.HelpView([1, 2, 3])
assert not h.keypress((0, 0), "q")
assert not h.keypress((0, 0), "?")

View File

@ -209,7 +209,6 @@ class TestMatching:
def test_method(self):
q = self.req()
s = self.resp()
assert self.q("~m get", q)
assert not self.q("~m post", q)
@ -218,7 +217,6 @@ class TestMatching:
def test_domain(self):
q = self.req()
s = self.resp()
assert self.q("~d host", q)
assert not self.q("~d none", q)

View File

@ -350,7 +350,7 @@ class TestFlow(object):
def test_copy(self):
f = tutils.tflow(resp=True)
a0 = f.get_state()
f.get_state()
f2 = f.copy()
a = f.get_state()
b = f2.get_state()
@ -532,7 +532,6 @@ class TestState:
assert c.flow_count() == 2
assert c.active_flow_count() == 1
_ = HTTPResponse.wrap(netlib.tutils.tresp())
assert not c.update_flow(None)
assert c.active_flow_count() == 1

View File

@ -1,7 +1,6 @@
from netlib.exceptions import HttpSyntaxException
from netlib.http import http1
from netlib.tcp import TCPClient
from netlib.tutils import treq, raises
from netlib.tutils import treq
from . import tutils, tservers

View File

@ -5,10 +5,8 @@ import pytest
import traceback
import os
import tempfile
import sys
from libmproxy.proxy.config import ProxyConfig
from libmproxy.proxy.server import ProxyServer
from libmproxy.cmdline import APP_HOST, APP_PORT
import logging
@ -24,9 +22,7 @@ from netlib import tservers as netlib_tservers
from netlib.utils import http2_read_raw_frame
import h2
from hyperframe.frame import Frame
from libmproxy import utils
from . import tservers
requires_alpn = pytest.mark.skipif(