Skip to content

Commit

Permalink
Manage options to build a date range depending on user inputs
Browse files Browse the repository at this point in the history
  • Loading branch information
@Bachibouzouk
Bachibouzouk committed Nov 30, 2023
1 parent 1826f38 commit bed4fd3
Showing 1 changed file with 80 additions and 54 deletions.
134 changes: 80 additions & 54 deletions ramp/core/core.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -69,12 +69,12 @@ def __init__(
"""
self.name = name
self._date_start = (
datetime.date.fromisoformat(date_start)
datetime.datetime.fromisoformat(date_start)
if isinstance(date_start, str)
else date_start
)
self._date_end = (
datetime.date.fromisoformat(date_end)
datetime.datetime.fromisoformat(date_end)
if isinstance(date_end, str)
else date_end
)
Expand Down Expand Up @@ -103,7 +103,7 @@ def date_start(self):
@date_start.setter
def date_start(self, new_date):
self._date_start = (
datetime.date.fromisoformat(new_date)
datetime.datetime.fromisoformat(new_date)
if isinstance(new_date, str)
else new_date
)
Expand All @@ -115,7 +115,7 @@ def date_end(self):
@date_end.setter
def date_end(self, new_date):
self._date_end = (
datetime.date.fromisoformat(new_date)
datetime.datetime.fromisoformat(new_date)
if isinstance(new_date, str)
else new_date
)
Expand Down Expand Up @@ -174,70 +174,96 @@ def datetimeindex(self):
def initialize(self, num_days=None, peak_enlarge=None, force=False):
# TODO allow calendar years multi-year inputs

if num_days is not None:
self.__num_days = num_days
if self.days is not None:
# logging.error
print(
"You want to initialize the usecase with num_days but you already have provided days. This might be the case if you use UseCase.load before you used UseCase.initialize()"
)
self.__num_days = None

if self.date_start is not None and self.date_end is not None:
self.days = pd.date_range(
start=self.date_start, end=self.date_end
) # TODO add one extra day
self.__num_days = len(self.days)

if self.__num_days is None:
# asks the user how many days (i.e. code runs) they want
self.__num_days = int(
input("please indicate the number of days to be generated: ")
if self.is_initialized is True and force is False:
# logging.warning
print(
f"The usecase '{self.name}' is already initialized. If you want to force the reinitialization, use argument `force=True`\n"
f"You will typically see this message if you provide 'date_start' and 'date_end' to a UseCase instance and want to call the method initialize() on top of that"
)
print("Please wait...")

if self.days is None:
# TODO add 24 hours to date end in the display
if self.date_start is not None:
self.days = pd.date_range(
start=self.date_start, periods=self.__num_days
)
# logging info
else:
if self.is_initialized is True and force is True:
# logging.info
print(
f"You will simulate {self.__num_days} days from {self.date_start} until {self.days[-1]}"
f"The usecase '{self.name}' is already initialized but argument `force=True` was provided, reinitializing..."
)
else:
if self.date_end is not None:
self.days = pd.date_range(
end=self.date_end, periods=self.__num_days
)
# logging info
self.days = None

if num_days is not None:
self.__num_days = num_days
if self.days is not None:
# logging.error
print(
f"You will simulate {self.__num_days} days from {self.days[0]} until {self.date_end}"
f"You want to initialize the usecase '{self.name}' with num_days but you already have provided days. This might be the case if you use UseCase.load before you used UseCase.initialize()"
)
self.__num_days = None

if self.date_start is not None and self.date_end is not None:
self.days = pd.date_range(
start=self.date_start, end=self.date_end
) # TODO add one extra day
if self.__num_days is not None:
if self.__num_days != len(self.days):
# logging.warning
print(
"You provided arguments 'date_start' and 'date_end' to your usecase. However you provided argument 'num_days' to 'initialize' method, this will build a date range from 'date_start' with 'num_days' days"
)
self.days = None
else:
self.__num_days = len(self.days)
else:
self.__num_days = len(self.days)

if self.__num_days is None:
# asks the user how many days (i.e. code runs) they want
self.__num_days = int(
input("please indicate the number of days to be generated: ")
)
print("Please wait...")

if self.days is None:
# TODO add 24 hours to date end in the display
if self.date_start is not None:
self.days = pd.date_range(
start=datetime.datetime.today(), periods=self.__num_days
start=self.date_start, periods=self.__num_days
)
# logging info
print(
f"You will simulate {self.__num_days} days from {self.days[0]} until {self.days[-1]}"
f"You will simulate {self.__num_days} day(s) from {self.date_start} until {self.days[-1]+datetime.timedelta(days=1)}"
)
else:
if self.date_end is not None:
self.days = pd.date_range(
end=self.date_end, periods=self.__num_days
)
# logging info
print(
f"You will simulate {self.__num_days} day(s) from {self.days[0]} until {self.date_end+datetime.timedelta(days=1)}"
)

else:
print(
f"You will simulate {self.__num_days} days from {self.days[0]} until {self.days[-1]}"
else:
today = datetime.datetime.today()

self.days = pd.date_range(
start=datetime.datetime(today.year, today.month, today.day),
periods=self.__num_days,
)
# logging info
print(
f"You will simulate {self.__num_days} day(s) from {self.days[0]} until {self.days[-1]+datetime.timedelta(days=1)}"
)

else:
print(
f"You will simulate {self.__num_days} day(s) from {self.days[0]} until {self.days[-1]+datetime.timedelta(days=1)}"
)
self.peak_time_range = self.calc_peak_time_range(peak_enlarge=peak_enlarge)
# format datetimeindex in minutes
self.__datetimeindex = pd.date_range(
start=self.days[0],
end=self.days[-1] + pd.Timedelta(1, "d") - pd.Timedelta(1, "T"),
freq="T",
)
self.peak_time_range = self.calc_peak_time_range(peak_enlarge=peak_enlarge)
# format datetimeindex in minutes
self.__datetimeindex = pd.date_range(
start=self.days[0],
end=self.days[-1] + pd.Timedelta(1, "d") - pd.Timedelta(1, "T"),
freq="T",
)

def calc_peak_time_range(self, peak_enlarge=0.15):
@property
def is_initialized(self):
answer = False
Expand Down Expand Up @@ -301,7 +327,7 @@ def calc_peak_time_range(self, peak_enlarge=None):
return np.arange(peak_time - rand_peak_enlarge, peak_time + rand_peak_enlarge)

def generate_daily_load_profiles(
self, days=None, flat=False, cases=None, verbose=False
self, days=None, flat=True, cases=None, verbose=False
):
"""
Iterate over the days and generate a daily profile for each of the days
Expand Down

0 comments on commit bed4fd3

Please sign in to comment.