company.py

from load_files import update_data
import csv
import math

class Company:
    """
    Represents a Company
    """
    def __init__(self, id: str):
        """
        Initialize a new company
        """
        self.id = id
        self.dates = []
        self.volumes = []
        self.close_prices = []
        self.high_prices = []
        self.low_prices = []
        self.open_prices = []
        self.company_name = ""
        self.description = ""
        self.sector = ""
        self.market_cap = 0
        self.get_market_cap()
        self.populate_storage()
    def populate_storage(self):
        """
        Fills up self.dates,volumes,and prices with info
        """
        path = 'company_library/' + self.id + '.csv'
        try:
            with open(path,'r') as csv_file:
                csv_reader = csv.reader(csv_file)
                for line in csv_reader:
                    temp_volume = line[8]
                    temp_close_price = line[2]
                    temp_high_price = line[3]
                    temp_open_price = line[6]
                    temp_low_price = line[5]
                    temp_date = line[1]
                    try:
                        temp_dates = temp_date.split("-")
                        temp_date = temp_dates[0] + temp_dates[1] + temp_dates[2]
                        temp_date = int(temp_date)

                        self.low_prices.append(float(temp_low_price))
                        self.high_prices.append(float(temp_high_price))
                        self.dates.append(temp_date)
                        self.volumes.append(float(temp_volume))
                        self.close_prices.append(float(temp_close_price))
                        self.open_prices.append(float(temp_open_price))
                    except:
                        #removes top element
                        pass
        except:
            print(self.id, "ommitted from update (id not found in company_library folder)")

    def get_market_cap(self):
        path = "companylist.csv"
        try:
            with open(path,'r') as csv_file:
                csv_reader = csv.reader(csv_file)
                for line in csv_reader:
                    if line[0] == self.id.upper():
                        self.company_name = line[1]
                        self.sector = line[6]
                        self.description = line[7]
                        self.market_cap = float(line[3])

                        break
        except:
            pass
    def get_change_in_price(self, start, end):
        """
        Return the change in price over the start and end dates
        :param start:
        :param end:
        :return:
        """
        if start not in self.dates or end not in self.dates:
            raise IndexError
        start_index = self.dates.index(start)
        end_index = self.dates.index(end)
        return (self.close_prices[end_index] - self.open_prices[start_index])/self.open_prices[start_index]



    def getavg(self, start: int, end: int, category: str):
        """
        RETURN THE AVG VOLUME/PRICE
        """
        if start not in self.dates or end not in self.dates:
            raise IndexError

        start_index = self.dates.index(start)
        end_index = self.dates.index(end)
        if category == "PRICE":
            sum = 0
            count = 0
            for i in range(start_index,end_index+1):
                sum += self.close_prices[i]
                count += 1
            sum = sum/count
            return sum
        else:
            sum = 0
            count = 0
            for i in range(start_index,end_index+1):
                sum += self.volumes[i]
                count += 1
            sum = sum/count
            return sum

    def standard_deviation(self):
        x = self.getavg(self.dates[len(self.dates)-9], self.dates[len(self.dates)-1], "PRICE")
        print("AVERAGE: ", x)
        sum = 0
        for i in range(len(self.dates)-9,len(self.dates)):
            print("CLOSE PRICE: ", self.close_prices[i])
            sum += (self.close_prices[i] - x)*(self.close_prices[i]-x)
        sum = sum/9
        return sum

    def get_true_range(self,date,n):
        day = self.dates.index(date)-1
        TR_total = 0
        for i in range(1,n+1):
            today = day-i #TODAY
            yesterday = day - i - 1 #YESTERDAY




            today_high = self.high_prices[today]
            today_low = self.low_prices[today]


            yesterday_close =self.close_prices[yesterday]

            TR = max(today_high-today_low,abs(today_high-yesterday_close),abs(today_low-yesterday_close))
            TR_total += TR
        return TR_total/(n)


    def __str__(self):
        """
        Return the ID of the company
        """
        return self.id



if __name__ == "__main__":
    x = Company('JASO')
    print(x.market_cap)