21 lines
778 B
Python
21 lines
778 B
Python
import regex as re
|
|
|
|
from watfag.parsers.generic import GenericParser
|
|
from watfag.parsers.generic.parsers import CheckParser
|
|
from watfag.parsers.generic.watfag import VideoCodec
|
|
|
|
checks = {
|
|
re.compile(r"avc|[hx][\. -]?264", re.IGNORECASE): VideoCodec.AVC,
|
|
re.compile(r"hevc|[hx][\. -]?265", re.IGNORECASE): VideoCodec.HEVC,
|
|
re.compile(r"vp9", re.IGNORECASE): VideoCodec.VP9,
|
|
re.compile(r"av1", re.IGNORECASE): VideoCodec.AV1,
|
|
re.compile(r"mpeg-4|mpeg4|mp4v|xvid", re.IGNORECASE): VideoCodec.XVID
|
|
}
|
|
|
|
class VideoCodecParser(CheckParser, GenericParser):
|
|
def __init__(self, release):
|
|
super().__init__(release)
|
|
self.checks = checks
|
|
self.remove_checks = [] # No remove checks for video codec
|
|
self.attribute_name = "video_codec"
|