dt_range.py

from datetime import datetime, timedelta
from functools import total_ordering


@total_ordering
class dt_range:
    def __init__(self, start_dt, end_dt):
        if start_dt > end_dt:
            raise ValueError('start_dt must be before end_dt')

        self.start_dt = start_dt
        self.end_dt = end_dt

    @property
    def delta(self):
        return timedelta(self.end_dt - self.start_dt)

    def __contains__(self, other):
        if isinstance(type(self)):
            return (self.start_dt <= other.start_dt) and (self.end_dt >= other.end_dt)
        elif isinstance(datetime):
            return (self.start_dt <= other <= self.end_dt)

        raise ValueError('invalid operand')

    def __eq__(self, other):
        if not isinstance(type(self)):
            return NotImplemented
        
        return (self.start_dt == other.start_dt) and (self.end_dt == other.end_dt)

    def __hash__(self):
        return hash((self.start_dt, self.end_dt))

    def __lt__(self, other):
        if not isinstance(type(self)):
            return NotImplemented
        
        return (self.start_dt < other.start_dt) and (self.end_dt < other.end_dt)
    
    def __repr__(self):
        return 'dt_range({}, {})'.format(repr(self.start_dt), repr(self.end_dt))

    def __str__(self):
        return '[{}, {}]'.format(self.start_dt, self.end_dt)