from params import avoid_table, marana_yoga_table
from params import NAKSHATRAS
import numpy as np
user_nakshatra = NAKSHATRAS.MAGAM
num_days = 90
mins_step = 30
a = Almanac()
t0 = a.time_scale.utc(2020, 4, 1)
t1 = a.time_scale.utc(2020, 4, 1, 0, 24 * 60 * 366)
month_times, months = a._get_month_start_end(t0, t1)
# sun_rise, sun_set = a._compute_sun_rise_sun_set(t0, t1)
tframe = a.time_scale.utc(2020, 4, 1, 0, range(0, 24 * 60 * num_days,
mins_step))
a.time = tframe
a.compute_almanac()
mask = np.ones(tframe.shape, dtype=np.bool)
print(f'started with {mask.sum()}')
for t in avoid_thitis_list:
mask[(t == a.thithi.name)] = False
print(f'after removing thithis {avoid_thitis_list} -- {mask.sum()}')
for n in nakshaktra_ignore_list:
mask[n == a.nakshaktra.name] = False
print(f'after removing nakshaktras {nakshaktra_ignore_list} -- {mask.sum()}')
for day, thithis in avoid_thithi_on_days.items():
cur_days = (a.vara == day)
for t in thithis:
mask[(t == a.thithi.name) & cur_days] = False
print(f'after removing thithi on days {avoid_thithi_on_days} -- {mask.sum()}')
for day, nakshaktras in avoid_nakshaktras_on_day.items():
