from datetime import datetime, timedelta
from unittest import result
from dateutil.relativedelta import relativedelta
import pytz

class utc_time():
    # Get current time
    def __init__(self) -> None:
        self.current_time = datetime.now()
        
    def get(self):
       return self.current_time.astimezone(pytz.UTC)
   
    def add(self, amount, unit):
        if unit == 'year':
            self.current_time += relativedelta(years=int(amount))
        elif unit == 'month':
            self.current_time += relativedelta(months=int(amount))
        elif unit == 'week':
            self.current_time += relativedelta(weeks=int(amount))
        elif unit == 'day':
            self.current_time += relativedelta(days=int(amount))
        elif unit == 'hour':
            self.current_time += relativedelta(hours=int(amount))
        elif unit == 'minute':
            self.current_time += relativedelta(minutes=int(amount))
        elif unit == 'second':
            self.current_time += relativedelta(seconds=int(amount))
        elif unit == 'millisecond':
            self.current_time += relativedelta(microseconds=int(amount)*1000)
        else: raise Exception("[date_utils] Error: Add unit not found.")
        # Add more units as needed...
        return self
    
    
def convert_utc(dt):
    utc_time = dt.astimezone(pytz.UTC)
    return utc_time

def startOf(dt):
    start_of = datetime(dt.year, dt.month, dt.day)
    return start_of

def endOf(dt):
    end_of = startOf(dt) + timedelta(days=1, seconds=-1)
    return end_of