lib_one.py

from tabulate import tabulate
from datetime import date
import pandas as pd
import time, os

import custom_exceptions

'''
11013: 1분기
11012: 반기
11014: 3분기
11011: 사업보고서
'''
    
def timer(func):
    def Wrapper(*args, **kwargs):
        print(func.__name__)
        start = time.time()
        obj = func(*args, **kwargs)
        end = time.time()
        print(end - start)
        print("-" * 20)
        return obj
    return Wrapper

def get_date_today():
    today = date.today()
    x = today.weekday()
    today = pd.to_datetime(today)
    while x > 4:
        today -= pd.to_timedelta(1, 'D')
        x -= 1
    return today

def get_last_week(today: pd.Timestamp) -> pd.Timestamp:
    return (today - pd.to_timedelta(7, 'D'))

def get_six_years_list(year: str) -> list:
    '''
    Returns: List of strings, past six yrs.
    '''
    year = int(year)
    year_list = []
    for i in range(year-5, year+1):
        year_list.append(str(i))
    return year_list

def get_stock_price(sp_data, date: str) -> dict:
    '''
    Arguments: a dataframe object containing all the stock information of the company, the date
    Returns: the high/low/close stock price closest from the past to the date given.
    '''
    date = pd.to_datetime(date)

    #HAVE TO DO THIS CUZ THE sp_data returns Date as default index. So changing it back to column
    if sp_data.index[0] != 0: #IF THE INDEX OF THE DATA IS THE TIME, NOT 0,1,...
        sp_data.reset_index(inplace=True)  #SET TIME AS COLUMN AND REPLACE DEFAULT

    #If the latest stock price is not within one week range
    no_new_data = (date - sp_data["Date"].max()) > pd.Timedelta(days=7)

    if no_new_data or sp_data['Date'].min() > date:
        #If the there isn't any stock value equivalent
        raise custom_exceptions.StockPriceError("Given date not in the dataframe 'sp_data'", date)
    else:
        value = {}
        while True:
            if sum(sp_data['Date'] == date) > 0:
                sp_data = sp_data[sp_data['Date'] == date]
                value["Low"] = sp_data['Low'].iloc[0]
                value["High"] = sp_data['High'].iloc[0]
                value["Close"] = sp_data['Close'].iloc[0]
                break
            else:
                date -= pd.to_timedelta(1, 'day')
        return value
    
def get_cmpny_stock(cmpnyname: str):
    try:
        file_path = os.path.join(os.getcwd(), 'Data', 'Stocks', f"{cmpnyname}_stock.csv")
        df = pd.read_csv(file_path, encoding="euc-kr", parse_dates=["Date"])
    except FileNotFoundError:
        raise custom_exceptions.YoungCmpny(corp_code=cmpnyname, end_year="2022") #When the Stock doesn't match
    return df