Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-30 03:14:22 +00:00
Code Issues Projects Releases Wiki Activity

minor improvements

Browse Source
This commit is contained in:
Maximilian Hils 2016-05-19 22:05:50 -07:00
parent f3bedc103a
commit 395b11d288
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mitmproxy/flow.py
View File

@ -13,6 +13,8 @@ from six.moves import http_cookies, http_cookiejar, urllib
import os import os
import re import re
from typing import List, Optional, Set
from netlib import wsgi, odict from netlib import wsgi, odict
from netlib.exceptions import HttpException from netlib.exceptions import HttpException
from netlib.http import Headers, http1, cookies from netlib.http import Headers, http1, cookies
@ -378,8 +380,11 @@ class StickyAuthState:
f.request.headers["authorization"] = self.hosts[host] f.request.headers["authorization"] = self.hosts[host]
@six.add_metaclass(ABCMeta)
class FlowList(object): class FlowList(object):
__metaclass__ = ABCMeta
def __init__(self):
self._list = [] # type: List[Flow]
def __iter__(self): def __iter__(self):
return iter(self._list) return iter(self._list)
@ -418,7 +423,7 @@ class FlowList(object):
class FlowView(FlowList): class FlowView(FlowList):
def __init__(self, store, filt=None): def __init__(self, store, filt=None):
self._list = [] super(FlowView, self).__init__()
if not filt: if not filt:
filt = lambda flow: True filt = lambda flow: True
self._build(store, filt) self._build(store, filt)
@ -460,7 +465,7 @@ class FlowStore(FlowList):
""" """
def __init__(self): def __init__(self):
self._list = [] super(FlowStore, self).__init__()
self._set = set() # Used for O(1) lookups self._set = set() # Used for O(1) lookups
self.views = [] self.views = []
self._recalculate_views() self._recalculate_views()
@ -651,18 +656,18 @@ class FlowMaster(controller.ServerMaster):
self.server_playback = None self.server_playback = None
self.client_playback = None self.client_playback = None
self.kill_nonreplay = False self.kill_nonreplay = False
self.scripts = [] self.scripts = [] # type: List[script.Script]
self.pause_scripts = False self.pause_scripts = False
self.stickycookie_state = False self.stickycookie_state = None # type: Optional[StickyCookieState]
self.stickycookie_txt = None self.stickycookie_txt = None
self.stickyauth_state = False self.stickyauth_state = False # type: Optional[StickyAuthState]
self.stickyauth_txt = None self.stickyauth_txt = None
self.anticache = False self.anticache = False
self.anticomp = False self.anticomp = False
self.stream_large_bodies = False self.stream_large_bodies = None # type: Optional[StreamLargeBodies]
self.refresh_server_playback = False self.refresh_server_playback = False
self.replacehooks = ReplaceHooks() self.replacehooks = ReplaceHooks()
self.setheaders = SetHeaders() self.setheaders = SetHeaders()