# Licensed under a 3-clause BSD style license - see LICENSE.rst

# Standard library

import re

import textwrap

import warnings

from datetime import datetime

from urllib.request import Request, urlopen

# Third-party

from astropy import time as atime

from astropy.utils.console import _color_text, color_print

from .funcs import get_sun

__all__ = []

class HumanError(ValueError):

pass

class CelestialError(ValueError):

pass

def get_sign(dt):

""" """

if (int(dt.month) == 12 and int(dt.day) >= 22) or (

int(dt.month) == 1 and int(dt.day) <= 19

):

zodiac_sign = "capricorn"

elif (int(dt.month) == 1 and int(dt.day) >= 20) or (

int(dt.month) == 2 and int(dt.day) <= 17

):

zodiac_sign = "aquarius"

elif (int(dt.month) == 2 and int(dt.day) >= 18) or (

int(dt.month) == 3 and int(dt.day) <= 19

):

zodiac_sign = "pisces"

elif (int(dt.month) == 3 and int(dt.day) >= 20) or (

int(dt.month) == 4 and int(dt.day) <= 19

):

zodiac_sign = "aries"

elif (int(dt.month) == 4 and int(dt.day) >= 20) or (

int(dt.month) == 5 and int(dt.day) <= 20

):

zodiac_sign = "taurus"

elif (int(dt.month) == 5 and int(dt.day) >= 21) or (

int(dt.month) == 6 and int(dt.day) <= 20

):

zodiac_sign = "gemini"

elif (int(dt.month) == 6 and int(dt.day) >= 21) or (

int(dt.month) == 7 and int(dt.day) <= 22

):

zodiac_sign = "cancer"

elif (int(dt.month) == 7 and int(dt.day) >= 23) or (

int(dt.month) == 8 and int(dt.day) <= 22

):

zodiac_sign = "leo"

elif (int(dt.month) == 8 and int(dt.day) >= 23) or (

int(dt.month) == 9 and int(dt.day) <= 22

):

zodiac_sign = "virgo"

elif (int(dt.month) == 9 and int(dt.day) >= 23) or (

int(dt.month) == 10 and int(dt.day) <= 22

):

zodiac_sign = "libra"

elif (int(dt.month) == 10 and int(dt.day) >= 23) or (

int(dt.month) == 11 and int(dt.day) <= 21

):

zodiac_sign = "scorpio"

elif (int(dt.month) == 11 and int(dt.day) >= 22) or (

int(dt.month) == 12 and int(dt.day) <= 21

):

zodiac_sign = "sagittarius"

return zodiac_sign

_VALID_SIGNS = [

"capricorn",

"aquarius",

"pisces",

"aries",

"taurus",

"gemini",

"cancer",

"leo",

"virgo",

"libra",

"scorpio",

"sagittarius",

]

# Some of the constellation names map to different astrological "sign names".

# Astrologers really needs to talk to the IAU...

_CONST_TO_SIGNS = {"capricornus": "capricorn", "scorpius": "scorpio"}

_ZODIAC = (

(1900, "rat"),

(1901, "ox"),

(1902, "tiger"),

(1903, "rabbit"),

(1904, "dragon"),

(1905, "snake"),

(1906, "horse"),

(1907, "goat"),

(1908, "monkey"),

(1909, "rooster"),

(1910, "dog"),

(1911, "pig"),

)

# https://stackoverflow.com/questions/12791871/chinese-zodiac-python-program

def _get_zodiac(yr):

return _ZODIAC[(yr - _ZODIAC[0][0]) % 12][1]

def horoscope(birthday, corrected=True, chinese=False):

"""

Enter your birthday as an `astropy.time.Time` object and

receive a mystical horoscope about things to come.

Parameters

----------

birthday : `astropy.time.Time` or str

Your birthday as a `datetime.datetime` or `astropy.time.Time` object

or "YYYY-MM-DD"string.

corrected : bool

Whether to account for the precession of the Earth instead of using the

ancient Greek dates for the signs. After all, you do want your *real*

horoscope, not a cheap inaccurate approximation, right?

chinese : bool

Chinese annual zodiac wisdom instead of Western one.

Returns

-------

Infinite wisdom, condensed into astrologically precise prose.

Notes

-----

This function was implemented on April 1. Take note of that date.

"""

from bs4 import BeautifulSoup

today = datetime.now()

err_msg = "Invalid response from celestial gods (failed to load horoscope)."

headers = {"User-Agent": "foo/bar"}

special_words = {

"([sS]tar[s^ ]*)": "yellow",

"([yY]ou[^ ]*)": "magenta",

"([pP]lay[^ ]*)": "blue",

"([hH]eart)": "red",

"([fF]ate)": "lightgreen",

}

if isinstance(birthday, str):

birthday = datetime.strptime(birthday, "%Y-%m-%d")

if chinese:

# TODO: Make this more accurate by using the actual date, not just year

# Might need third-party tool like https://pypi.org/project/lunardate

zodiac_sign = _get_zodiac(birthday.year)

url = (

"https://www.horoscope.com/us/horoscopes/yearly/"

f"{today.year}-chinese-horoscope-{zodiac_sign}.aspx"

)

summ_title_sfx = f"in {today.year}"

try:

res = Request(url, headers=headers)

with urlopen(res) as f:

try:

doc = BeautifulSoup(f, "html.parser")

# TODO: Also include Love, Family & Friends, Work, Money, More?

item = doc.find(id="overview")

desc = item.getText()

except Exception:

raise CelestialError(err_msg)

except Exception:

raise CelestialError(err_msg)

else:

birthday = atime.Time(birthday)

if corrected:

with warnings.catch_warnings():

warnings.simplefilter("ignore") # Ignore ErfaWarning

zodiac_sign = get_sun(birthday).get_constellation().lower()

zodiac_sign = _CONST_TO_SIGNS.get(zodiac_sign, zodiac_sign)

if zodiac_sign not in _VALID_SIGNS:

raise HumanError(

f"On your birthday the sun was in {zodiac_sign.title()}, which is"

" not a sign of the zodiac. You must not exist. Or maybe you can"

" settle for corrected=False."

)

else:

zodiac_sign = get_sign(birthday.to_datetime())

url = f"https://astrology.com/horoscope/daily/{zodiac_sign}.html"

summ_title_sfx = f"on {today.strftime('%Y-%m-%d')}"

res = Request(url, headers=headers)

with urlopen(res) as f:

try:

doc = BeautifulSoup(f, "html.parser")

item = doc.find("div", {"id": "content"})

desc = item.getText()

except Exception:

raise CelestialError(err_msg)

print("*" * 79)

color_print(f"Horoscope for {zodiac_sign.capitalize()} {summ_title_sfx}:", "green")

print("*" * 79)

for block in textwrap.wrap(desc, 79):

split_block = block.split()

for i, word in enumerate(split_block):

for re_word in special_words.keys():

match = re.search(re_word, word)

if match is None:

continue

split_block[i] = _color_text(match.groups()[0], special_words[re_word])

print(" ".join(split_block))

def inject_horoscope():

import astropy

astropy._yourfuture = horoscope

inject_horoscope()