lunar_data_with_ns_year_2078.py

#-*- encoding: utf-8 -*-

'''
    ONLY FOR 2078 BS
'''

import json
import copy

OUT_FILE = "2078_lunar_data.json"
# OUT_FILE_2 = "2078_detailed.json"
YEAR = 2078


lunar_month_list = ["चौलाथ्व","चौलागा","बछलाथ्व","बछलागा","तछलाथ्व","तछलागा","दिल्लाथ्व","दिल्लागा","गुँलाथ्व","गुँलागा","ञलाथ्व","ञलागा","कौलाथ्व","कौलागा","कछलाथ्व","कछलागा","थिल्लाथ्व","थिल्लागा","पोहेलाथ्व","पोहेलागा","सिल्लाथ्व","सिल्लागा","चिल्लाथ्व","चिल्लागा","चौलाथ्व","चौलागा"]
pakshya_list = ["चैत्र शुक्लपक्ष","वैशाख कृष्णपक्ष", "वैशाख शुक्लपक्ष", "ज्येष्ठ कृष्णपक्ष", "ज्येष्ठ शुक्लपक्ष", "आषाढ कृष्णपक्ष", "आषाढ शुक्लपक्ष", "श्रावण कृष्णपक्ष", "श्रावण शुक्लपक्ष", "भाद्र कृष्णपक्ष", "भाद्र शुक्लपक्ष", "आश्विन कृष्णपक्ष", "आश्विन शुक्लपक्ष", "कार्तिक कृष्णपक्ष", "कार्तिक शुक्लपक्ष", "मंसिर कृष्णपक्ष", "मंसिर शुक्लपक्ष", "पौष कृष्णपक्ष", "पौष शुक्लपक्ष", "माघ कृष्णपक्ष", "माघ शुक्लपक्ष", "फाल्गुण कृष्णपक्ष", "फाल्गुण शुक्लपक्ष", "चैत्र कृष्णपक्ष", "चैत्र शुक्लपक्ष","वैशाख कृष्णपक्ष"]


def get_lunar_month(date):
    ''' only valid  for 2078 BS '''
    date_split = date.split("-")
    date_month = int(date_split[1])
    date_day = int(date_split[2])

    if date_month == 1 and (date_day >= 1 and date_day <= 14):
        return lunar_month_list[0]
    elif date_month == 1 and (date_day >= 15 and date_day <= 28):
        return lunar_month_list[1]
    elif (date_month == 1 and (date_day >= 29 and date_day <= 31)) or (date_month == 2 and (date_day >= 1 and date_day <= 12)):
        return lunar_month_list[2]
    elif date_month == 2 and (date_day >= 13 and date_day <= 27):
        return lunar_month_list[3]
    elif date_month == 2 and (date_day >= 28 and date_day <= 31) or (date_month == 3 and (date_day >= 1 and date_day <= 10)):
        return lunar_month_list[4]
    elif date_month == 3 and (date_day >= 11 and date_day <= 26):
        return lunar_month_list[5]
    elif date_month == 3 and (date_day >= 27 and date_day <= 31) or (date_month == 4 and (date_day >= 1 and date_day <= 9)):
        return lunar_month_list[6]
    elif date_month == 4 and (date_day >= 10 and date_day <= 24):
        return lunar_month_list[7]
    elif date_month == 4 and (date_day >= 25 and date_day <= 32) or (date_month == 5 and (date_day >= 1 and date_day <= 6)):
        return lunar_month_list[8]
    elif date_month == 5 and (date_day >= 7 and date_day <= 22):
        return lunar_month_list[9]
    elif date_month == 5 and (date_day >= 23 and date_day <= 31) or (date_month == 6 and (date_day >= 1 and date_day <= 4)):
        return lunar_month_list[10]
    elif date_month == 6 and (date_day >= 5 and date_day <= 20):
        return lunar_month_list[11]
    elif date_month == 6 and (date_day >= 21 and date_day <= 31) or (date_month == 7 and (date_day >= 1 and date_day <= 3)):
        return lunar_month_list[12]
    elif date_month == 7 and (date_day >= 4 and date_day <= 18):
        return lunar_month_list[13]
    elif date_month == 7 and (date_day >= 19 and date_day <= 30) or (date_month == 8 and (date_day >= 1 and date_day <= 3)):
        return lunar_month_list[14]
    elif date_month == 8 and (date_day >= 4 and date_day <= 18):
        return lunar_month_list[15]
    elif date_month == 8 and (date_day >= 19 and date_day <= 29) or (date_month == 9 and (date_day >= 1 and date_day <= 4)):
        return lunar_month_list[16]
    elif date_month == 9 and (date_day >= 5 and date_day <= 18):
        return lunar_month_list[17]
    elif date_month == 9 and (date_day >= 19 and date_day <= 30) or (date_month == 10 and (date_day >= 1 and date_day <= 3)):
        return lunar_month_list[18]
    elif date_month == 10 and (date_day >= 4 and date_day <= 18):
        return lunar_month_list[19]
    elif date_month == 10 and (date_day >= 19 and date_day <= 29) or (date_month == 11 and (date_day >= 1 and date_day <= 4)):
        return lunar_month_list[20]
    elif date_month == 11 and (date_day >= 5 and date_day <= 18):
        return lunar_month_list[21]
    elif date_month == 11 and (date_day >= 19 and date_day <= 30) or (date_month == 12 and (date_day >= 1 and date_day <= 4)):
        return lunar_month_list[22]
    elif date_month == 12 and (date_day >= 5 and date_day <= 18):
        return lunar_month_list[23]
    elif date_month == 12 and (date_day >= 19 and date_day <= 30):
        return lunar_month_list[24]
    return date + " undefined lunar month"


def get_pakshya(date):
    ''' only valid  for 2078 BS '''
    date_split = date.split("-")
    date_month = int(date_split[1])
    date_day = int(date_split[2])

    if date_month == 1 and (date_day >= 1 and date_day <= 14):
        return pakshya_list[0]
    elif date_month == 1 and (date_day >= 15 and date_day <= 28):
        return pakshya_list[1]
    elif (date_month == 1 and (date_day >= 29 and date_day <= 31)) or (date_month == 2 and (date_day >= 1 and date_day <= 12)):
        return pakshya_list[2]
    elif date_month == 2 and (date_day >= 13 and date_day <= 27):
        return pakshya_list[3]
    elif date_month == 2 and (date_day >= 28 and date_day <= 31) or (date_month == 3 and (date_day >= 1 and date_day <= 10)):
        return pakshya_list[4]
    elif date_month == 3 and (date_day >= 11 and date_day <= 26):
        return pakshya_list[5]
    elif date_month == 3 and (date_day >= 27 and date_day <= 31) or (date_month == 4 and (date_day >= 1 and date_day <= 9)):
        return pakshya_list[6]
    elif date_month == 4 and (date_day >= 10 and date_day <= 24):
        return pakshya_list[7]
    elif date_month == 4 and (date_day >= 25 and date_day <= 32) or (date_month == 5 and (date_day >= 1 and date_day <= 6)):
        return pakshya_list[8]
    elif date_month == 5 and (date_day >= 7 and date_day <= 22):
        return pakshya_list[9]
    elif date_month == 5 and (date_day >= 23 and date_day <= 31) or (date_month == 6 and (date_day >= 1 and date_day <= 4)):
        return pakshya_list[10]
    elif date_month == 6 and (date_day >= 5 and date_day <= 20):
        return pakshya_list[11]
    elif date_month == 6 and (date_day >= 21 and date_day <= 31) or (date_month == 7 and (date_day >= 1 and date_day <= 3)):
        return pakshya_list[12]
    elif date_month == 7 and (date_day >= 4 and date_day <= 18):
        return pakshya_list[13]
    elif date_month == 7 and (date_day >= 19 and date_day <= 30) or (date_month == 8 and (date_day >= 1 and date_day <= 3)):
        return pakshya_list[14]
    elif date_month == 8 and (date_day >= 4 and date_day <= 18):
        return pakshya_list[15]
    elif date_month == 8 and (date_day >= 19 and date_day <= 29) or (date_month == 9 and (date_day >= 1 and date_day <= 4)):
        return pakshya_list[16]
    elif date_month == 9 and (date_day >= 5 and date_day <= 18):
        return pakshya_list[17]
    elif date_month == 9 and (date_day >= 19 and date_day <= 30) or (date_month == 10 and (date_day >= 1 and date_day <= 3)):
        return pakshya_list[18]
    elif date_month == 10 and (date_day >= 4 and date_day <= 18):
        return pakshya_list[19]
    elif date_month == 10 and (date_day >= 19 and date_day <= 29) or (date_month == 11 and (date_day >= 1 and date_day <= 4)):
        return pakshya_list[20]
    elif date_month == 11 and (date_day >= 5 and date_day <= 18):
        return pakshya_list[21]
    elif date_month == 11 and (date_day >= 19 and date_day <= 30) or (date_month == 12 and (date_day >= 1 and date_day <= 4)):
        return pakshya_list[22]
    elif date_month == 12 and (date_day >= 5 and date_day <= 18):
        return pakshya_list[23]
    elif date_month == 12 and (date_day >= 19 and date_day <= 30):
        return pakshya_list[24]
    return date + " undefined pakshya"

def get_nepal_sambat(date):
    ''' only valid  for 2078 BS '''
    date_split = date.split("-")
    date_month = int(date_split[1])
    date_day = int(date_split[2])

    ''' this is for 2078 '''
    if date_month < 7:
        return "1141"
    elif date_month >= 8:
        return "1142"
    elif date_month == 7:
        if date_day < 19:
            return "1141"
        else:
            return "1142"
    return " n/a "

if __name__ == "__main__":
    in_file = open("2078.json", "r+")
    prev_data = json.load(in_file)
    out_dict = {}

    for month in range(12):
        for day in range(32):
            try:
                day_dict = prev_data["data"][month][day]
                day_date = day_dict["date"]
                day_dict["lunar_month"] = get_lunar_month(day_date)
                day_dict["pakshya"] = get_pakshya(day_date)
                day_dict["ns_year"] = get_nepal_sambat(day_date)

                out_dict[day_date] = [day_dict["lunar_month"], day_dict["pakshya"], day_dict["tithi"], day_dict["ns_year"]]
                # out_dict[day_date] = [day_dict["lunar_month"], day_dict["pakshya"], day_dict["tithi"]]

                # print(day_date),
                # print("\t"),
                # print(get_lunar_month(day_date))
            except:
                pass

    with open(OUT_FILE, 'w') as file:
        json.dump(out_dict, file)

    # with open(OUT_FILE_2, 'w') as file2:
    #     json.dump(prev_data, file2)