gourav/PyFacts
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Changed frequency into a dataclass

Browse Source 
For more compact and pythonic code
This commit is contained in:
Gourav Kumar 2022-02-17 22:50:19 +05:30
parent 1f2b75282b
commit b6b2381163
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
fincal/fincal.py
View File

@ -11,15 +11,12 @@ class Options:
closest: str = 'before' # after closest: str = 'before' # after
@dataclass(frozen=True)
class Frequency: class Frequency:
def __init__(self, name, interval_type, interval_value, interval_days_value): name: str
self.name = name freq_type: str
self.type = interval_type value: int
self.value = interval_value days: int
self.days = interval_days_value
def __repr__(self):
return f"Frequency({self.name}, {self.type}, {self.value}, {self.days})"
class AllFrequencies: class AllFrequencies:
@ -43,7 +40,7 @@ def create_date_series(
dates = [] dates = []
for i in range(0, int(datediff)): for i in range(0, int(datediff)):
diff = {frequency.type: frequency.value*i} diff = {frequency.freq_type: frequency.value*i}
dates.append(start_date + relativedelta(**diff)) dates.append(start_date + relativedelta(**diff))
return dates return dates