Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Implement utils.irc.hostmask_match() as regex

Browse source
This commit is contained in:
jesopo 2019-06-20 14:10:10 +01:00
parent a9514dfc44
commit 792a42be6e
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/utils/irc/__init__.py
View file

@ -1,4 +1,4 @@
import fnmatch, json, string, re, typing, uuid import json, string, re, typing, uuid
from src import IRCLine, utils from src import IRCLine, utils
from . import protocol from . import protocol
@ -331,4 +331,11 @@ class BatchType(object):
return t[0] if t else None return t[0] if t else None
def hostmask_match(hostmask: str, pattern: str) -> bool: def hostmask_match(hostmask: str, pattern: str) -> bool:
return fnmatch.fnmatchcase(hostmask, pattern) part1_out = []
for part1 in pattern.split("?"):
part2_out = []
for part2 in part1.split("*"):
part2_out.append(re.escape(part2))
part1_out.append(".*".join(part2_out))
pattern_parsed = ".".join(part1_out)
return not re.match(pattern_parsed, hostmask) == None