exracted parser mappings
This commit is contained in:
29
src/parsers/parser_mapping.py
Normal file
29
src/parsers/parser_mapping.py
Normal file
@ -0,0 +1,29 @@
|
||||
from collections import OrderedDict
|
||||
import re
|
||||
|
||||
from src.parsers.MyMail.my_mail_parser import MyMailParser
|
||||
from src.parsers.Okru.ok_parser import OkParser
|
||||
from src.parsers.Yappy.yappy_parser import YappyParser
|
||||
from src.parsers.base_parser import BaseParser
|
||||
|
||||
def compile_regex(regex):
|
||||
return re.compile(regex, re.IGNORECASE | re.DOTALL | re.MULTILINE)
|
||||
|
||||
parser_mapping = OrderedDict(
|
||||
{
|
||||
compile_regex(r"^my.mail.ru/") : MyMailParser,
|
||||
compile_regex(r"^(?:www.)?(?:youtube.com|youtu.be)/"): BaseParser,
|
||||
compile_regex(r"^vk.com/"): BaseParser,
|
||||
compile_regex(r"^ok.ru/okvideo/topic"): OkParser,
|
||||
compile_regex(r"^ok.ru/video"): BaseParser,
|
||||
compile_regex(r"^...?likee.video/"): BaseParser,
|
||||
compile_regex(r"^dzen.ru/"): BaseParser,
|
||||
compile_regex(r"^yappy.media/"): BaseParser,
|
||||
compile_regex(r"^yandex.ru/"): BaseParser,
|
||||
}
|
||||
)
|
||||
|
||||
def get_parser(uri):
|
||||
for regex in parser_mapping:
|
||||
if regex.match(uri):
|
||||
return parser_mapping[regex]
|
Reference in New Issue
Block a user