22 lines
645 B
Python
22 lines
645 B
Python
|
from mitmproxy import flowfilter
|
||
|
|
|
||
|
|
|
||
|
|
def parse_spec(option: str) -> tuple[flowfilter.TFilter, str, str]:
|
||
|
|
"""
|
||
|
|
Parse strings in the following format:
|
||
|
|
|
||
|
|
[/flow-filter]/subject/replacement
|
||
|
|
|
||
|
|
"""
|
||
|
|
sep, rem = option[0], option[1:]
|
||
|
|
parts = rem.split(sep, 2)
|
||
|
|
if len(parts) == 2:
|
||
|
|
subject, replacement = parts
|
||
|
|
return flowfilter.match_all, subject, replacement
|
||
|
|
elif len(parts) == 3:
|
||
|
|
patt, subject, replacement = parts
|
||
|
|
flow_filter = flowfilter.parse(patt)
|
||
|
|
return flow_filter, subject, replacement
|
||
|
|
else:
|
||
|
|
raise ValueError("Invalid number of parameters (2 or 3 are expected)")
|