You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.8 KiB
88 lines
2.8 KiB
from abc import ABCMeta
|
|
from ipaddress import ip_address
|
|
from typing import Iterable
|
|
|
|
|
|
class BaseStruct(object, metaclass=ABCMeta):
|
|
__slots__ = ()
|
|
|
|
|
|
# Как обслуживается абонент
|
|
class TariffStruct(BaseStruct):
|
|
__slots__ = ('tid', 'speedIn', 'speedOut')
|
|
|
|
def __init__(self, tariff_id=0, speed_in=None, speed_out=None):
|
|
self.tid = int(tariff_id)
|
|
self.speedIn = speed_in or 0
|
|
self.speedOut = speed_out or 0
|
|
|
|
# Yes, if all variables is zeroed
|
|
def is_empty(self):
|
|
return self.tid == 0 and self.speedIn == 0 and self.speedOut == 0
|
|
|
|
def __eq__(self, other):
|
|
# не сравниваем id, т.к. тарифы с одинаковыми скоростями для NAS одинаковы
|
|
# Да и иногда не удобно доставать из nas id тарифы из базы
|
|
return self.speedIn == other.speedIn and self.speedOut == other.speedOut
|
|
|
|
def __str__(self):
|
|
return "Id=%d, speedIn=%.2f, speedOut=%.2f" % (self.tid, self.speedIn, self.speedOut)
|
|
|
|
# нужно чтоб хеши тарифов In10,Out20 и In20,Out10 были разными
|
|
# поэтому сначала float->str и потом хеш
|
|
def __hash__(self):
|
|
return hash(str(self.speedIn) + str(self.speedOut))
|
|
|
|
|
|
# Abon from database
|
|
class AbonStruct(BaseStruct):
|
|
__slots__ = ('uid', '_ips', 'tariff', 'is_access', 'queue_id')
|
|
|
|
def __init__(self, uid=0, ips=None, tariff=None, is_access=True):
|
|
self.uid = int(uid or 0)
|
|
if ips is None:
|
|
self._ips = ()
|
|
else:
|
|
self._ips = tuple(ip_address(ip) for ip in ips)
|
|
self.tariff = tariff
|
|
self.is_access = is_access
|
|
self.queue_id = 0
|
|
|
|
def get_ips(self):
|
|
return self._ips
|
|
|
|
def set_ips(self, v):
|
|
self._ips = set(v)
|
|
|
|
ips = property(get_ips, set_ips, doc='Ip addresses')
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, AbonStruct):
|
|
raise TypeError
|
|
r = self.uid == other.uid and self._ips == other._ips
|
|
r = r and self.tariff == other.tariff
|
|
return r
|
|
|
|
def __str__(self):
|
|
return "uid=%d, ips=[%s], tariff=%s" % (self.uid, ';'.join(str(i) for i in self._ips), self.tariff or '<No Service>')
|
|
|
|
def __hash__(self):
|
|
return hash(hash(self._ips) + hash(self.tariff)) if self.tariff is not None else 0
|
|
|
|
|
|
# Shape rule from NAS(Network Access Server)
|
|
class ShapeItem(BaseStruct):
|
|
__slots__ = ('abon', 'sid')
|
|
|
|
def __init__(self, abon, sid):
|
|
self.abon = abon
|
|
self.sid = sid
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, ShapeItem):
|
|
raise TypeError
|
|
return self.sid == other.sid and self.abon == other.abon
|
|
|
|
|
|
VectorAbon = Iterable[AbonStruct]
|
|
VectorTariff = Iterable[TariffStruct]
|