mirror of https://github.com/morpheus65535/bazarr
106 lines
2.2 KiB
Python
106 lines
2.2 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Quantities: Size
|
|
"""
|
|
from abc import abstractmethod
|
|
|
|
from rebulk.remodule import re
|
|
|
|
from ..common import seps
|
|
|
|
|
|
class Quantity:
|
|
"""
|
|
Represent a quantity object with magnitude and units.
|
|
"""
|
|
|
|
parser_re = re.compile(r'(?P<magnitude>\d+(?:[.]\d+)?)(?P<units>[^\d]+)?')
|
|
|
|
def __init__(self, magnitude, units):
|
|
self.magnitude = magnitude
|
|
self.units = units
|
|
|
|
@classmethod
|
|
@abstractmethod
|
|
def parse_units(cls, value):
|
|
"""
|
|
Parse a string to a proper unit notation.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@classmethod
|
|
def fromstring(cls, string):
|
|
"""
|
|
Parse the string into a quantity object.
|
|
:param string:
|
|
:return:
|
|
"""
|
|
values = cls.parser_re.match(string).groupdict()
|
|
try:
|
|
magnitude = int(values['magnitude'])
|
|
except ValueError:
|
|
magnitude = float(values['magnitude'])
|
|
units = cls.parse_units(values['units'])
|
|
|
|
return cls(magnitude, units)
|
|
|
|
def __hash__(self):
|
|
return hash(str(self))
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, str):
|
|
return str(self) == other
|
|
if not isinstance(other, self.__class__):
|
|
return NotImplemented
|
|
return self.magnitude == other.magnitude and self.units == other.units
|
|
|
|
def __ne__(self, other):
|
|
return not self == other
|
|
|
|
def __repr__(self):
|
|
return f'<{self.__class__.__name__} [{self}]>'
|
|
|
|
def __str__(self):
|
|
return f'{self.magnitude}{self.units}'
|
|
|
|
|
|
class Size(Quantity):
|
|
"""
|
|
Represent size.
|
|
|
|
e.g.: 1.1GB, 300MB
|
|
"""
|
|
|
|
@classmethod
|
|
def parse_units(cls, value):
|
|
return value.strip(seps).upper()
|
|
|
|
|
|
class BitRate(Quantity):
|
|
"""
|
|
Represent bit rate.
|
|
|
|
e.g.: 320Kbps, 1.5Mbps
|
|
"""
|
|
|
|
@classmethod
|
|
def parse_units(cls, value):
|
|
value = value.strip(seps).capitalize()
|
|
for token in ('bits', 'bit'):
|
|
value = value.replace(token, 'bps')
|
|
|
|
return value
|
|
|
|
|
|
class FrameRate(Quantity):
|
|
"""
|
|
Represent frame rate.
|
|
|
|
e.g.: 24fps, 60fps
|
|
"""
|
|
|
|
@classmethod
|
|
def parse_units(cls, value):
|
|
return 'fps'
|